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