alexisbcook / ResNetCAM-keras

Keras implementation of a ResNet-CAM model

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ResNetCAM-keras

Keras implementation of a ResNet-CAM model

Motivation

The original Matlab implementation and paper (for AlexNet, GoogLeNet, and VGG16) can be found here. A Keras implementation of VGG-CAM can be found here.

This implementation is written in Keras and uses ResNet-50, which was not explored in the original paper.

Requirements

  • keras with tensorflow backend (keras version 2.0.0 or later)
  • numpy
  • ast
  • scipy
  • matplotlib
  • opencv3

Usage

git clone https://github.com/alexisbcook/ResNetCAM-keras.git
cd ResNetCAM-keras
python ResNet_CAM.py images/dog.png

Feel free to try out your own image by replacing images/dog.png with a file path to another image! :)

Example plots

Dog Localization

Blog Post

This repository is discussed in the blog post here.