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

About the pre-trained model. #3

Closed
GengDavid opened this issue Jul 9, 2021 · 4 comments
Closed

About the pre-trained model. #3

GengDavid opened this issue Jul 9, 2021 · 4 comments

Comments

@GengDavid
Copy link

Hi @jd730, thanks for sharing your wonderful work.
In the README, you said the training requires "void-suppression" pre-trained model. However, I find that in the config file, you still use the standard pre-trained model here.
If I want to train your EOPSN, which pre-trained model should I use?
Thx.

@jd730
Copy link
Owner

jd730 commented Jul 9, 2021

Hi @GengDavid,
Thank you for reporting it. 'void-suppression' pre-trained model is the correct one.

jd730 added a commit that referenced this issue Jul 9, 2021
@jd730 jd730 closed this as completed Jul 9, 2021
@GengDavid
Copy link
Author

@jd730 Got it. Thanks for your reply. I have a further question about the 'void-suppression' pre-trained model. Does it mean pre-training the backbone model on ImageNet without the "Unknown" classes (i.e., using 1000-16=984 classes for pre-training)?

@jd730
Copy link
Owner

jd730 commented Jul 9, 2021

No, it is trained on COCO without unknown class and with suppression loss for void proposals:
image
Please refer to the main paper for more details.

@GengDavid
Copy link
Author

I see, 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