项目作者: moserle

项目描述 :
an implementation of Grad-CAM for tensorflowJS
高级语言: TypeScript
项目地址: git://github.com/moserle/TSJS-Grad-CAM.git
创建时间: 2019-09-06T20:56:39Z
项目社区:https://github.com/moserle/TSJS-Grad-CAM

开源协议:MIT License

下载


TSJS-Grad-CAM

An implementation of Grad-CAM for tensorflowJS.
See the paper: https://arxiv.org/pdf/1610.02391v1.pdf
Gradient class activation maps (Grad-CAM) is a technique to visualize the decisions of deep learning networks.

Acknowledgement

This tensorflow-js implementation is based on the following repositories:

GradCAM implementation in keras:

https://github.com/jacobgil/keras-grad-cam

Tensorflow-js implementation of the CAM algorithm:

https://github.com/tensorflow/tfjs-examples/tree/master/visualize-convnet

Run Grad-CAM

The ‘runGradCAM’ file train a model on a pecified dataset and computes the grad-CAM for a given image.