【文件属性】:
文件名称:pytorch-saliency:Pytorch插件生成显着性
文件大小:203KB
文件格式:ZIP
更新时间:2021-05-14 12:16:10
deep-learning pytorch saliency-map Python
显着性地图的Pytorch扩展
Pytorch实现不同的显着性方法。
依存关系
火炬4.0
正在安装
从来源
git clone https://github.com/Ema93sh/pytorch-saliency
cd pytorch-saliency
python setup.py install
用法示例
有关更详细的示例,请查看main.py
from torchvision import models
from saliency import SaliencyMethod, MapType, generate_saliency
# Load your image
input_image = ...
# Load your model
model = models.vgg16(pretrained=True)
# Choose your saliency method
sali
【文件预览】:
pytorch-saliency-master
----setup.py(438B)
----.gitignore(48B)
----images()
--------4.jpg(34KB)
--------3.jpg(10KB)
--------6.jpg(27KB)
--------10.jpg(22KB)
--------1.jpg(24KB)
--------7.jpg(8KB)
--------9.jpg(22KB)
--------2.jpg(17KB)
--------8.jpg(11KB)
--------5.jpg(19KB)
----saliency()
--------saliency.py(284B)
--------excitation_bp()
--------__init__.py(919B)
--------types.py(211B)
--------vanilla()
--------guided()
--------deconv()
----LICENSE(1KB)
----README.md(1KB)
----main.py(2KB)