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

Classification Feature Specification #2

Closed
curiosity654 opened this issue Feb 23, 2021 · 4 comments
Closed

Classification Feature Specification #2

curiosity654 opened this issue Feb 23, 2021 · 4 comments

Comments

@curiosity654
Copy link

Hi there, thanks for the amazing work. I have a question about the classification script. save_feat in network.py is in test mode, but in forward_test, the model only returns acc and an empty feature dict, could you please provide the components of the feature you used in classification?
image
image

@curiosity654
Copy link
Author

Specifically, I just want to ensure if the classification is done with the concatenation of per capsule descriptor and regressed pose.

@wsunid
Copy link
Contributor

wsunid commented Feb 23, 2021

Hi, thanks for your interest in our work! Yes, we generate the final feature fed to the classifier by flattening multiple capsules (with the concatenation of pose and descriptor).

You could customize the returned feature dict. I removed the feature dict for brevity. Actually, you could also have a look at our classification pipeline. It shows how we customized the returned feature dict and used capsules in classifications.

@curiosity654
Copy link
Author

Thank you for the quick response, now I've reproduced the results in the paper😃.

@wsunid
Copy link
Contributor

wsunid commented May 11, 2021 via email

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