boncey / Flickr4Java

Java API For Flickr. Fork of FlickrJ
BSD 2-Clause "Simplified" License
176 stars 155 forks source link

YOLOv5 Object Detection Model #550

Closed lilitqocharyan closed 2 years ago

lilitqocharyan commented 2 years ago

I have a problem with training yolov5. I can train a model with multiple classes, for example ['cat','dog','rabbit']. I want to add one more class to this model, ['snake'], so my model can detect all four at once. I have tried model ensembling and freezing layers, but these didn't give me the desired result. please can anyone help me figure out how to do this. thanks in advance

object_detection, yolov5, python