Cat-Dog-CNN-Classifier
Cat-Dog-CNN-Classifier copied to clipboard
Convolutional Neural Network to classify images as either cat or dog, along with using attention heatmaps for localization. Written in python with keras.
Cat-Dog-CNN-Classifier
Convolutional Neural Network to classify images as either cat or dog. Written in python with keras.
Trained with keras using tensorflow backend.
localization.py uses a heatmap to figure out where the classified object is in the image, and draws the bounding box over the image.