Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

COCO number of classes #112

Open
Bala93 opened this issue Jun 12, 2023 · 4 comments
Open

COCO number of classes #112

Bala93 opened this issue Jun 12, 2023 · 4 comments

Comments

@Bala93
Copy link

Bala93 commented Jun 12, 2023

Why does COCO have 91 classes, compared to 80 classes ?

@kazuto1011
Copy link
Owner

COCO has labels for only 80 classes, while the indices range from 0 to 90 (11 classes have no label).
To deal with this, the model simply outputs 91 class logits, and only 80 classes are supervised.
I understand that's kind of redundant, but I just followed the official Caffe implementation so that the pre-trained weights could be loaded as is.

@Bala93
Copy link
Author

Bala93 commented Jun 12, 2023

Thank you for the clarification.

@Bala93 Bala93 closed this as completed Jun 12, 2023
@Bala93 Bala93 reopened this Jun 18, 2023
@Bala93
Copy link
Author

Bala93 commented Jun 18, 2023

I would also like to know one more thing. How do I train for deeplab with 90 classes ? In the following file (https://github.com/kazuto1011/deeplab-pytorch/blob/master/libs/datasets/__init__.py), coco.yaml provision is not provided. I am working with train2014 dataset.

@Italy2006
Copy link

I would also like to know one more thing. How do I train for deeplab with 90 classes ? In the following file (https://github.com/kazuto1011/deeplab-pytorch/blob/master/libs/datasets/__init__.py), coco.yaml provision is not provided. I am working with train2014 dataset.

Sorry but do you solve this problem? I have the same question as you too, thanks for your reply

# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants