├── LICENSE ├── README.md ├── pics ├── sota_pic.png └── sota_table.png └── transreid_pytorch ├── config ├── __init__.py └── defaults.py ├── configs ├── dukemtmc │ ├── vit_base.yml │ └── vit_small.yml ├── market │ ├── debug.yml │ ├── vit_base.yml │ ├── vit_base_ics_384.yml │ ├── vit_small.yml │ ├── vit_small_ics.yml │ └── vit_small_ics_ddp.yml ├── msmt17 │ ├── vit_base.yml │ ├── vit_base_ics_384.yml │ ├── vit_small.yml │ ├── vit_small_ics.yml │ └── vit_small_ics_ddp.yml └── occ_duke │ ├── vit_base.yml │ └── vit_small.yml ├── datasets ├── __init__.py ├── bases.py ├── dukemtmcreid.py ├── make_dataloader.py ├── market1501.py ├── mm.py ├── msmt17.py ├── occ_duke.py ├── preprocessing.py ├── sampler.py ├── sampler_ddp.py └── transforms.py ├── 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 │ ├── resnet_ibn_a.py │ ├── swin_transformer.py │ ├── transformer_layers.py │ └── vit_pytorch.py └── make_model.py ├── processor ├── __init__.py └── processor.py ├── run.sh ├── run_batch.sh ├── run_epochs.sh ├── run_epochs_300.sh ├── run_epochs_msmt.sh ├── setup.py ├── solver ├── __init__.py ├── cosine_lr.py ├── lr_scheduler.py ├── make_optimizer.py ├── scheduler.py └── scheduler_factory.py ├── test.py ├── train.py ├── utils ├── __init__.py ├── faiss_rerank.py ├── faiss_utils.py ├── iotools.py ├── logger.py ├── meter.py ├── metrics.py └── reranking.py ├── vis_eval.sh └── vis_rank.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/README.md -------------------------------------------------------------------------------- /pics/sota_pic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/pics/sota_pic.png -------------------------------------------------------------------------------- /pics/sota_table.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/pics/sota_table.png -------------------------------------------------------------------------------- /transreid_pytorch/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/config/__init__.py -------------------------------------------------------------------------------- /transreid_pytorch/config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/config/defaults.py -------------------------------------------------------------------------------- /transreid_pytorch/configs/dukemtmc/vit_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/configs/dukemtmc/vit_base.yml -------------------------------------------------------------------------------- /transreid_pytorch/configs/dukemtmc/vit_small.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/configs/dukemtmc/vit_small.yml -------------------------------------------------------------------------------- /transreid_pytorch/configs/market/debug.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/configs/market/debug.yml -------------------------------------------------------------------------------- /transreid_pytorch/configs/market/vit_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/configs/market/vit_base.yml -------------------------------------------------------------------------------- /transreid_pytorch/configs/market/vit_base_ics_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/configs/market/vit_base_ics_384.yml -------------------------------------------------------------------------------- /transreid_pytorch/configs/market/vit_small.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/configs/market/vit_small.yml -------------------------------------------------------------------------------- /transreid_pytorch/configs/market/vit_small_ics.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/configs/market/vit_small_ics.yml -------------------------------------------------------------------------------- /transreid_pytorch/configs/market/vit_small_ics_ddp.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/configs/market/vit_small_ics_ddp.yml -------------------------------------------------------------------------------- /transreid_pytorch/configs/msmt17/vit_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/configs/msmt17/vit_base.yml -------------------------------------------------------------------------------- /transreid_pytorch/configs/msmt17/vit_base_ics_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/configs/msmt17/vit_base_ics_384.yml -------------------------------------------------------------------------------- /transreid_pytorch/configs/msmt17/vit_small.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/configs/msmt17/vit_small.yml -------------------------------------------------------------------------------- /transreid_pytorch/configs/msmt17/vit_small_ics.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/configs/msmt17/vit_small_ics.yml -------------------------------------------------------------------------------- /transreid_pytorch/configs/msmt17/vit_small_ics_ddp.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/configs/msmt17/vit_small_ics_ddp.yml -------------------------------------------------------------------------------- /transreid_pytorch/configs/occ_duke/vit_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/configs/occ_duke/vit_base.yml -------------------------------------------------------------------------------- /transreid_pytorch/configs/occ_duke/vit_small.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/configs/occ_duke/vit_small.yml -------------------------------------------------------------------------------- /transreid_pytorch/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/datasets/__init__.py -------------------------------------------------------------------------------- /transreid_pytorch/datasets/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/datasets/bases.py -------------------------------------------------------------------------------- /transreid_pytorch/datasets/dukemtmcreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/datasets/dukemtmcreid.py -------------------------------------------------------------------------------- /transreid_pytorch/datasets/make_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/datasets/make_dataloader.py -------------------------------------------------------------------------------- /transreid_pytorch/datasets/market1501.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/datasets/market1501.py -------------------------------------------------------------------------------- /transreid_pytorch/datasets/mm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/datasets/mm.py -------------------------------------------------------------------------------- /transreid_pytorch/datasets/msmt17.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/datasets/msmt17.py -------------------------------------------------------------------------------- /transreid_pytorch/datasets/occ_duke.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/datasets/occ_duke.py -------------------------------------------------------------------------------- /transreid_pytorch/datasets/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/datasets/preprocessing.py -------------------------------------------------------------------------------- /transreid_pytorch/datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/datasets/sampler.py -------------------------------------------------------------------------------- /transreid_pytorch/datasets/sampler_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/datasets/sampler_ddp.py -------------------------------------------------------------------------------- /transreid_pytorch/datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/datasets/transforms.py -------------------------------------------------------------------------------- /transreid_pytorch/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/loss/__init__.py -------------------------------------------------------------------------------- /transreid_pytorch/loss/arcface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/loss/arcface.py -------------------------------------------------------------------------------- /transreid_pytorch/loss/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/loss/center_loss.py -------------------------------------------------------------------------------- /transreid_pytorch/loss/make_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/loss/make_loss.py -------------------------------------------------------------------------------- /transreid_pytorch/loss/metric_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/loss/metric_learning.py -------------------------------------------------------------------------------- /transreid_pytorch/loss/softmax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/loss/softmax_loss.py -------------------------------------------------------------------------------- /transreid_pytorch/loss/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/loss/triplet_loss.py -------------------------------------------------------------------------------- /transreid_pytorch/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/model/__init__.py -------------------------------------------------------------------------------- /transreid_pytorch/model/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /transreid_pytorch/model/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/model/backbones/resnet.py -------------------------------------------------------------------------------- /transreid_pytorch/model/backbones/resnet_ibn_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/model/backbones/resnet_ibn_a.py -------------------------------------------------------------------------------- /transreid_pytorch/model/backbones/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/model/backbones/swin_transformer.py -------------------------------------------------------------------------------- /transreid_pytorch/model/backbones/transformer_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/model/backbones/transformer_layers.py -------------------------------------------------------------------------------- /transreid_pytorch/model/backbones/vit_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/model/backbones/vit_pytorch.py -------------------------------------------------------------------------------- /transreid_pytorch/model/make_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/model/make_model.py -------------------------------------------------------------------------------- /transreid_pytorch/processor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/processor/__init__.py -------------------------------------------------------------------------------- /transreid_pytorch/processor/processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/processor/processor.py -------------------------------------------------------------------------------- /transreid_pytorch/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/run.sh -------------------------------------------------------------------------------- /transreid_pytorch/run_batch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/run_batch.sh -------------------------------------------------------------------------------- /transreid_pytorch/run_epochs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/run_epochs.sh -------------------------------------------------------------------------------- /transreid_pytorch/run_epochs_300.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/run_epochs_300.sh -------------------------------------------------------------------------------- /transreid_pytorch/run_epochs_msmt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/run_epochs_msmt.sh -------------------------------------------------------------------------------- /transreid_pytorch/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/setup.py -------------------------------------------------------------------------------- /transreid_pytorch/solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/solver/__init__.py -------------------------------------------------------------------------------- /transreid_pytorch/solver/cosine_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/solver/cosine_lr.py -------------------------------------------------------------------------------- /transreid_pytorch/solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/solver/lr_scheduler.py -------------------------------------------------------------------------------- /transreid_pytorch/solver/make_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/solver/make_optimizer.py -------------------------------------------------------------------------------- /transreid_pytorch/solver/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/solver/scheduler.py -------------------------------------------------------------------------------- /transreid_pytorch/solver/scheduler_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/solver/scheduler_factory.py -------------------------------------------------------------------------------- /transreid_pytorch/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/test.py -------------------------------------------------------------------------------- /transreid_pytorch/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/train.py -------------------------------------------------------------------------------- /transreid_pytorch/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /transreid_pytorch/utils/faiss_rerank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/utils/faiss_rerank.py -------------------------------------------------------------------------------- /transreid_pytorch/utils/faiss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/utils/faiss_utils.py -------------------------------------------------------------------------------- /transreid_pytorch/utils/iotools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/utils/iotools.py -------------------------------------------------------------------------------- /transreid_pytorch/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/utils/logger.py -------------------------------------------------------------------------------- /transreid_pytorch/utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/utils/meter.py -------------------------------------------------------------------------------- /transreid_pytorch/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/utils/metrics.py -------------------------------------------------------------------------------- /transreid_pytorch/utils/reranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/utils/reranking.py -------------------------------------------------------------------------------- /transreid_pytorch/vis_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/vis_eval.sh -------------------------------------------------------------------------------- /transreid_pytorch/vis_rank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/PersonViT/HEAD/transreid_pytorch/vis_rank.py --------------------------------------------------------------------------------