Image Classification

Image Classification

This page is for cat/dog image classification. Upload an image and try.

The model is trained on 20,000+ images of cats and dogs with the following structure:

            dogcat_model = Sequential()
            dogcat_model.add(Conv2D(64, (3, 3), activation='relu', input_shape=(224, 224, 3)))
            dogcat_model.add(MaxPooling2D(pool_size=(2, 2)))
            dogcat_model.add(Conv2D(64, (3, 3), activation='relu'))
            dogcat_model.add(MaxPooling2D(pool_size=(2, 2)))
            dogcat_model.add(Conv2D(64, (3, 3), activation='relu'))
            dogcat_model.add(MaxPooling2D(pool_size=(2, 2)))
            dogcat_model.add(Flatten())
            dogcat_model.add(Dense(64, activation='relu'))
            dogcat_model.add(Dense(2, activation='softmax'))