Skip to content
New issue

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

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

Already on GitHub? Sign in to your account

Why the pretrained VCOCO model has 81 object classes? #6

Closed
weiyunfei opened this issue Mar 23, 2021 · 3 comments
Closed

Why the pretrained VCOCO model has 81 object classes? #6

weiyunfei opened this issue Mar 23, 2021 · 3 comments

Comments

@weiyunfei
Copy link

When I tried to evaluate the VCOCO model you provide, I have to set the parameter --num_obj_classes to 81. What is the reason for this setting? And should I set --num_obj_classes 81 during training?

Thanks for your reply!

@weiyunfei
Copy link
Author

I've checked the code, I found that the VCOCO model you released has an 82-way object head. I just wanner verify whether it is a technical error.

@tamtamz
Copy link

tamtamz commented Mar 23, 2021

Thank you for pointing it out. The correct number is 81. The number of object classes in the training command for V-COCO, which was 80, was incorrect, and I fixed it to 81. The one additional class is for the missing object case in the V-COCO dataset, where a person is acting, but a target object is missing.

@weiyunfei
Copy link
Author

Thanks!

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

No branches or pull requests

2 participants