Search code examples
tensorflowyolodarknetpre-trained-model

Train with YOLO pretrained weights on Darknet


I'm attempting to train my Yolo object detector using the Darknet CNN. I'm using Yolov4 pre-trained weights which can predict Cars, Traffic Lights, and Stop Signs with these COCO Classes. Just wondering how I can add an extra layer so my model can also pick up Traffic Signs. Code below to train:

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

Google Colab I'm using as reference: Google Colab


Solution

  • You have to train the model from the beginning with images of all previous objects, Cars, Traffic Lights, and Stop Signs, in addition to the new Traffic Signs.