PyTorch_YOLOv2

custom augmentation for training yolov2 model with voc dataset

Setting up for the training

sh setup1.sh

conda activate simp
sh setup2.sh

This is compatible for training using Google collab (which I used while training this), but making a ssh connection with the GPU is recommended

sh train.sh

Name		Name	Last commit message	Last commit date
Latest commit History 51 Commits
backbone		backbone
config		config
data		data
evaluator		evaluator
models		models
setup		setup
utils		utils
.gitignore		.gitignore
README.md		README.md
eval.py		eval.py
test.py		test.py
tools.py		tools.py
train.py		train.py
train.sh		train.sh