├── .gitignore ├── README.md ├── config ├── PAT.yml ├── __init__.py ├── defaults.py └── vit.yml ├── data ├── __init__.py ├── build_DG_dataloader.py ├── common.py ├── data_utils.py ├── datasets │ ├── AirportALERT.py │ ├── DG_cuhk02.py │ ├── DG_cuhk03_detected.py │ ├── DG_cuhk03_labeled.py │ ├── DG_cuhk_sysu.py │ ├── DG_dukemtmcreid.py │ ├── DG_grid.py │ ├── DG_iLIDS.py │ ├── DG_market1501.py │ ├── DG_prid.py │ ├── DG_viper.py │ ├── __init__.py │ ├── bases.py │ ├── caviara.py │ ├── cuhk03.py │ ├── dukemtmcreid.py │ ├── grid.py │ ├── iLIDS.py │ ├── lpw.py │ ├── market1501.py │ ├── msmt17.py │ ├── pes3d.py │ ├── pku.py │ ├── prai.py │ ├── prid.py │ ├── randperson.py │ ├── sensereid.py │ ├── shinpuhkan.py │ ├── sysu_mm.py │ ├── thermalworld.py │ ├── vehicleid.py │ ├── veri.py │ ├── veri_keypoint.py │ ├── veriwild.py │ └── viper.py ├── samplers │ ├── __init__.py │ ├── data_sampler.py │ └── triplet_sampler.py └── transforms │ ├── __init__.py │ ├── autoaugment.py │ ├── build.py │ ├── functional.py │ └── transforms.py ├── enviroments.sh ├── loss ├── __init__.py ├── arcface.py ├── build_loss.py ├── ce_labelSmooth.py ├── center_loss.py ├── make_loss.py ├── metric_learning.py ├── myloss.py ├── smooth.py ├── softmax_loss.py └── triplet_loss.py ├── model ├── __init__.py ├── backbones │ ├── IBN.py │ ├── __init__.py │ ├── resnet.py │ ├── resnet_ibn.py │ └── vit_pytorch.py └── make_model.py ├── processor ├── __init__.py ├── ori_vit_processor_with_amp.py └── part_attention_vit_processor.py ├── run.sh ├── solver ├── __init__.py ├── cosine_lr.py ├── lr_scheduler.py ├── make_optimizer.py ├── scheduler.py └── scheduler_factory.py ├── test.py ├── train.py ├── utils ├── __init__.py ├── comm.py ├── file_io.py ├── iotools.py ├── logger.py ├── meter.py ├── metrics.py ├── registry.py └── reranking.py └── visualization ├── config_vis ├── __init__.py └── vit_b.py ├── good_samples_market_query.json ├── readme.md ├── test.jpg ├── vit_explain.py └── vit_rollout ├── vit_example.py ├── vit_grad_rollout.py └── vit_rollout.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | tb_log 3 | .vscode 4 | *.yml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/README.md -------------------------------------------------------------------------------- /config/PAT.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/config/PAT.yml -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/config/defaults.py -------------------------------------------------------------------------------- /config/vit.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/config/vit.yml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/build_DG_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/build_DG_dataloader.py -------------------------------------------------------------------------------- /data/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/common.py -------------------------------------------------------------------------------- /data/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/data_utils.py -------------------------------------------------------------------------------- /data/datasets/AirportALERT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/AirportALERT.py -------------------------------------------------------------------------------- /data/datasets/DG_cuhk02.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/DG_cuhk02.py -------------------------------------------------------------------------------- /data/datasets/DG_cuhk03_detected.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/DG_cuhk03_detected.py -------------------------------------------------------------------------------- /data/datasets/DG_cuhk03_labeled.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/DG_cuhk03_labeled.py -------------------------------------------------------------------------------- /data/datasets/DG_cuhk_sysu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/DG_cuhk_sysu.py -------------------------------------------------------------------------------- /data/datasets/DG_dukemtmcreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/DG_dukemtmcreid.py -------------------------------------------------------------------------------- /data/datasets/DG_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/DG_grid.py -------------------------------------------------------------------------------- /data/datasets/DG_iLIDS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/DG_iLIDS.py -------------------------------------------------------------------------------- /data/datasets/DG_market1501.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/DG_market1501.py -------------------------------------------------------------------------------- /data/datasets/DG_prid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/DG_prid.py -------------------------------------------------------------------------------- /data/datasets/DG_viper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/DG_viper.py -------------------------------------------------------------------------------- /data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/__init__.py -------------------------------------------------------------------------------- /data/datasets/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/bases.py -------------------------------------------------------------------------------- /data/datasets/caviara.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/caviara.py -------------------------------------------------------------------------------- /data/datasets/cuhk03.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/cuhk03.py -------------------------------------------------------------------------------- /data/datasets/dukemtmcreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/dukemtmcreid.py -------------------------------------------------------------------------------- /data/datasets/grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/grid.py -------------------------------------------------------------------------------- /data/datasets/iLIDS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/iLIDS.py -------------------------------------------------------------------------------- /data/datasets/lpw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/lpw.py -------------------------------------------------------------------------------- /data/datasets/market1501.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/market1501.py -------------------------------------------------------------------------------- /data/datasets/msmt17.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/msmt17.py -------------------------------------------------------------------------------- /data/datasets/pes3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/pes3d.py -------------------------------------------------------------------------------- /data/datasets/pku.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/pku.py -------------------------------------------------------------------------------- /data/datasets/prai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/prai.py -------------------------------------------------------------------------------- /data/datasets/prid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/prid.py -------------------------------------------------------------------------------- /data/datasets/randperson.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/randperson.py -------------------------------------------------------------------------------- /data/datasets/sensereid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/sensereid.py -------------------------------------------------------------------------------- /data/datasets/shinpuhkan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/shinpuhkan.py -------------------------------------------------------------------------------- /data/datasets/sysu_mm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/sysu_mm.py -------------------------------------------------------------------------------- /data/datasets/thermalworld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/thermalworld.py -------------------------------------------------------------------------------- /data/datasets/vehicleid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/vehicleid.py -------------------------------------------------------------------------------- /data/datasets/veri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/veri.py -------------------------------------------------------------------------------- /data/datasets/veri_keypoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/veri_keypoint.py -------------------------------------------------------------------------------- /data/datasets/veriwild.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/veriwild.py -------------------------------------------------------------------------------- /data/datasets/viper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/datasets/viper.py -------------------------------------------------------------------------------- /data/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/samplers/__init__.py -------------------------------------------------------------------------------- /data/samplers/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/samplers/data_sampler.py -------------------------------------------------------------------------------- /data/samplers/triplet_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/samplers/triplet_sampler.py -------------------------------------------------------------------------------- /data/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/transforms/__init__.py -------------------------------------------------------------------------------- /data/transforms/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/transforms/autoaugment.py -------------------------------------------------------------------------------- /data/transforms/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/transforms/build.py -------------------------------------------------------------------------------- /data/transforms/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/transforms/functional.py -------------------------------------------------------------------------------- /data/transforms/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/data/transforms/transforms.py -------------------------------------------------------------------------------- /enviroments.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/enviroments.sh -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/loss/__init__.py -------------------------------------------------------------------------------- /loss/arcface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/loss/arcface.py -------------------------------------------------------------------------------- /loss/build_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/loss/build_loss.py -------------------------------------------------------------------------------- /loss/ce_labelSmooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/loss/ce_labelSmooth.py -------------------------------------------------------------------------------- /loss/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/loss/center_loss.py -------------------------------------------------------------------------------- /loss/make_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/loss/make_loss.py -------------------------------------------------------------------------------- /loss/metric_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/loss/metric_learning.py -------------------------------------------------------------------------------- /loss/myloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/loss/myloss.py -------------------------------------------------------------------------------- /loss/smooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/loss/smooth.py -------------------------------------------------------------------------------- /loss/softmax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/loss/softmax_loss.py -------------------------------------------------------------------------------- /loss/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/loss/triplet_loss.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/backbones/IBN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/model/backbones/IBN.py -------------------------------------------------------------------------------- /model/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/model/backbones/__init__.py -------------------------------------------------------------------------------- /model/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/model/backbones/resnet.py -------------------------------------------------------------------------------- /model/backbones/resnet_ibn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/model/backbones/resnet_ibn.py -------------------------------------------------------------------------------- /model/backbones/vit_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/model/backbones/vit_pytorch.py -------------------------------------------------------------------------------- /model/make_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/model/make_model.py -------------------------------------------------------------------------------- /processor/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /processor/ori_vit_processor_with_amp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/processor/ori_vit_processor_with_amp.py -------------------------------------------------------------------------------- /processor/part_attention_vit_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/processor/part_attention_vit_processor.py -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/run.sh -------------------------------------------------------------------------------- /solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/solver/__init__.py -------------------------------------------------------------------------------- /solver/cosine_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/solver/cosine_lr.py -------------------------------------------------------------------------------- /solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/solver/lr_scheduler.py -------------------------------------------------------------------------------- /solver/make_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/solver/make_optimizer.py -------------------------------------------------------------------------------- /solver/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/solver/scheduler.py -------------------------------------------------------------------------------- /solver/scheduler_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/solver/scheduler_factory.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/utils/comm.py -------------------------------------------------------------------------------- /utils/file_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/utils/file_io.py -------------------------------------------------------------------------------- /utils/iotools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/utils/iotools.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/utils/meter.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/utils/registry.py -------------------------------------------------------------------------------- /utils/reranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/utils/reranking.py -------------------------------------------------------------------------------- /visualization/config_vis/__init__.py: -------------------------------------------------------------------------------- 1 | from .vit_b import _C as cfg -------------------------------------------------------------------------------- /visualization/config_vis/vit_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/visualization/config_vis/vit_b.py -------------------------------------------------------------------------------- /visualization/good_samples_market_query.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/visualization/good_samples_market_query.json -------------------------------------------------------------------------------- /visualization/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/visualization/readme.md -------------------------------------------------------------------------------- /visualization/test.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/visualization/test.jpg -------------------------------------------------------------------------------- /visualization/vit_explain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/visualization/vit_explain.py -------------------------------------------------------------------------------- /visualization/vit_rollout/vit_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/visualization/vit_rollout/vit_example.py -------------------------------------------------------------------------------- /visualization/vit_rollout/vit_grad_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/visualization/vit_rollout/vit_grad_rollout.py -------------------------------------------------------------------------------- /visualization/vit_rollout/vit_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liyuke65535/Part-Aware-Transformer/HEAD/visualization/vit_rollout/vit_rollout.py --------------------------------------------------------------------------------