gitextract_fsax4gel/ ├── .flake8 ├── .gitignore ├── .isort.cfg ├── .style.yapf ├── LICENSE ├── README.md ├── Torchreid_original_README.rst ├── configs/ │ └── bpbreid/ │ ├── bpbreid_dukemtmc_test.yaml │ ├── bpbreid_dukemtmc_train.yaml │ ├── bpbreid_market1501_test.yaml │ ├── bpbreid_market1501_train.yaml │ ├── bpbreid_occ_duke_test.yaml │ ├── bpbreid_occ_duke_train.yaml │ ├── bpbreid_occ_reid_test.yaml │ ├── bpbreid_occ_reid_train.yaml │ ├── bpbreid_p_dukemtmc_test.yaml │ ├── bpbreid_p_dukemtmc_train.yaml │ ├── pcb_market1501_train.yaml │ └── pcb_occ_duke_train.yaml ├── docs/ │ ├── AWESOME_REID.md │ ├── MODEL_ZOO.md │ ├── Makefile │ ├── conf.py │ ├── datasets.rst │ ├── evaluation.rst │ ├── index.rst │ ├── pkg/ │ │ ├── data.rst │ │ ├── engine.rst │ │ ├── losses.rst │ │ ├── metrics.rst │ │ ├── models.rst │ │ ├── optim.rst │ │ └── utils.rst │ ├── requirements.txt │ └── user_guide.rst ├── linter.sh ├── pyproject.toml ├── requirements.txt ├── requirements_labels.txt ├── setup.py └── torchreid/ ├── __init__.py ├── data/ │ ├── __init__.py │ ├── data_augmentation/ │ │ ├── __init__.py │ │ └── random_occlusion.py │ ├── datamanager.py │ ├── datasets/ │ │ ├── __init__.py │ │ ├── dataset.py │ │ ├── image/ │ │ │ ├── __init__.py │ │ │ ├── cuhk01.py │ │ │ ├── cuhk02.py │ │ │ ├── cuhk03.py │ │ │ ├── dukemtmcreid.py │ │ │ ├── grid.py │ │ │ ├── ilids.py │ │ │ ├── market1501.py │ │ │ ├── msmt17.py │ │ │ ├── occluded_dukemtmc.py │ │ │ ├── occluded_reid.py │ │ │ ├── p_ETHZ.py │ │ │ ├── p_dukemtmc_reid.py │ │ │ ├── partial_ilids.py │ │ │ ├── partial_reid.py │ │ │ ├── prid.py │ │ │ ├── sensereid.py │ │ │ └── viper.py │ │ └── video/ │ │ ├── __init__.py │ │ ├── dukemtmcvidreid.py │ │ ├── ilidsvid.py │ │ ├── mars.py │ │ └── prid2011.py │ ├── masks_transforms/ │ │ ├── __init__.py │ │ ├── coco_keypoints_transforms.py │ │ ├── mask_transform.py │ │ ├── pcb_transforms.py │ │ └── pifpaf_mask_transform.py │ ├── sampler.py │ └── transforms.py ├── engine/ │ ├── __init__.py │ ├── engine.py │ ├── image/ │ │ ├── __init__.py │ │ ├── part_based_engine.py │ │ ├── softmax.py │ │ └── triplet.py │ └── video/ │ ├── __init__.py │ ├── softmax.py │ └── triplet.py ├── hyperparameter/ │ ├── custom_hyperparameter_optimizer.py │ ├── hyperparameter_optimizer.py │ └── optuna_hyperparameter_optimizer.py ├── losses/ │ ├── GiLt_loss.py │ ├── __init__.py │ ├── body_part_attention_loss.py │ ├── cross_entropy_loss.py │ ├── hard_mine_triplet_loss.py │ ├── inter_parts_triplet_loss.py │ ├── part_averaged_triplet_loss.py │ ├── part_individual_triplet_loss.py │ ├── part_max_min_triplet_loss.py │ ├── part_max_triplet_loss.py │ ├── part_min_triplet_loss.py │ └── part_random_max_min_triplet_loss.py ├── metrics/ │ ├── __init__.py │ ├── accuracy.py │ ├── distance.py │ ├── rank.py │ └── rank_cylib/ │ ├── Makefile │ ├── __init__.py │ ├── rank_cy.pyx │ ├── setup.py │ └── test_cython.py ├── models/ │ ├── __init__.py │ ├── bpbreid.py │ ├── compact_bilinear_pooling.py │ ├── densenet.py │ ├── hacnn.py │ ├── hrnet.py │ ├── inceptionresnetv2.py │ ├── inceptionv4.py │ ├── mlfn.py │ ├── mobilenetv2.py │ ├── mudeep.py │ ├── nasnet.py │ ├── osnet.py │ ├── osnet_ain.py │ ├── pcb.py │ ├── pvpm.py │ ├── resnet.py │ ├── resnet_fastreid.py │ ├── resnet_ibn_a.py │ ├── resnet_ibn_b.py │ ├── resnetmid.py │ ├── senet.py │ ├── shufflenet.py │ ├── shufflenetv2.py │ ├── squeezenet.py │ └── xception.py ├── optim/ │ ├── __init__.py │ ├── lr_scheduler.py │ ├── optimizer.py │ └── radam.py ├── scripts/ │ ├── __init__.py │ ├── default_config.py │ ├── get_labels.py │ └── main.py ├── tools/ │ ├── __init__.py │ ├── compute_mean_std.py │ ├── extract_part_based_features.py │ └── feature_extractor.py └── utils/ ├── __init__.py ├── avgmeter.py ├── constants.py ├── distribution.py ├── engine_state.py ├── imagetools.py ├── logging/ │ ├── __init__.py │ ├── deprecated_loggers.py │ └── logger.py ├── model_complexity.py ├── reidtools.py ├── rerank.py ├── tensortools.py ├── tools.py ├── torch_receptive_field/ │ ├── __init__.py │ └── receptive_field.py ├── torchtools.py ├── visualization/ │ ├── __init__.py │ ├── display_batch_triplets.py │ ├── embeddings_projection.py │ ├── feature_map_visualization.py │ └── visualize_query_gallery_rankings.py └── writer.py