├── .gitignore ├── LICENSE ├── README.md ├── config ├── __init__.py └── defaults.py ├── configs ├── camera_view │ ├── camera_101a.yml │ └── view_101a.yml ├── stage1 │ ├── 101a_384.yml │ ├── 101a_384_recrop.yml │ ├── 101a_384_spgan.yml │ ├── densenet169a_384.yml │ ├── resnext101a_384.yml │ ├── s101_384.yml │ ├── se_resnet101a_384.yml │ └── transreid_256.yml └── stage2 │ ├── 101a_384.yml │ ├── 101a_384_recrop.yml │ ├── 101a_384_spgan.yml │ ├── densenet169a_384.yml │ ├── resnext101a_384.yml │ ├── s101_384.yml │ ├── se_resnet101a_384.yml │ └── transreid_256.yml ├── datasets ├── __init__.py ├── aic.py ├── aic_sim.py ├── aic_sim_spgan.py ├── autoaugment.py ├── bases.py ├── make_dataloader.py ├── preprocessing.py ├── sampler.py ├── sampler_ddp.py └── sim_view.py ├── ensemble.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 │ ├── densenet_ibn.py │ ├── resnest.py │ ├── resnet.py │ ├── resnet_ibn_a.py │ ├── resnext_ibn.py │ ├── se_module.py │ ├── se_resnet_ibn_a.py │ ├── transformer_block.py │ └── vit_pytorch.py ├── layers │ └── pooling.py └── make_model.py ├── paper.pdf ├── processor ├── __init__.py ├── processor.py └── uda_processor.py ├── requirements.txt ├── run.sh ├── solver ├── __init__.py ├── lr_scheduler.py └── make_optimizer.py ├── test.py ├── track_cam_rk.npy ├── track_view_rk.npy ├── train.py ├── train_cam.py ├── train_stage2_v1.py ├── train_stage2_v2.py ├── train_view.py └── utils ├── __init__.py ├── faiss_rerank.py ├── faiss_utils.py ├── ficfac_torch.py ├── iotools.py ├── logger.py ├── meter.py ├── metrics.py └── reranking.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/config/defaults.py -------------------------------------------------------------------------------- /configs/camera_view/camera_101a.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/camera_view/camera_101a.yml -------------------------------------------------------------------------------- /configs/camera_view/view_101a.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/camera_view/view_101a.yml -------------------------------------------------------------------------------- /configs/stage1/101a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage1/101a_384.yml -------------------------------------------------------------------------------- /configs/stage1/101a_384_recrop.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage1/101a_384_recrop.yml -------------------------------------------------------------------------------- /configs/stage1/101a_384_spgan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage1/101a_384_spgan.yml -------------------------------------------------------------------------------- /configs/stage1/densenet169a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage1/densenet169a_384.yml -------------------------------------------------------------------------------- /configs/stage1/resnext101a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage1/resnext101a_384.yml -------------------------------------------------------------------------------- /configs/stage1/s101_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage1/s101_384.yml -------------------------------------------------------------------------------- /configs/stage1/se_resnet101a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage1/se_resnet101a_384.yml -------------------------------------------------------------------------------- /configs/stage1/transreid_256.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage1/transreid_256.yml -------------------------------------------------------------------------------- /configs/stage2/101a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage2/101a_384.yml -------------------------------------------------------------------------------- /configs/stage2/101a_384_recrop.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage2/101a_384_recrop.yml -------------------------------------------------------------------------------- /configs/stage2/101a_384_spgan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage2/101a_384_spgan.yml -------------------------------------------------------------------------------- /configs/stage2/densenet169a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage2/densenet169a_384.yml -------------------------------------------------------------------------------- /configs/stage2/resnext101a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage2/resnext101a_384.yml -------------------------------------------------------------------------------- /configs/stage2/s101_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage2/s101_384.yml -------------------------------------------------------------------------------- /configs/stage2/se_resnet101a_384.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage2/se_resnet101a_384.yml -------------------------------------------------------------------------------- /configs/stage2/transreid_256.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/configs/stage2/transreid_256.yml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/aic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/datasets/aic.py -------------------------------------------------------------------------------- /datasets/aic_sim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/datasets/aic_sim.py -------------------------------------------------------------------------------- /datasets/aic_sim_spgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/datasets/aic_sim_spgan.py -------------------------------------------------------------------------------- /datasets/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/datasets/autoaugment.py -------------------------------------------------------------------------------- /datasets/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/datasets/bases.py -------------------------------------------------------------------------------- /datasets/make_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/datasets/make_dataloader.py -------------------------------------------------------------------------------- /datasets/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/datasets/preprocessing.py -------------------------------------------------------------------------------- /datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/datasets/sampler.py -------------------------------------------------------------------------------- /datasets/sampler_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/datasets/sampler_ddp.py -------------------------------------------------------------------------------- /datasets/sim_view.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/datasets/sim_view.py -------------------------------------------------------------------------------- /ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/ensemble.py -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/loss/__init__.py -------------------------------------------------------------------------------- /loss/arcface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/loss/arcface.py -------------------------------------------------------------------------------- /loss/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/loss/center_loss.py -------------------------------------------------------------------------------- /loss/make_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/loss/make_loss.py -------------------------------------------------------------------------------- /loss/metric_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/loss/metric_learning.py -------------------------------------------------------------------------------- /loss/softmax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/loss/softmax_loss.py -------------------------------------------------------------------------------- /loss/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/loss/triplet_loss.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/backbones/densenet_ibn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/model/backbones/densenet_ibn.py -------------------------------------------------------------------------------- /model/backbones/resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/model/backbones/resnest.py -------------------------------------------------------------------------------- /model/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/model/backbones/resnet.py -------------------------------------------------------------------------------- /model/backbones/resnet_ibn_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/model/backbones/resnet_ibn_a.py -------------------------------------------------------------------------------- /model/backbones/resnext_ibn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/model/backbones/resnext_ibn.py -------------------------------------------------------------------------------- /model/backbones/se_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/model/backbones/se_module.py -------------------------------------------------------------------------------- /model/backbones/se_resnet_ibn_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/model/backbones/se_resnet_ibn_a.py -------------------------------------------------------------------------------- /model/backbones/transformer_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/model/backbones/transformer_block.py -------------------------------------------------------------------------------- /model/backbones/vit_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/model/backbones/vit_pytorch.py -------------------------------------------------------------------------------- /model/layers/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/model/layers/pooling.py -------------------------------------------------------------------------------- /model/make_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/model/make_model.py -------------------------------------------------------------------------------- /paper.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/paper.pdf -------------------------------------------------------------------------------- /processor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/processor/__init__.py -------------------------------------------------------------------------------- /processor/processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/processor/processor.py -------------------------------------------------------------------------------- /processor/uda_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/processor/uda_processor.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/run.sh -------------------------------------------------------------------------------- /solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/solver/__init__.py -------------------------------------------------------------------------------- /solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/solver/lr_scheduler.py -------------------------------------------------------------------------------- /solver/make_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/solver/make_optimizer.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/test.py -------------------------------------------------------------------------------- /track_cam_rk.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/track_cam_rk.npy -------------------------------------------------------------------------------- /track_view_rk.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/track_view_rk.npy -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/train.py -------------------------------------------------------------------------------- /train_cam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/train_cam.py -------------------------------------------------------------------------------- /train_stage2_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/train_stage2_v1.py -------------------------------------------------------------------------------- /train_stage2_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/train_stage2_v2.py -------------------------------------------------------------------------------- /train_view.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/train_view.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/faiss_rerank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/utils/faiss_rerank.py -------------------------------------------------------------------------------- /utils/faiss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/utils/faiss_utils.py -------------------------------------------------------------------------------- /utils/ficfac_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/utils/ficfac_torch.py -------------------------------------------------------------------------------- /utils/iotools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/utils/iotools.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/utils/meter.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/reranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/michuanhaohao/AICITY2021_Track2_DMT/HEAD/utils/reranking.py --------------------------------------------------------------------------------