facebookresearch / unbiased-teacher-v2

PyTorch code for CVPR 2022 paper Unbiased Teacher v2 Semi-supervised Object Detection for Anchor-free and Anchor-based Detectors
MIT License
90 stars 19 forks source link

An issue about training my own dataset #11

Open Donghuan777 opened 1 year ago

Donghuan777 commented 1 year ago

Thanks for your work! When I trian fcos-based model using my own dataset with only 1 class 'building', after training, I got 'airplane' class in visualization result. It is strange, i change the NUM_CLASS in config file, anything else i need to change? Thanks for your reply!