yolov3中,如果要使得训练模型识别的更准确,可以修改anchors之后再重新训练,接下来如何去获取我们自己的数据的anchors

首先需要两样东西

一:存放着自己数据集图片路径的2007_train.txt文件,如下图,这是我的在ubuntu系统下的.txt文件(涂抹去了我的个人信息)

yolov3获取自己数据集的anchors

二:生成anchors的脚本文件gen_anchors.py

然后就是对gen_anchors.py文件的修改,首先找到文件里如下两处的地方

(1)
yolov3获取自己数据集的anchors
(2)
yolov3获取自己数据集的anchors

找到(1)处地方后将其修改为你训练时候输入的尺寸大小(和训练的.cfg文件中的width,height一致)

找到(2)处地方后,可将①位置的2007_train.txt修改为你的.txt文件名,或者你将你的.txt文件改成2007_train.txt也可以(这样就不用修改①位置),同时要把txt文件和py文件放在同一目录中

找到(2)处地方后,在该目录下新建一个名为②位置(generated_anchors)的文件夹

做完如上步骤后,python环境下运行gen_anchors.py,运行完成后,在新建的文件夹(generated_anchors)会生成名字为anchors9.txt的文件,里面存放9组根据你的数据集生成anchors宽高值

将9组生成的anchors宽高值分别*32后,修改至你训练的.cfg文件里

相关文章: