You can see more details about Yolo by clicking the link below:
https://zhuanlan.zhihu.com/p/64051698
You can download pretrained ResNet-Backbone YOLOv1 weight here:
https://pan.baidu.com/s/1YnPqOepzAbr9T_z4Ux-Ocg
- auto-save and load mechanism, default dir is './model'
- base-net can be chose from pre-trained ResNet18、ResNet50 and ResNet101
the data set dir should be like this:
base_dir
VOC2007
Annotations
ImageSets
JPEGImages
SegmentationClass
SegmentationObject
VOC2012
Annotations
ImageSets
JPEGImages
SegmentationClass
SegmentationObject
you can setup the base_dir by this:
python train.py --voc_data_set_root /base_dir
-
you can train model by this:
-
python train.py --voc_data_set_root /media/charles/750GB/VOC0712trainval --num_workers 12 --batch_size 16 --backbone resnet50 --save_step 500
-
assign the dir of pic in predict.py and run to see the demo
-
model weights will be uploaded afterwards