Train with YOLO pretrained weights on Darknet

645 Views Asked by At

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

1

There are 1 best solutions below

1
AbdelAziz AbdelLatef On BEST ANSWER

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.