yolov3训练自己的数据集
从图片和xml的保存,到训练
保存地址随便,但图片名,要和xml的命名对应。因为convert_to_txt.py文件是以xml的命名为切分,而voc_annotation.py文件是以convert_to_txt.py生成的文件来遍历生成文件。
下面是详细步骤:
1,图片保存
在convert_to_txt.py调整自己的路径,和生成文件的路径,自己记得就行,
然后运行。就生成了如下
然后执行voc_annotation.py,要记得之前自己生成文件的位置,和图片,xml文件保存的位置,voc_annotation.py文件会生成图片路径与预测框,这能使图片与预测框对应,方便烂序。
这里voc_annotation.py中的classes = [“person”, “car”, “flower”]要与xml
中的name对应,否则就遍历不到框。
然后就会生成
生成的文件中是图片的链接与框等信息
然后修改yolov3.cfg文件
一共有3个【yolo】和3个【convolutional】要改
检查train.py是否调用voc_annotation.py生成的训练文件
最后运行train.py文件就可以了
可能的报错
可能运行voc_annotation.py文件后,生成的文件中只有图片的路径,例如:
这是因为有xml文件,没有图片,有图片没有xml文件。
进行改进和防止报错
怎么改源文件防止有xml文件,没有图片,有图片没有xml文件。
在文件中添加