search

AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )
http://pjreddie.com/darknet/
Other
21.67k stars 7.96k forks source link

Help in understanding the way to insert classes #8235

Open gd1925 opened 2 years ago

gd1925 commented 2 years ago

Hi all,

I wish to understand a concept in training. I want to add some objects for detections on top of the objects that the model already detects. (The model for example which is trained on 80 classes and I wish to add maybe 2-3 more classes to it). So is there any way to do it without having to train it from the start? Can transfer learning be useful in this case? Thank you so much.

thuangb commented 2 years ago

If your custom dataset is from natural images and/or the objects in your dataset are similar to objects in COCO dataset then you can use transfer learning, else you have to train from scratch

gd1925 commented 2 years ago

Thank you so much for your reply.

fettahyildizz commented 2 years ago

Did you do it? how are the results? was transfer learning succesfull or you trained it from scratch?

Thanks in advance.