arjunsingh88 / image_classification_cats_dogs

Image Classification problem, Cats v/s Dogs Model. Browse to https://imgclassification.herokuapp.com/ for the deployment via Heroku

A Simple Cats/Dogs Classifier

CNN: How it works

• Model 1: simple CNN with 3 Convolution layer and 3 connected layer with regularization and avoiding overfit

• Model 2: Tranfer Learning using VGG16 as base model and using it for this problem

Streamlit app

website

image_classification_cats_dogs's Issues

.h5 file of the custom CNN

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.