├── .gitignore ├── LICENSE ├── README.md ├── config ├── __init__.py └── defaults.py ├── configs ├── DukeMTMC │ ├── deit_transreid_stride.yml │ ├── vit_base.yml │ ├── vit_jpm.yml │ ├── vit_sie.yml │ ├── vit_transreid.yml │ ├── vit_transreid_384.yml │ ├── vit_transreid_stride.yml │ └── vit_transreid_stride_384.yml ├── MSMT17 │ ├── deit_small.yml │ ├── deit_transreid_stride.yml │ ├── vit_base.yml │ ├── vit_jpm.yml │ ├── vit_sie.yml │ ├── vit_small.yml │ ├── vit_transreid.yml │ ├── vit_transreid_384.yml │ ├── vit_transreid_stride.yml │ └── vit_transreid_stride_384.yml ├── Market │ ├── deit_transreid_stride.yml │ ├── vit_base.yml │ ├── vit_jpm.yml │ ├── vit_sie.yml │ ├── vit_transreid.yml │ ├── vit_transreid_384.yml │ ├── vit_transreid_stride.yml │ └── vit_transreid_stride_384.yml ├── OCC_Duke │ ├── deit_transreid_stride.yml │ ├── vit_base.yml │ ├── vit_jpm.yml │ ├── vit_sie.yml │ ├── vit_transreid.yml │ └── vit_transreid_stride.yml ├── VeRi │ ├── deit_transreid.yml │ ├── deit_transreid_stride.yml │ ├── vit_base.yml │ ├── vit_transreid.yml │ └── vit_transreid_stride.yml ├── VehicleID │ ├── deit_transreid.yml │ ├── deit_transreid_stride.yml │ ├── vit_base.yml │ ├── vit_transreid.yml │ └── vit_transreid_stride.yml └── transformer_base.yml ├── datasets ├── __init__.py ├── bases.py ├── dukemtmcreid.py ├── keypoint_test.txt ├── keypoint_train.txt ├── make_dataloader.py ├── market1501.py ├── msmt17.py ├── occ_duke.py ├── preprocessing.py ├── sampler.py ├── sampler_ddp.py ├── vehicleid.py └── veri.py ├── dist_train.sh ├── figs ├── ablation.png ├── framework.png └── sota.png ├── loss ├── __init__.py ├── arcface.py ├── center_loss.py ├── make_loss.py ├── metric_learning.py ├── softmax_loss.py └── triplet_loss.py ├── model ├── __init__.py ├── backbones │ ├── __init__.py │ ├── resnet.py │ └── vit_pytorch.py └── make_model.py ├── processor ├── __init__.py └── processor.py ├── requirements.txt ├── solver ├── __init__.py ├── cosine_lr.py ├── lr_scheduler.py ├── make_optimizer.py ├── scheduler.py └── scheduler_factory.py ├── test.py ├── train.py └── utils ├── __init__.py ├── iotools.py ├── logger.py ├── meter.py ├── metrics.py └── reranking.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/config/defaults.py -------------------------------------------------------------------------------- /configs/DukeMTMC/deit_transreid_stride.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/DukeMTMC/deit_transreid_stride.yml -------------------------------------------------------------------------------- /configs/DukeMTMC/vit_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/DukeMTMC/vit_base.yml -------------------------------------------------------------------------------- /configs/DukeMTMC/vit_jpm.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/DukeMTMC/vit_jpm.yml -------------------------------------------------------------------------------- /configs/DukeMTMC/vit_sie.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/DukeMTMC/vit_sie.yml -------------------------------------------------------------------------------- /configs/DukeMTMC/vit_transreid.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/DukeMTMC/vit_transreid.yml -------------------------------------------------------------------------------- /configs/DukeMTMC/vit_transreid_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/DukeMTMC/vit_transreid_384.yml -------------------------------------------------------------------------------- /configs/DukeMTMC/vit_transreid_stride.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/DukeMTMC/vit_transreid_stride.yml -------------------------------------------------------------------------------- /configs/DukeMTMC/vit_transreid_stride_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/DukeMTMC/vit_transreid_stride_384.yml -------------------------------------------------------------------------------- /configs/MSMT17/deit_small.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/MSMT17/deit_small.yml -------------------------------------------------------------------------------- /configs/MSMT17/deit_transreid_stride.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/MSMT17/deit_transreid_stride.yml -------------------------------------------------------------------------------- /configs/MSMT17/vit_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/MSMT17/vit_base.yml -------------------------------------------------------------------------------- /configs/MSMT17/vit_jpm.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/MSMT17/vit_jpm.yml -------------------------------------------------------------------------------- /configs/MSMT17/vit_sie.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/MSMT17/vit_sie.yml -------------------------------------------------------------------------------- /configs/MSMT17/vit_small.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/MSMT17/vit_small.yml -------------------------------------------------------------------------------- /configs/MSMT17/vit_transreid.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/MSMT17/vit_transreid.yml -------------------------------------------------------------------------------- /configs/MSMT17/vit_transreid_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/MSMT17/vit_transreid_384.yml -------------------------------------------------------------------------------- /configs/MSMT17/vit_transreid_stride.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/MSMT17/vit_transreid_stride.yml -------------------------------------------------------------------------------- /configs/MSMT17/vit_transreid_stride_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/MSMT17/vit_transreid_stride_384.yml -------------------------------------------------------------------------------- /configs/Market/deit_transreid_stride.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/Market/deit_transreid_stride.yml -------------------------------------------------------------------------------- /configs/Market/vit_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/Market/vit_base.yml -------------------------------------------------------------------------------- /configs/Market/vit_jpm.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/Market/vit_jpm.yml -------------------------------------------------------------------------------- /configs/Market/vit_sie.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/Market/vit_sie.yml -------------------------------------------------------------------------------- /configs/Market/vit_transreid.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/Market/vit_transreid.yml -------------------------------------------------------------------------------- /configs/Market/vit_transreid_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/Market/vit_transreid_384.yml -------------------------------------------------------------------------------- /configs/Market/vit_transreid_stride.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/Market/vit_transreid_stride.yml -------------------------------------------------------------------------------- /configs/Market/vit_transreid_stride_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/Market/vit_transreid_stride_384.yml -------------------------------------------------------------------------------- /configs/OCC_Duke/deit_transreid_stride.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/OCC_Duke/deit_transreid_stride.yml -------------------------------------------------------------------------------- /configs/OCC_Duke/vit_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/OCC_Duke/vit_base.yml -------------------------------------------------------------------------------- /configs/OCC_Duke/vit_jpm.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/OCC_Duke/vit_jpm.yml -------------------------------------------------------------------------------- /configs/OCC_Duke/vit_sie.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/OCC_Duke/vit_sie.yml -------------------------------------------------------------------------------- /configs/OCC_Duke/vit_transreid.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/OCC_Duke/vit_transreid.yml -------------------------------------------------------------------------------- /configs/OCC_Duke/vit_transreid_stride.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/OCC_Duke/vit_transreid_stride.yml -------------------------------------------------------------------------------- /configs/VeRi/deit_transreid.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/VeRi/deit_transreid.yml -------------------------------------------------------------------------------- /configs/VeRi/deit_transreid_stride.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/VeRi/deit_transreid_stride.yml -------------------------------------------------------------------------------- /configs/VeRi/vit_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/VeRi/vit_base.yml -------------------------------------------------------------------------------- /configs/VeRi/vit_transreid.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/VeRi/vit_transreid.yml -------------------------------------------------------------------------------- /configs/VeRi/vit_transreid_stride.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/VeRi/vit_transreid_stride.yml -------------------------------------------------------------------------------- /configs/VehicleID/deit_transreid.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/VehicleID/deit_transreid.yml -------------------------------------------------------------------------------- /configs/VehicleID/deit_transreid_stride.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/VehicleID/deit_transreid_stride.yml -------------------------------------------------------------------------------- /configs/VehicleID/vit_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/VehicleID/vit_base.yml -------------------------------------------------------------------------------- /configs/VehicleID/vit_transreid.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/VehicleID/vit_transreid.yml -------------------------------------------------------------------------------- /configs/VehicleID/vit_transreid_stride.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/VehicleID/vit_transreid_stride.yml -------------------------------------------------------------------------------- /configs/transformer_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/configs/transformer_base.yml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/datasets/bases.py -------------------------------------------------------------------------------- /datasets/dukemtmcreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/datasets/dukemtmcreid.py -------------------------------------------------------------------------------- /datasets/keypoint_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/datasets/keypoint_test.txt -------------------------------------------------------------------------------- /datasets/keypoint_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/datasets/keypoint_train.txt -------------------------------------------------------------------------------- /datasets/make_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/datasets/make_dataloader.py -------------------------------------------------------------------------------- /datasets/market1501.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/datasets/market1501.py -------------------------------------------------------------------------------- /datasets/msmt17.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/datasets/msmt17.py -------------------------------------------------------------------------------- /datasets/occ_duke.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/datasets/occ_duke.py -------------------------------------------------------------------------------- /datasets/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/datasets/preprocessing.py -------------------------------------------------------------------------------- /datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/datasets/sampler.py -------------------------------------------------------------------------------- /datasets/sampler_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/datasets/sampler_ddp.py -------------------------------------------------------------------------------- /datasets/vehicleid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/datasets/vehicleid.py -------------------------------------------------------------------------------- /datasets/veri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/datasets/veri.py -------------------------------------------------------------------------------- /dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/dist_train.sh -------------------------------------------------------------------------------- /figs/ablation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/figs/ablation.png -------------------------------------------------------------------------------- /figs/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/figs/framework.png -------------------------------------------------------------------------------- /figs/sota.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/figs/sota.png -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/loss/__init__.py -------------------------------------------------------------------------------- /loss/arcface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/loss/arcface.py -------------------------------------------------------------------------------- /loss/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/loss/center_loss.py -------------------------------------------------------------------------------- /loss/make_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/loss/make_loss.py -------------------------------------------------------------------------------- /loss/metric_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/loss/metric_learning.py -------------------------------------------------------------------------------- /loss/softmax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/loss/softmax_loss.py -------------------------------------------------------------------------------- /loss/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/loss/triplet_loss.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/model/backbones/resnet.py -------------------------------------------------------------------------------- /model/backbones/vit_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/model/backbones/vit_pytorch.py -------------------------------------------------------------------------------- /model/make_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/model/make_model.py -------------------------------------------------------------------------------- /processor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/processor/__init__.py -------------------------------------------------------------------------------- /processor/processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/processor/processor.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/requirements.txt -------------------------------------------------------------------------------- /solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/solver/__init__.py -------------------------------------------------------------------------------- /solver/cosine_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/solver/cosine_lr.py -------------------------------------------------------------------------------- /solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/solver/lr_scheduler.py -------------------------------------------------------------------------------- /solver/make_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/solver/make_optimizer.py -------------------------------------------------------------------------------- /solver/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/solver/scheduler.py -------------------------------------------------------------------------------- /solver/scheduler_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/solver/scheduler_factory.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/iotools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/utils/iotools.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/utils/meter.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/reranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/damo-cv/TransReID/HEAD/utils/reranking.py --------------------------------------------------------------------------------