search

OlafenwaMoses / ImageAI

A python library built to empower developers to build applications and systems with self-contained Computer Vision capabilities
https://www.genxr.co/#products
MIT License
8.57k stars 2.19k forks source link

Transfer Learning on Yolo trained on Coco dataset #378

Open Rakeshchandrajoshi opened 4 years ago

Rakeshchandrajoshi commented 4 years ago

Sir, Can I get add one class on that model? Sir, I want to add one class more on that model but without training again on the coco dataset. Can I do with only the pre-trained model file and dataset of a single class to train the model for 81 class?

OlafenwaMoses commented 4 years ago

If you want to add one more class with the pre-existing classes, you will have to include the COCO images and annotation in your new dataset.