search

facebookresearch / unbiased-teacher

PyTorch code for ICLR 2021 paper Unbiased Teacher for Semi-Supervised Object Detection
https://arxiv.org/abs/2102.09480
MIT License
415 stars 83 forks source link

About organized own dataset #59

Closed dhhcj closed 3 years ago

dhhcj commented 3 years ago

HI, I really appreciate this work, but I faced a problem when I used my own dataset for training

I refer to some previous issues, modify the COCO data format and category (only 1 class), and make sure the training is fine, but in the eval stage, I faced the problem with prediction still results using COCO categories, so how can I fixed... very looking forward your reply

Does it seem the training stage is fine? I suppose image then image then image finally ... image

and I checked the results, found the category becomes the COCO

dhhcj commented 3 years ago

image