资源分类
》 资源 》算法》 smoothgrad-pytorch

smoothgrad-pytorch

2020-01-19 | |  38 |   0 |   0

SmoothGrad with PyTorch

PyTorch implementation of SmoothGrad [1]. WIP, not tested on GPU.

Dependencies

  • Python 2.7

  • PyTorch

  • torchvision

  • tqdm

Examples

python main.py --image samples/cat_dog.png [--no-cuda] [--guided]

With the --guided option, you can generate smoothed maps from guided backproped gradients.

图片.png

Model: ResNet-152 pre-trained on ImageNet
Prediction: bull mastiff - 54.3% @1
#samples: 100

图片.png

References

[1] D. Smikov, N. Thorat, B. Kim, F. Viégas, M. Wattenberg. "SmoothGrad: removing noise by adding noise". arXiv, 2017


上一篇：NMN-MultiHopQA

下一篇：deep-head-pose

用户评价
全部评价

热门资源

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • inferno-boilerplate

    This is a very basic boilerplate example for pe...