很多博主的文章里提到想在之前的基础上训练,在训练最后一句加上预训练模型就可以,通常是
./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,这样就不会报错了