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

Could the pre-trained weight be available? #2

Open
yankungou opened this issue Nov 22, 2022 · 2 comments
Open

Could the pre-trained weight be available? #2

yankungou opened this issue Nov 22, 2022 · 2 comments

Comments

@yankungou
Copy link

Hi, thank you for your great paper. I'm wondering if the pre-trained weight could be available? Thank you very much!

@sidgairo18
Copy link
Owner

Hi YK711, thank you for your interest in our work.

Unfortunately, we could not make the weights available at the time due to certain license restrictions. And now I have left my university (losing access to the data).

However following the instructions in the README, it should be easy to train the models. :)

Please do reach out incase any part is not clear, or you are stuck with something.

Thanks once again :-)

@yankungou
Copy link
Author

yankungou commented Dec 8, 2022

Hi @sidgairo18, thank you for your quick reply.

May I ask how many clusters you use in the WikiArt dataset? I got bad results, and I found it was not good just after training the classification net (the first stage). So I thought maybe something wrong with the cluster number. I got the optimal cluster number=3 according to the elbow method. The gram matrix is computed on conv5.

Thank you very much!

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