简体   繁体   中英

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

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.

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM