Yet-Another-EfficientDet-Pytorch icon indicating copy to clipboard operation
Yet-Another-EfficientDet-Pytorch copied to clipboard

Guidance for using pretrained weights with a different number of classes

Open michaeldwong opened this issue 2 years ago • 1 comments

I want a trained d0 model that only detects cars/people (and no other classes). I'm currently loading in the pretrained weights provided and am doing additional training with only the car/people detections in MS COCO. I've modified the project yml file obj_list to be ['person, 'car']. Is this the correct approach to doing this? Thank you.

michaeldwong avatar Sep 05 '22 20:09 michaeldwong

I am also working on that plz tell if you know

sag3011kumbari avatar Nov 16 '22 09:11 sag3011kumbari