Yet-Another-EfficientDet-Pytorch
Yet-Another-EfficientDet-Pytorch copied to clipboard
Guidance for using pretrained weights with a different number of classes
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.
I am also working on that plz tell if you know