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

Pretrained weights for pytorch version #27

Closed
ancorasir opened this issue May 13, 2020 · 2 comments
Closed

Pretrained weights for pytorch version #27

ancorasir opened this issue May 13, 2020 · 2 comments

Comments

@ancorasir
Copy link

Thanks the sharing the code with the community. Would you share the pretrained weights for pytorch as well? I'd like to test trained model on images of novel object for example.

@dougsm
Copy link
Owner

dougsm commented Jun 17, 2020

Hi @ancorasir
Sorry for the super slow reply, I completely missed this.
Pretrained weights for some different conditions are available here: https://github.com/dougsm/ggcnn/releases/download/v1.0/ggcnn2_weights.zip

Note that if you have a specific application in mind, retraining for that scenario might provide better results, especiall aspects like field of view and data augmentation.

@dougsm dougsm closed this as completed Jun 17, 2020
@trungpham2606
Copy link

trungpham2606 commented Jul 2, 2020

@dougsm
Seem there's something wrong that I cant load your saved file.
Can you specify the pytorch version you used for this project ?
Your saved model cant be loaded in other PCs excluding yours.

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

3 participants