在Darknet上使用YOLO预训练重量进行训练

Question

我正试图使用暗网CNN训练我的Yolo对象检测器。我使用Yolov4预先训练过的权重，它可以预测这些COCO类的汽车、红绿灯和停车标志。只是想知道如何添加一个额外的图层，这样我的模型也可以拾取交通标志。下面的训练代码：

!./darknet detector train data/obj.data cfg/yolov4-obj.cfg yolov4.conv.137 -dont_show -map

谷歌Colab我正在使用作为参考：谷歌Colab

Answer 1

除了新的交通标志外，您还必须从一开始就使用所有以前对象、汽车、交通灯和停车标志的图像来训练模型。