├── LICENSE ├── README.md ├── attention_data └── img.jpg ├── configs ├── CUB │ ├── conformer_tscam_small_patch16_224.yaml │ ├── deit_tscam_base_patch16_224.yaml │ ├── deit_tscam_small_patch16_224.yaml │ └── deit_tscam_tiny_patch16_224.yaml └── ILSVRC │ ├── conformer_tscam_small_patch16_224.yaml │ ├── deit_tscam_base_patch16_224.yaml │ ├── deit_tscam_small_patch16_224.yaml │ └── deit_tscam_tiny_patch16_224.yaml ├── data └── ImageNet_ILSVRC2012 │ └── ILSVRC2012_list │ ├── train.txt │ ├── val_folder.txt │ └── val_folder_new.txt ├── figures ├── attention_many_layer_cub.png ├── attention_many_layer_ilsvrc.png ├── localization_results.png ├── ts-cam.png └── ts-cam_mechnism.png ├── lib ├── __init__.py ├── __pycache__ │ ├── cams.cpython-36.pyc │ ├── cams_deit.cpython-36.pyc │ ├── cams_deit.cpython-37.pyc │ ├── utils.cpython-36.pyc │ └── utils.cpython-37.pyc ├── cams.py ├── cams_deit.py ├── config │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── __init__.cpython-37.pyc │ │ ├── default.cpython-36.pyc │ │ └── default.cpython-37.pyc │ ├── default.py │ └── utils.py ├── core │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── __init__.cpython-37.pyc │ │ ├── engine.cpython-36.pyc │ │ ├── engine.cpython-37.pyc │ │ ├── functions.cpython-36.pyc │ │ └── functions.cpython-37.pyc │ ├── engine.py │ └── functions.py ├── datasets │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── __init__.cpython-37.pyc │ │ ├── cub.cpython-36.pyc │ │ ├── cub.cpython-37.pyc │ │ ├── imagenet.cpython-36.pyc │ │ └── imagenet.cpython-37.pyc │ ├── cub.py │ └── imagenet.py ├── models │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── __init__.cpython-37.pyc │ │ ├── deit.cpython-36.pyc │ │ ├── deit.cpython-37.pyc │ │ ├── vgg.cpython-36.pyc │ │ ├── vgg.cpython-37.pyc │ │ ├── vision_transformer.cpython-36.pyc │ │ └── vision_transformer.cpython-37.pyc │ ├── conformer.py │ ├── deit.py │ ├── vgg.py │ └── vision_transformer.py └── utils.py ├── tools_cam ├── .ipynb_checkpoints │ ├── visualization_attention_map_cub-checkpoint.ipynb │ └── visualization_attention_map_imaget-checkpoint.ipynb ├── __init__.py ├── __pycache__ │ ├── _init_paths.cpython-36.pyc │ └── _init_paths.cpython-37.pyc ├── _init_paths.py ├── test_cam.py ├── train_cam.py ├── visualization_attention_map_cub.ipynb └── visualization_attention_map_imaget.ipynb ├── train_val_cub.sh ├── train_val_ilsvrc.sh ├── val_cub.sh └── val_ilsvrc.sh /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/README.md -------------------------------------------------------------------------------- /attention_data/img.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/attention_data/img.jpg -------------------------------------------------------------------------------- /configs/CUB/conformer_tscam_small_patch16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/configs/CUB/conformer_tscam_small_patch16_224.yaml -------------------------------------------------------------------------------- /configs/CUB/deit_tscam_base_patch16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/configs/CUB/deit_tscam_base_patch16_224.yaml -------------------------------------------------------------------------------- /configs/CUB/deit_tscam_small_patch16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/configs/CUB/deit_tscam_small_patch16_224.yaml -------------------------------------------------------------------------------- /configs/CUB/deit_tscam_tiny_patch16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/configs/CUB/deit_tscam_tiny_patch16_224.yaml -------------------------------------------------------------------------------- /configs/ILSVRC/conformer_tscam_small_patch16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/configs/ILSVRC/conformer_tscam_small_patch16_224.yaml -------------------------------------------------------------------------------- /configs/ILSVRC/deit_tscam_base_patch16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/configs/ILSVRC/deit_tscam_base_patch16_224.yaml -------------------------------------------------------------------------------- /configs/ILSVRC/deit_tscam_small_patch16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/configs/ILSVRC/deit_tscam_small_patch16_224.yaml -------------------------------------------------------------------------------- /configs/ILSVRC/deit_tscam_tiny_patch16_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/configs/ILSVRC/deit_tscam_tiny_patch16_224.yaml -------------------------------------------------------------------------------- /data/ImageNet_ILSVRC2012/ILSVRC2012_list/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/data/ImageNet_ILSVRC2012/ILSVRC2012_list/train.txt -------------------------------------------------------------------------------- /data/ImageNet_ILSVRC2012/ILSVRC2012_list/val_folder.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/data/ImageNet_ILSVRC2012/ILSVRC2012_list/val_folder.txt -------------------------------------------------------------------------------- /data/ImageNet_ILSVRC2012/ILSVRC2012_list/val_folder_new.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/data/ImageNet_ILSVRC2012/ILSVRC2012_list/val_folder_new.txt -------------------------------------------------------------------------------- /figures/attention_many_layer_cub.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/figures/attention_many_layer_cub.png -------------------------------------------------------------------------------- /figures/attention_many_layer_ilsvrc.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/figures/attention_many_layer_ilsvrc.png -------------------------------------------------------------------------------- /figures/localization_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/figures/localization_results.png -------------------------------------------------------------------------------- /figures/ts-cam.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/figures/ts-cam.png -------------------------------------------------------------------------------- /figures/ts-cam_mechnism.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/figures/ts-cam_mechnism.png -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- 1 | from .models import * 2 | -------------------------------------------------------------------------------- /lib/__pycache__/cams.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/__pycache__/cams.cpython-36.pyc -------------------------------------------------------------------------------- /lib/__pycache__/cams_deit.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/__pycache__/cams_deit.cpython-36.pyc -------------------------------------------------------------------------------- /lib/__pycache__/cams_deit.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/__pycache__/cams_deit.cpython-37.pyc -------------------------------------------------------------------------------- /lib/__pycache__/utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/__pycache__/utils.cpython-36.pyc -------------------------------------------------------------------------------- /lib/__pycache__/utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/__pycache__/utils.cpython-37.pyc -------------------------------------------------------------------------------- /lib/cams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/cams.py -------------------------------------------------------------------------------- /lib/cams_deit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/cams_deit.py -------------------------------------------------------------------------------- /lib/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/config/__init__.py -------------------------------------------------------------------------------- /lib/config/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/config/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/config/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/config/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /lib/config/__pycache__/default.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/config/__pycache__/default.cpython-36.pyc -------------------------------------------------------------------------------- /lib/config/__pycache__/default.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/config/__pycache__/default.cpython-37.pyc -------------------------------------------------------------------------------- /lib/config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/config/default.py -------------------------------------------------------------------------------- /lib/config/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/config/utils.py -------------------------------------------------------------------------------- /lib/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/core/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/core/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/core/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/core/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /lib/core/__pycache__/engine.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/core/__pycache__/engine.cpython-36.pyc -------------------------------------------------------------------------------- /lib/core/__pycache__/engine.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/core/__pycache__/engine.cpython-37.pyc -------------------------------------------------------------------------------- /lib/core/__pycache__/functions.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/core/__pycache__/functions.cpython-36.pyc -------------------------------------------------------------------------------- /lib/core/__pycache__/functions.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/core/__pycache__/functions.cpython-37.pyc -------------------------------------------------------------------------------- /lib/core/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/core/engine.py -------------------------------------------------------------------------------- /lib/core/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/core/functions.py -------------------------------------------------------------------------------- /lib/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/datasets/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/datasets/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/datasets/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/cub.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/datasets/__pycache__/cub.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/cub.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/datasets/__pycache__/cub.cpython-37.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/imagenet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/datasets/__pycache__/imagenet.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/imagenet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/datasets/__pycache__/imagenet.cpython-37.pyc -------------------------------------------------------------------------------- /lib/datasets/cub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/datasets/cub.py -------------------------------------------------------------------------------- /lib/datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/datasets/imagenet.py -------------------------------------------------------------------------------- /lib/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/models/__init__.py -------------------------------------------------------------------------------- /lib/models/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/models/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/models/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/models/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /lib/models/__pycache__/deit.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/models/__pycache__/deit.cpython-36.pyc -------------------------------------------------------------------------------- /lib/models/__pycache__/deit.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/models/__pycache__/deit.cpython-37.pyc -------------------------------------------------------------------------------- /lib/models/__pycache__/vgg.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/models/__pycache__/vgg.cpython-36.pyc -------------------------------------------------------------------------------- /lib/models/__pycache__/vgg.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/models/__pycache__/vgg.cpython-37.pyc -------------------------------------------------------------------------------- /lib/models/__pycache__/vision_transformer.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/models/__pycache__/vision_transformer.cpython-36.pyc -------------------------------------------------------------------------------- /lib/models/__pycache__/vision_transformer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/models/__pycache__/vision_transformer.cpython-37.pyc -------------------------------------------------------------------------------- /lib/models/conformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/models/conformer.py -------------------------------------------------------------------------------- /lib/models/deit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/models/deit.py -------------------------------------------------------------------------------- /lib/models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/models/vgg.py -------------------------------------------------------------------------------- /lib/models/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/models/vision_transformer.py -------------------------------------------------------------------------------- /lib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/lib/utils.py -------------------------------------------------------------------------------- /tools_cam/.ipynb_checkpoints/visualization_attention_map_cub-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/tools_cam/.ipynb_checkpoints/visualization_attention_map_cub-checkpoint.ipynb -------------------------------------------------------------------------------- /tools_cam/.ipynb_checkpoints/visualization_attention_map_imaget-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/tools_cam/.ipynb_checkpoints/visualization_attention_map_imaget-checkpoint.ipynb -------------------------------------------------------------------------------- /tools_cam/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools_cam/__pycache__/_init_paths.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/tools_cam/__pycache__/_init_paths.cpython-36.pyc -------------------------------------------------------------------------------- /tools_cam/__pycache__/_init_paths.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/tools_cam/__pycache__/_init_paths.cpython-37.pyc -------------------------------------------------------------------------------- /tools_cam/_init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/tools_cam/_init_paths.py -------------------------------------------------------------------------------- /tools_cam/test_cam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/tools_cam/test_cam.py -------------------------------------------------------------------------------- /tools_cam/train_cam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/tools_cam/train_cam.py -------------------------------------------------------------------------------- /tools_cam/visualization_attention_map_cub.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/tools_cam/visualization_attention_map_cub.ipynb -------------------------------------------------------------------------------- /tools_cam/visualization_attention_map_imaget.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/tools_cam/visualization_attention_map_imaget.ipynb -------------------------------------------------------------------------------- /train_val_cub.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/train_val_cub.sh -------------------------------------------------------------------------------- /train_val_ilsvrc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/train_val_ilsvrc.sh -------------------------------------------------------------------------------- /val_cub.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/val_cub.sh -------------------------------------------------------------------------------- /val_ilsvrc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vasgaowei/TS-CAM/HEAD/val_ilsvrc.sh --------------------------------------------------------------------------------