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 retraining cnn explainer #31

Closed
BobBogart opened this issue Jun 13, 2021 · 2 comments
Closed

about retraining cnn explainer #31

BobBogart opened this issue Jun 13, 2021 · 2 comments

Comments

@BobBogart
Copy link

Hello,it's really happy using cnn explainer.Yet I have a question,how could we use it to train a cnn for things like sword with around 200 kinds?

@xiaohk
Copy link
Member

xiaohk commented Jun 13, 2021

Hello @BobBogart,

I am happy to hear that you like CNN Explainer!

To use CNN Explainer with your own model and dataset, you can check out these related issues: #2, #8, #15. Some functions you want to look into are listed in this comment.

If you want to train Tiny-VGG, the model used in CNN Explainer, on the sword dataset, you can check out issue #14. They have successfully trained Tiny-VGG on their plant dataset and visualize the CNN with CNN Explainer. Since 200 classes are a lot, perhaps you can try to choose 10 most distinct classes as a first step. Later you can modify the CNN Explainer UI code to have more output neurons (you can keep the number of neurons in hidden layers as 10).

I will close the issues for now. Feel free to reach out if you need further help :)

@xiaohk xiaohk closed this as completed Jun 13, 2021
@BobBogart
Copy link
Author

BobBogart commented Jun 13, 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