yolov3如何在之前训练的基础上继续训练

很多博主的文章里提到想在之前的基础上训练,在训练最后一句加上预训练模型就可以,通常是


./darknet detector train cfg/voc.data cfg/yolov3-voc.cfg scripts/darknet53.conv.74 

最后一个darknet53.conv.74是下载下来的预训练模型。

但是我们训练自己的模型的时候,得到的训练文件是一个.weights文件,一般是yolov3_last.weights,想在这个基础上训练,不能直接把这个文件当做预训练模型,如果直接将这个放到训练语句的最后,训练会报错

要多运行下面这句话,做一个微调,将yolov3_last.weights变为yolov3_last.conv.23

./darknet partial cfg/darknet19_448.cfg yolov3_last.weights yolov3_last.conv.23 23

然后再运行训练:./darknet detector train cfg/voc.data cfg/yolov3-voc.cfg yolov3_last.conv.23,这样就不会报错了


 

发布了59 篇原创文章 · 获赞 46 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/sinat_41852207/article/details/102964648