├── .gitignore ├── README.md ├── misc ├── PAP_rank_list_M_to_M │ ├── dataset_market1501_Market-1501-v15.09.15_query_0051_c2s1_006051_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0058_c3s1_006826_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0096_c1s1_015851_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0157_c5s1_026726_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0422_c5s3_081437_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0455_c5s1_116245_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0501_c1s2_064721_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0514_c3s1_141483_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0568_c1s3_019626_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0602_c6s2_010968_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0617_c2s2_016562_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_1180_c5s3_006593_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_1256_c1s5_047166_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_1270_c4s5_052385_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_1302_c6s4_019777_00.jpg │ └── dataset_market1501_Market-1501-v15.09.15_query_1497_c1s6_013421_00.jpg ├── PCB_rank_list_M_to_M │ ├── dataset_market1501_Market-1501-v15.09.15_query_0051_c2s1_006051_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0058_c3s1_006826_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0096_c1s1_015851_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0157_c5s1_026726_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0422_c5s3_081437_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0455_c5s1_116245_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0501_c1s2_064721_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0514_c3s1_141483_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0568_c1s3_019626_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0602_c6s2_010968_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_0617_c2s2_016562_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_1180_c5s3_006593_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_1256_c1s5_047166_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_1270_c4s5_052385_00.jpg │ ├── dataset_market1501_Market-1501-v15.09.15_query_1302_c6s4_019777_00.jpg │ └── dataset_market1501_Market-1501-v15.09.15_query_1497_c1s6_013421_00.jpg └── model.png ├── package ├── __init__.py ├── config │ ├── __init__.py │ └── default.py ├── data │ ├── __init__.py │ ├── create_dataset.py │ ├── dataloader.py │ ├── dataset.py │ ├── datasets │ │ ├── __init__.py │ │ ├── coco.py │ │ ├── cuhk03_np_detected_jpg.py │ │ ├── cuhk03_np_detected_png.py │ │ ├── duke.py │ │ ├── market1501.py │ │ ├── msmt17.py │ │ ├── partial_ilids.py │ │ └── partial_reid.py │ ├── kpt_to_pap_mask.py │ ├── multitask_dataloader.py │ ├── random_identity_sampler.py │ └── transform.py ├── eval │ ├── __init__.py │ ├── eval_dataloader.py │ ├── eval_feat.py │ ├── extract_feat.py │ ├── metric.py │ ├── np_distance.py │ ├── re_ranking.py │ └── torch_distance.py ├── loss │ ├── __init__.py │ ├── id_loss.py │ ├── loss.py │ ├── ps_loss.py │ └── triplet_loss.py ├── model │ ├── __init__.py │ ├── backbone.py │ ├── base_model.py │ ├── global_pool.py │ ├── model.py │ ├── pa_pool.py │ ├── pcb_pool.py │ ├── ps_head.py │ └── resnet.py ├── optim │ ├── __init__.py │ ├── cft_trainer.py │ ├── eanet_trainer.py │ ├── lr_scheduler.py │ ├── optimizer.py │ ├── reid_trainer.py │ └── trainer.py └── utils │ ├── __init__.py │ ├── arg_parser.py │ ├── cfg.py │ ├── file.py │ ├── image.py │ ├── init_path.py │ ├── log.py │ ├── meter.py │ ├── misc.py │ ├── model.py │ ├── rank_list.py │ └── torch_utils.py ├── paper_configs ├── GlobalPool.txt ├── PAP.txt ├── PAP_6P.txt ├── PAP_ST_PS.txt ├── PAP_ST_PS_SPGAN.txt ├── PAP_ST_PS_SPGAN_CFT.txt ├── PAP_S_PS.txt ├── PAP_S_PS_Triplet_Loss_Market1501.txt ├── PAP_StC_PS.txt └── PCB.txt ├── requirements.txt └── script └── exp ├── infer_dataloader_example.py ├── remove_optim_lr_s_in_ckpt.py ├── test_PAP_S_PS_reranking.sh ├── test_all.sh ├── train.sh ├── train_PAP_S_PS_Triplet_Loss_Market1501.sh ├── train_all.sh └── visualize_rank_list.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/README.md -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0051_c2s1_006051_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0051_c2s1_006051_00.jpg -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0058_c3s1_006826_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0058_c3s1_006826_00.jpg -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0096_c1s1_015851_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0096_c1s1_015851_00.jpg -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0157_c5s1_026726_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0157_c5s1_026726_00.jpg -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0422_c5s3_081437_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0422_c5s3_081437_00.jpg -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0455_c5s1_116245_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0455_c5s1_116245_00.jpg -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0501_c1s2_064721_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0501_c1s2_064721_00.jpg -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0514_c3s1_141483_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0514_c3s1_141483_00.jpg -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0568_c1s3_019626_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0568_c1s3_019626_00.jpg -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0602_c6s2_010968_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0602_c6s2_010968_00.jpg -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0617_c2s2_016562_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0617_c2s2_016562_00.jpg -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1180_c5s3_006593_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1180_c5s3_006593_00.jpg -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1256_c1s5_047166_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1256_c1s5_047166_00.jpg -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1270_c4s5_052385_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1270_c4s5_052385_00.jpg -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1302_c6s4_019777_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1302_c6s4_019777_00.jpg -------------------------------------------------------------------------------- /misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1497_c1s6_013421_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PAP_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1497_c1s6_013421_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0051_c2s1_006051_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0051_c2s1_006051_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0058_c3s1_006826_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0058_c3s1_006826_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0096_c1s1_015851_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0096_c1s1_015851_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0157_c5s1_026726_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0157_c5s1_026726_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0422_c5s3_081437_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0422_c5s3_081437_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0455_c5s1_116245_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0455_c5s1_116245_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0501_c1s2_064721_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0501_c1s2_064721_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0514_c3s1_141483_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0514_c3s1_141483_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0568_c1s3_019626_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0568_c1s3_019626_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0602_c6s2_010968_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0602_c6s2_010968_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0617_c2s2_016562_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_0617_c2s2_016562_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1180_c5s3_006593_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1180_c5s3_006593_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1256_c1s5_047166_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1256_c1s5_047166_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1270_c4s5_052385_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1270_c4s5_052385_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1302_c6s4_019777_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1302_c6s4_019777_00.jpg -------------------------------------------------------------------------------- /misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1497_c1s6_013421_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/PCB_rank_list_M_to_M/dataset_market1501_Market-1501-v15.09.15_query_1497_c1s6_013421_00.jpg -------------------------------------------------------------------------------- /misc/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/misc/model.png -------------------------------------------------------------------------------- /package/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /package/config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /package/config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/config/default.py -------------------------------------------------------------------------------- /package/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /package/data/create_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/data/create_dataset.py -------------------------------------------------------------------------------- /package/data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/data/dataloader.py -------------------------------------------------------------------------------- /package/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/data/dataset.py -------------------------------------------------------------------------------- /package/data/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /package/data/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/data/datasets/coco.py -------------------------------------------------------------------------------- /package/data/datasets/cuhk03_np_detected_jpg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/data/datasets/cuhk03_np_detected_jpg.py -------------------------------------------------------------------------------- /package/data/datasets/cuhk03_np_detected_png.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/data/datasets/cuhk03_np_detected_png.py -------------------------------------------------------------------------------- /package/data/datasets/duke.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/data/datasets/duke.py -------------------------------------------------------------------------------- /package/data/datasets/market1501.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/data/datasets/market1501.py -------------------------------------------------------------------------------- /package/data/datasets/msmt17.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/data/datasets/msmt17.py -------------------------------------------------------------------------------- /package/data/datasets/partial_ilids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/data/datasets/partial_ilids.py -------------------------------------------------------------------------------- /package/data/datasets/partial_reid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/data/datasets/partial_reid.py -------------------------------------------------------------------------------- /package/data/kpt_to_pap_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/data/kpt_to_pap_mask.py -------------------------------------------------------------------------------- /package/data/multitask_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/data/multitask_dataloader.py -------------------------------------------------------------------------------- /package/data/random_identity_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/data/random_identity_sampler.py -------------------------------------------------------------------------------- /package/data/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/data/transform.py -------------------------------------------------------------------------------- /package/eval/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /package/eval/eval_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/eval/eval_dataloader.py -------------------------------------------------------------------------------- /package/eval/eval_feat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/eval/eval_feat.py -------------------------------------------------------------------------------- /package/eval/extract_feat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/eval/extract_feat.py -------------------------------------------------------------------------------- /package/eval/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/eval/metric.py -------------------------------------------------------------------------------- /package/eval/np_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/eval/np_distance.py -------------------------------------------------------------------------------- /package/eval/re_ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/eval/re_ranking.py -------------------------------------------------------------------------------- /package/eval/torch_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/eval/torch_distance.py -------------------------------------------------------------------------------- /package/loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /package/loss/id_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/loss/id_loss.py -------------------------------------------------------------------------------- /package/loss/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/loss/loss.py -------------------------------------------------------------------------------- /package/loss/ps_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/loss/ps_loss.py -------------------------------------------------------------------------------- /package/loss/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/loss/triplet_loss.py -------------------------------------------------------------------------------- /package/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /package/model/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/model/backbone.py -------------------------------------------------------------------------------- /package/model/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/model/base_model.py -------------------------------------------------------------------------------- /package/model/global_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/model/global_pool.py -------------------------------------------------------------------------------- /package/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/model/model.py -------------------------------------------------------------------------------- /package/model/pa_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/model/pa_pool.py -------------------------------------------------------------------------------- /package/model/pcb_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/model/pcb_pool.py -------------------------------------------------------------------------------- /package/model/ps_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/model/ps_head.py -------------------------------------------------------------------------------- /package/model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/model/resnet.py -------------------------------------------------------------------------------- /package/optim/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /package/optim/cft_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/optim/cft_trainer.py -------------------------------------------------------------------------------- /package/optim/eanet_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/optim/eanet_trainer.py -------------------------------------------------------------------------------- /package/optim/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/optim/lr_scheduler.py -------------------------------------------------------------------------------- /package/optim/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/optim/optimizer.py -------------------------------------------------------------------------------- /package/optim/reid_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/optim/reid_trainer.py -------------------------------------------------------------------------------- /package/optim/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/optim/trainer.py -------------------------------------------------------------------------------- /package/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /package/utils/arg_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/utils/arg_parser.py -------------------------------------------------------------------------------- /package/utils/cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/utils/cfg.py -------------------------------------------------------------------------------- /package/utils/file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/utils/file.py -------------------------------------------------------------------------------- /package/utils/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/utils/image.py -------------------------------------------------------------------------------- /package/utils/init_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/utils/init_path.py -------------------------------------------------------------------------------- /package/utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/utils/log.py -------------------------------------------------------------------------------- /package/utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/utils/meter.py -------------------------------------------------------------------------------- /package/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/utils/misc.py -------------------------------------------------------------------------------- /package/utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/utils/model.py -------------------------------------------------------------------------------- /package/utils/rank_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/utils/rank_list.py -------------------------------------------------------------------------------- /package/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/package/utils/torch_utils.py -------------------------------------------------------------------------------- /paper_configs/GlobalPool.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /paper_configs/PAP.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/paper_configs/PAP.txt -------------------------------------------------------------------------------- /paper_configs/PAP_6P.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/paper_configs/PAP_6P.txt -------------------------------------------------------------------------------- /paper_configs/PAP_ST_PS.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/paper_configs/PAP_ST_PS.txt -------------------------------------------------------------------------------- /paper_configs/PAP_ST_PS_SPGAN.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/paper_configs/PAP_ST_PS_SPGAN.txt -------------------------------------------------------------------------------- /paper_configs/PAP_ST_PS_SPGAN_CFT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/paper_configs/PAP_ST_PS_SPGAN_CFT.txt -------------------------------------------------------------------------------- /paper_configs/PAP_S_PS.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/paper_configs/PAP_S_PS.txt -------------------------------------------------------------------------------- /paper_configs/PAP_S_PS_Triplet_Loss_Market1501.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/paper_configs/PAP_S_PS_Triplet_Loss_Market1501.txt -------------------------------------------------------------------------------- /paper_configs/PAP_StC_PS.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/paper_configs/PAP_StC_PS.txt -------------------------------------------------------------------------------- /paper_configs/PCB.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/paper_configs/PCB.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/requirements.txt -------------------------------------------------------------------------------- /script/exp/infer_dataloader_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/script/exp/infer_dataloader_example.py -------------------------------------------------------------------------------- /script/exp/remove_optim_lr_s_in_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/script/exp/remove_optim_lr_s_in_ckpt.py -------------------------------------------------------------------------------- /script/exp/test_PAP_S_PS_reranking.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/script/exp/test_PAP_S_PS_reranking.sh -------------------------------------------------------------------------------- /script/exp/test_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/script/exp/test_all.sh -------------------------------------------------------------------------------- /script/exp/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/script/exp/train.sh -------------------------------------------------------------------------------- /script/exp/train_PAP_S_PS_Triplet_Loss_Market1501.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/script/exp/train_PAP_S_PS_Triplet_Loss_Market1501.sh -------------------------------------------------------------------------------- /script/exp/train_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/script/exp/train_all.sh -------------------------------------------------------------------------------- /script/exp/visualize_rank_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/huanghoujing/EANet/HEAD/script/exp/visualize_rank_list.py --------------------------------------------------------------------------------