├── .flake8 ├── .gitignore ├── .isort.cfg ├── .style.yapf ├── Interact, Embed, and EnlargE-Boosting Modality-specific Representations for Multi-Modal Person Re-identification.pdf ├── LICENSE ├── README.md ├── README.rst ├── configs └── RGBNT_ieee_part_margin.yaml ├── linter.sh ├── requirements.txt ├── scripts ├── __pycache__ │ ├── default_config.cpython-36.pyc │ ├── default_config.cpython-37.pyc │ └── default_config.cpython-38.pyc ├── default_config.py ├── main.py ├── mainMultiModal.py └── mainMultiModalCudnn.py ├── setup.py ├── tools ├── compute_mean_std.py ├── parse_test_res.py ├── visualize_actmap.py └── visualize_actmap_ori.py └── torchreid ├── __init__.py ├── data ├── __init__.py ├── datamanager.py ├── datasets │ ├── __init__.py │ ├── dataset.py │ ├── image │ │ ├── AllDay.py │ │ ├── RGBNT201.py │ │ ├── UAV.py │ │ ├── __init__.py │ │ ├── cuhk01.py │ │ ├── cuhk02.py │ │ ├── cuhk03.py │ │ ├── cuhksysu.py │ │ ├── dukemtmcreid.py │ │ ├── grid.py │ │ ├── ilids.py │ │ ├── market1501.py │ │ ├── market_to_RGBNT201.py │ │ ├── msmt17.py │ │ ├── prid.py │ │ ├── sensereid.py │ │ ├── university1652.py │ │ └── viper.py │ └── video │ │ ├── __init__.py │ │ ├── dukemtmcvidreid.py │ │ ├── ilidsvid.py │ │ ├── mars.py │ │ └── prid2011.py ├── sampler.py └── transforms.py ├── engine ├── __init__.py ├── engine.py ├── image │ ├── __init__.py │ ├── hcloss.py │ ├── margin.py │ ├── softmax.py │ └── triplet.py └── video │ ├── __init__.py │ ├── softmax.py │ └── triplet.py ├── losses ├── __init__.py ├── cross_entropy_loss.py ├── hard_mine_triplet_loss.py ├── hcloss.py ├── multi_modal_margin_loss_new.py └── time_loss.py ├── metrics ├── accuracy.py ├── distance.py ├── rank.py └── rank_cylib │ ├── Makefile │ ├── rank_cy.c │ ├── rank_cy.cpython-36m-x86_64-linux-gnu.so │ ├── rank_cy.pyx │ ├── setup.py │ └── test_cython.py ├── models ├── __init__.py ├── densenet.py ├── hacnn.py ├── ieee3modalPart.py ├── inceptionresnetv2.py ├── inceptionv4.py ├── layers.py ├── mlfn.py ├── mobilenetv2.py ├── mudeep.py ├── nasnet.py ├── osnet.py ├── osnet_ain.py ├── pcb.py ├── pfnet.py ├── resnet.py ├── resnet_ibn_a.py ├── resnet_ibn_b.py ├── resnetmid.py ├── senet.py ├── shufflenet.py ├── shufflenetv2.py ├── squeezenet.py ├── util.py └── xception.py ├── optim ├── __init__.py ├── lr_scheduler.py ├── optimizer.py └── radam.py └── utils ├── GPU-Re-Ranking ├── README.md ├── extension │ ├── adjacency_matrix │ │ ├── build_adjacency_matrix.cpp │ │ ├── build_adjacency_matrix_kernel.cu │ │ └── setup.py │ ├── make.sh │ └── propagation │ │ ├── gnn_propagate.cpp │ │ ├── gnn_propagate_kernel.cu │ │ └── setup.py ├── gnn_reranking.py ├── main.py └── utils.py ├── __init__.py ├── avgmeter.py ├── feature_extractor.py ├── loggers.py ├── model_complexity.py ├── reidtools.py ├── rerank.py ├── tools.py └── torchtools.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/.gitignore -------------------------------------------------------------------------------- /.isort.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/.isort.cfg -------------------------------------------------------------------------------- /.style.yapf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/.style.yapf -------------------------------------------------------------------------------- /Interact, Embed, and EnlargE-Boosting Modality-specific Representations for Multi-Modal Person Re-identification.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/Interact, Embed, and EnlargE-Boosting Modality-specific Representations for Multi-Modal Person Re-identification.pdf -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/README.md -------------------------------------------------------------------------------- /README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/README.rst -------------------------------------------------------------------------------- /configs/RGBNT_ieee_part_margin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/configs/RGBNT_ieee_part_margin.yaml -------------------------------------------------------------------------------- /linter.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/linter.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/__pycache__/default_config.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/scripts/__pycache__/default_config.cpython-36.pyc -------------------------------------------------------------------------------- /scripts/__pycache__/default_config.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/scripts/__pycache__/default_config.cpython-37.pyc -------------------------------------------------------------------------------- /scripts/__pycache__/default_config.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/scripts/__pycache__/default_config.cpython-38.pyc -------------------------------------------------------------------------------- /scripts/default_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/scripts/default_config.py -------------------------------------------------------------------------------- /scripts/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/scripts/main.py -------------------------------------------------------------------------------- /scripts/mainMultiModal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/scripts/mainMultiModal.py -------------------------------------------------------------------------------- /scripts/mainMultiModalCudnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/scripts/mainMultiModalCudnn.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/setup.py -------------------------------------------------------------------------------- /tools/compute_mean_std.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/tools/compute_mean_std.py -------------------------------------------------------------------------------- /tools/parse_test_res.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/tools/parse_test_res.py -------------------------------------------------------------------------------- /tools/visualize_actmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/tools/visualize_actmap.py -------------------------------------------------------------------------------- /tools/visualize_actmap_ori.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/tools/visualize_actmap_ori.py -------------------------------------------------------------------------------- /torchreid/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/__init__.py -------------------------------------------------------------------------------- /torchreid/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/__init__.py -------------------------------------------------------------------------------- /torchreid/data/datamanager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datamanager.py -------------------------------------------------------------------------------- /torchreid/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/__init__.py -------------------------------------------------------------------------------- /torchreid/data/datasets/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/dataset.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/AllDay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/AllDay.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/RGBNT201.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/RGBNT201.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/UAV.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/UAV.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/__init__.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/cuhk01.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/cuhk01.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/cuhk02.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/cuhk02.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/cuhk03.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/cuhk03.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/cuhksysu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/cuhksysu.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/dukemtmcreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/dukemtmcreid.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/grid.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/ilids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/ilids.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/market1501.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/market1501.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/market_to_RGBNT201.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/market_to_RGBNT201.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/msmt17.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/msmt17.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/prid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/prid.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/sensereid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/sensereid.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/university1652.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/university1652.py -------------------------------------------------------------------------------- /torchreid/data/datasets/image/viper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/image/viper.py -------------------------------------------------------------------------------- /torchreid/data/datasets/video/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/video/__init__.py -------------------------------------------------------------------------------- /torchreid/data/datasets/video/dukemtmcvidreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/video/dukemtmcvidreid.py -------------------------------------------------------------------------------- /torchreid/data/datasets/video/ilidsvid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/video/ilidsvid.py -------------------------------------------------------------------------------- /torchreid/data/datasets/video/mars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/video/mars.py -------------------------------------------------------------------------------- /torchreid/data/datasets/video/prid2011.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/datasets/video/prid2011.py -------------------------------------------------------------------------------- /torchreid/data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/sampler.py -------------------------------------------------------------------------------- /torchreid/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/data/transforms.py -------------------------------------------------------------------------------- /torchreid/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/engine/__init__.py -------------------------------------------------------------------------------- /torchreid/engine/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/engine/engine.py -------------------------------------------------------------------------------- /torchreid/engine/image/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/engine/image/__init__.py -------------------------------------------------------------------------------- /torchreid/engine/image/hcloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/engine/image/hcloss.py -------------------------------------------------------------------------------- /torchreid/engine/image/margin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/engine/image/margin.py -------------------------------------------------------------------------------- /torchreid/engine/image/softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/engine/image/softmax.py -------------------------------------------------------------------------------- /torchreid/engine/image/triplet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/engine/image/triplet.py -------------------------------------------------------------------------------- /torchreid/engine/video/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/engine/video/__init__.py -------------------------------------------------------------------------------- /torchreid/engine/video/softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/engine/video/softmax.py -------------------------------------------------------------------------------- /torchreid/engine/video/triplet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/engine/video/triplet.py -------------------------------------------------------------------------------- /torchreid/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/losses/__init__.py -------------------------------------------------------------------------------- /torchreid/losses/cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/losses/cross_entropy_loss.py -------------------------------------------------------------------------------- /torchreid/losses/hard_mine_triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/losses/hard_mine_triplet_loss.py -------------------------------------------------------------------------------- /torchreid/losses/hcloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/losses/hcloss.py -------------------------------------------------------------------------------- /torchreid/losses/multi_modal_margin_loss_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/losses/multi_modal_margin_loss_new.py -------------------------------------------------------------------------------- /torchreid/losses/time_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/losses/time_loss.py -------------------------------------------------------------------------------- /torchreid/metrics/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/metrics/accuracy.py -------------------------------------------------------------------------------- /torchreid/metrics/distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/metrics/distance.py -------------------------------------------------------------------------------- /torchreid/metrics/rank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/metrics/rank.py -------------------------------------------------------------------------------- /torchreid/metrics/rank_cylib/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/metrics/rank_cylib/Makefile -------------------------------------------------------------------------------- /torchreid/metrics/rank_cylib/rank_cy.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/metrics/rank_cylib/rank_cy.c -------------------------------------------------------------------------------- /torchreid/metrics/rank_cylib/rank_cy.cpython-36m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/metrics/rank_cylib/rank_cy.cpython-36m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /torchreid/metrics/rank_cylib/rank_cy.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/metrics/rank_cylib/rank_cy.pyx -------------------------------------------------------------------------------- /torchreid/metrics/rank_cylib/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/metrics/rank_cylib/setup.py -------------------------------------------------------------------------------- /torchreid/metrics/rank_cylib/test_cython.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/metrics/rank_cylib/test_cython.py -------------------------------------------------------------------------------- /torchreid/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/__init__.py -------------------------------------------------------------------------------- /torchreid/models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/densenet.py -------------------------------------------------------------------------------- /torchreid/models/hacnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/hacnn.py -------------------------------------------------------------------------------- /torchreid/models/ieee3modalPart.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/ieee3modalPart.py -------------------------------------------------------------------------------- /torchreid/models/inceptionresnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/inceptionresnetv2.py -------------------------------------------------------------------------------- /torchreid/models/inceptionv4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/inceptionv4.py -------------------------------------------------------------------------------- /torchreid/models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/layers.py -------------------------------------------------------------------------------- /torchreid/models/mlfn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/mlfn.py -------------------------------------------------------------------------------- /torchreid/models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/mobilenetv2.py -------------------------------------------------------------------------------- /torchreid/models/mudeep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/mudeep.py -------------------------------------------------------------------------------- /torchreid/models/nasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/nasnet.py -------------------------------------------------------------------------------- /torchreid/models/osnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/osnet.py -------------------------------------------------------------------------------- /torchreid/models/osnet_ain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/osnet_ain.py -------------------------------------------------------------------------------- /torchreid/models/pcb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/pcb.py -------------------------------------------------------------------------------- /torchreid/models/pfnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/pfnet.py -------------------------------------------------------------------------------- /torchreid/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/resnet.py -------------------------------------------------------------------------------- /torchreid/models/resnet_ibn_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/resnet_ibn_a.py -------------------------------------------------------------------------------- /torchreid/models/resnet_ibn_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/resnet_ibn_b.py -------------------------------------------------------------------------------- /torchreid/models/resnetmid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/resnetmid.py -------------------------------------------------------------------------------- /torchreid/models/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/senet.py -------------------------------------------------------------------------------- /torchreid/models/shufflenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/shufflenet.py -------------------------------------------------------------------------------- /torchreid/models/shufflenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/shufflenetv2.py -------------------------------------------------------------------------------- /torchreid/models/squeezenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/squeezenet.py -------------------------------------------------------------------------------- /torchreid/models/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/util.py -------------------------------------------------------------------------------- /torchreid/models/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/models/xception.py -------------------------------------------------------------------------------- /torchreid/optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/optim/__init__.py -------------------------------------------------------------------------------- /torchreid/optim/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/optim/lr_scheduler.py -------------------------------------------------------------------------------- /torchreid/optim/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/optim/optimizer.py -------------------------------------------------------------------------------- /torchreid/optim/radam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/optim/radam.py -------------------------------------------------------------------------------- /torchreid/utils/GPU-Re-Ranking/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/GPU-Re-Ranking/README.md -------------------------------------------------------------------------------- /torchreid/utils/GPU-Re-Ranking/extension/adjacency_matrix/build_adjacency_matrix.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/GPU-Re-Ranking/extension/adjacency_matrix/build_adjacency_matrix.cpp -------------------------------------------------------------------------------- /torchreid/utils/GPU-Re-Ranking/extension/adjacency_matrix/build_adjacency_matrix_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/GPU-Re-Ranking/extension/adjacency_matrix/build_adjacency_matrix_kernel.cu -------------------------------------------------------------------------------- /torchreid/utils/GPU-Re-Ranking/extension/adjacency_matrix/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/GPU-Re-Ranking/extension/adjacency_matrix/setup.py -------------------------------------------------------------------------------- /torchreid/utils/GPU-Re-Ranking/extension/make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/GPU-Re-Ranking/extension/make.sh -------------------------------------------------------------------------------- /torchreid/utils/GPU-Re-Ranking/extension/propagation/gnn_propagate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/GPU-Re-Ranking/extension/propagation/gnn_propagate.cpp -------------------------------------------------------------------------------- /torchreid/utils/GPU-Re-Ranking/extension/propagation/gnn_propagate_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/GPU-Re-Ranking/extension/propagation/gnn_propagate_kernel.cu -------------------------------------------------------------------------------- /torchreid/utils/GPU-Re-Ranking/extension/propagation/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/GPU-Re-Ranking/extension/propagation/setup.py -------------------------------------------------------------------------------- /torchreid/utils/GPU-Re-Ranking/gnn_reranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/GPU-Re-Ranking/gnn_reranking.py -------------------------------------------------------------------------------- /torchreid/utils/GPU-Re-Ranking/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/GPU-Re-Ranking/main.py -------------------------------------------------------------------------------- /torchreid/utils/GPU-Re-Ranking/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/GPU-Re-Ranking/utils.py -------------------------------------------------------------------------------- /torchreid/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/__init__.py -------------------------------------------------------------------------------- /torchreid/utils/avgmeter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/avgmeter.py -------------------------------------------------------------------------------- /torchreid/utils/feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/feature_extractor.py -------------------------------------------------------------------------------- /torchreid/utils/loggers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/loggers.py -------------------------------------------------------------------------------- /torchreid/utils/model_complexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/model_complexity.py -------------------------------------------------------------------------------- /torchreid/utils/reidtools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/reidtools.py -------------------------------------------------------------------------------- /torchreid/utils/rerank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/rerank.py -------------------------------------------------------------------------------- /torchreid/utils/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/tools.py -------------------------------------------------------------------------------- /torchreid/utils/torchtools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ziwang1121/IEEE/HEAD/torchreid/utils/torchtools.py --------------------------------------------------------------------------------