文件名称:integrated-gradient-pytorch:这是该论文的pytorch实施-深度网络的公理归因
文件大小:8.28MB
文件格式:ZIP
更新时间:2024-06-13 14:34:52
visualization pytorch integrated-gradient Python
综合梯度 这是的pytorch实现。 原始的tensorflow版本可以在找到。 致谢 要求 python-3.5.2 pytorch-0.4.1 的OpenCVPython的 待办事项清单 添加更多功能作为原始代码。 微调结果,使它们接近原始纸张。 指示 强烈建议使用GPU加速计算。 如果使用CPU,我建议选择一些小型网络,例如resnet18 。 您还需要将图像放在examples/ 。 支持的网络列表(当然,您可以自己添加任何网络) 起始 网路18 resnet152 vgg19 运行代码 python main.py --cuda --model-type= ' inception ' --img= ' 01.jpg ' 结果 结果与原始论文略有不同,它可能存在一些错误或需要进行一些调整。 我会不断更新,欢迎任何贡献! 盗梦空间v3 ResNet-18 ResNe
【文件预览】:
integrated-gradient-pytorch-master
----figures()
--------table.png(34KB)
--------resnet152.png(1.61MB)
--------inception.png(1.78MB)
--------vgg19.png(1.64MB)
--------resnet18.png(1.73MB)
----main.py(3KB)
----utils.py(2KB)
----LICENSE(1KB)
----results()
--------inception()
--------resnet18()
--------vgg19()
--------resnet152()
----examples()
--------01.jpg(7KB)
--------05.jpg(7KB)
--------04.jpg(21KB)
--------03.jpg(14KB)
--------02.jpg(11KB)
----.gitignore(1KB)
----visualization.py(3KB)
----README.md(1KB)
----integrated_gradients.py(1KB)