该库 fork 自 @github/marvis 的 pytorch-yolo2,不过作者没有直接修改或者更新 marvis 的源文件,因为很多文件已经改了文件名。所以本库和源文件有很大的差异,主要差异有以下几点:
如果您想了解训练和检测程序,请参阅 https://github.com/marvis/pytorch-yolo2 了解详细信息。
Github 链接:
https://github.com/andy-yun/pytorch-0.4-yolov3
按如下方式训练你的数据、coco、voc data:
python train.py -d cfg/coco.data -c cfg/yolo_v3.cfg -w yolov3.weights
python train.py -d cfg/my.data -c cfg/my.cfg -w yolov3.weights -r
最大的 epochs 选项是自动计算的,有时候会很小,所以你要在设置中修改 max_epochs。
yolo2 训练记录,链接:
https://www.youtube.com/watch?v=jhoaVeqtOQw
yolo3 训练记录,链接:
https://www.youtube.com/watch?v=zazKAm9FClc
详细信息请查阅官方 Github 页面:
https://github.com/andy-yun/pytorch-0.4-yolov3