Having a problem with multi class dataset

facebookresearch / detr

End-to-End Object Detection with Transformers

Apache License 2.0

13.63k stars 2.46k forks source link

Having a problem with multi class dataset #481

Open Usmani255 opened 2 years ago

Usmani255 commented 2 years ago

@likethesky @Celebio @colesbury @pdollar @haridas Hi sir how are you,Please help me regarding this I run this code on google colab for single class it works well but i failed to get results on multiple class 1.My datasets contains the 15 classes and my last class is is 15 so i change in detr.py to num_classes=16

dataset_class

detrfile

2.in every Epoch class rate is decreasing in training but it is remained to 100 in after every Epoch during test lossproblem1 loss problem2

3.I am also sharing my built checkpoint.pth file with you log file

cc-guowenchang commented 2 years ago

Hi, I have the same problem, did you solve it?

C1078617739 commented 2 years ago

我也有这样的问题，请问您解决了吗？