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

.h5 file of the custom CNN #17

Open
AshleyNikheelKanhye opened this issue Oct 3, 2023 · 1 comment
Open

.h5 file of the custom CNN #17

AshleyNikheelKanhye opened this issue Oct 3, 2023 · 1 comment

Comments

@AshleyNikheelKanhye
Copy link

Hi @arjunsingh88 , I was hoping if I could get the .h5 file of the custom CNN model if you still have it. You only uploaded the VGG16 transfer learning one.
Thanks anyways for your code.

@arjunsingh88
Copy link
Owner

Hi @AshleyNikheelKanhye In this project, the goal was to showcase the advantages of transfer learning compared to training a custom model from scratch. The results clearly indicated superior performance with transfer learning. I've provided the pre-trained model file (.h5) for the transfer learning approach. However, for the custom model, the .h5 file is not available.

You can still recreate the custom model by running the provided .ipynb file step by step. Following the instructions sequentially should enable you to train a model from scratch. I hope this explanation is helpful.

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