-
Notifications
You must be signed in to change notification settings - Fork 5.5k
Model is being successfully trained on custom dataset, but when testing it outputs COCO's class names instead of custom class names #471
Comments
Hi @netw0rkf10w, |
@ir413 Thanks a lot! As you suggested I replaced the line
Then after executing the command (as shown in the original question), I got the following error:
As you can see my custom dataset has only 6 classes. Maybe this is the reason? |
The error looks unrelated to the number of classes; it's thrown by a Conv op with grouped convolution where there's a mismatch between the number of input channels, number of groups, and number of filter channels. This indicates that you accidentally have used mismatched config and model files. |
@rbgirshick Thank you very much for your answer! To make sure everything is done properly, I have started the training again, and now it works. Sorry for the stupid question. |
Hello,
I have successfully trained a Faster R-CNN model on my custom dataset (my config file is based on
e2e_faster_rcnn_X-101-32x8d-FPN_1x.yaml
with modifications of theDATASETS
lines). However, when usingtools/infer_simple.py
to test the model:it runs without any errors, but the results are very poor, and more strangely, the names of the predicted objects are from COCO (e.g.
car
,air plane
, etc.) and not from my custom dataset.Could you please help?
Thank you very much in advance!
The text was updated successfully, but these errors were encountered: