Tensorflow版yolov3训练自己的数据
源代码:https://github.com/YunYang1994/TensorFlow2.0-Examples/tree/master/4-Object_Detection/YOLOV3
本机可运行的环境如下:
1. 下载完代码后首先在主/根目录下建立train和test文件夹,目录树如下:
test为测试集,train为训练集,与voc_annotation.py文件中设置的路径需一致才能生成voc_train.txt和voc_test.txt
2. 打开voc_annotation.py修改如下:
修改自己需要训练的类别