Skip to content

Files

Latest commit

 

History

History

CAM_pytorch

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 
 
 

class activation mapping

  • 环境:

    python版本 pytorch版本
    3.5 0.3.0
  • 作用:分类、定位(不使用真值框进行定位,论文证明 卷积层本身就有定位功能)


数据集

预训练模型

训练

1、在config.py中配置数据集等训练参数

2、执行main.py开始训练

可视化

1、在config.py中配置预训练模型

2、执行main.py可视化class_activation_map

训练过程

图片说明图片说明

效果

  • 网络分类时重点关注的区域(即网络的分类依据)
图片说明图片说明
图片说明图片说明
图片说明图片说明

参考