-
在PyTorch中实现类激活映射(Class Activation Maps)
资源介绍
PyTorch中的类激活图
如标题为的论文中所述,类激活映射的实现
输入
CAM(猫)
CAM(鹅)
支持的Torchvision型号
在编写CAM时,只能使用以下代码从Torchvision生成以下模型:
网路18
资源34
资源50
resnet101
resnet152
用法
python class_activation_map.py --model_name resnet18 --input_image data/car.jpg在results /目录中生成输出CAM覆盖图像。
要另外保存gif动画,请执行以下操作:
python class_activation_map.py --model_name resnet18 --input_image data/car.jpg --save_gif
例子
| | | |
- 上一篇: Doulci_Activator_2.3v
- 下一篇: iOS苹果推送apns测试工具.zip