I'm learning ML and I'm exploring object detection and classification. I discovered Yolo few months ago and it's impressively efficient and accurate. There are several pre-trained Yolo models on the web that can handle most of everyday objects.

Now, I'd like to train a Yolo model with custom objects. However... I want to keep already learned labels in the model. I would like to add new labels in the model, not to replace them. There are lot of tutorials that explain very well how to use custom labels from a pre-trained model but I did not find anything about increasing the number of classes inside a pre-trained model.

I barely found a discussion on a GitHub about this and the only solution was to train again all existing labels along with the new ones. I think this is quite tedious. I've read there is a mechanism called transfer learning but I'm not sure how this can be applied to add new labels in a model.

  • $\begingroup$ What version of YOLO are you planning to use? $\endgroup$ – Javier TG Feb 22 at 16:34
  • $\begingroup$ I'm planning to use YOLOv3 from AlexeyAB/darknet. $\endgroup$ – Spiralwise Feb 24 at 9:16

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Browse other questions tagged or ask your own question.