在 Darknet 上使用 YOLO 预训练权重进行训练
Train with YOLO pretrained weights on Darknet
我正在尝试使用 Darknet CNN 训练我的 Yolo 对象检测器。我正在使用 Yolov4 预训练权重,它可以使用这些 COCO 类 预测汽车、交通灯和停车标志。只是想知道如何添加一个额外的层,以便我的模型也可以拾取交通标志。下面的代码进行训练:
!./darknet detector train data/obj.data cfg/yolov4-obj.cfg yolov4.conv.137 -dont_show -map
Google Colab 我用作参考:Google Colab
除了新的交通标志外,您还必须从头开始使用所有先前对象、汽车、交通灯和停车标志的图像来训练模型。
我正在尝试使用 Darknet CNN 训练我的 Yolo 对象检测器。我正在使用 Yolov4 预训练权重,它可以使用这些 COCO 类 预测汽车、交通灯和停车标志。只是想知道如何添加一个额外的层,以便我的模型也可以拾取交通标志。下面的代码进行训练:
!./darknet detector train data/obj.data cfg/yolov4-obj.cfg yolov4.conv.137 -dont_show -map
Google Colab 我用作参考:Google Colab
除了新的交通标志外,您还必须从头开始使用所有先前对象、汽车、交通灯和停车标志的图像来训练模型。