[[TOC]]
-
下载数据集放入文件夹中
-
运行
write_txt.py
生成训练以及测试需要用到的文本文件 -
运行
train.py
开始训练
-
将
xml
文件放入\VOCdevkit\VOC2007\Annotations
-
将
.jpg
文件放入\VOCdevkit\VOC2007\JPEGImages
-
更改
write_txt.py
中的VOC_CLASSES
-
更改
yoloData.py
,yoloLoss.py
与new_resnet.py
中的CLASS_NUM
-
运行
write_txt.py
生成训练以及测试需要用到的文本文件 -
运行
train.py
开始训练
-
测试文件为
predict.py
-
加载直接训练好的权重文件放到
model.load_state_dict(torch.load(your model weight path))
,你的测试图片路径img_root
-
直接运行即可