/Cat-Dog-CNN-Classifier

Convolutional Neural Network to classify images as either cat or dog, along with using attention heatmaps for localization. Written in python with keras.

Primary LanguagePythonMIT LicenseMIT

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.