├── .gitignore ├── README.md ├── config ├── cfg-base.yaml ├── statistics.yaml ├── transreid-market-train.yaml ├── transreid-msmt17-train.yaml ├── transreid-occduke-train.yaml ├── transreid-occludedreid-test.yaml ├── transreid-partialreid-test.yaml ├── transreidssl-fwreid-train.yaml └── transreidssl-msmt17-train.yaml ├── dataset ├── __init__.py ├── dataset_dir.py ├── dataset_duke.py ├── dataset_fwreid.py ├── dataset_market.py ├── dataset_mars.py ├── dataset_msmt.py ├── dataset_occludedreid.py ├── dataset_partialreid.py ├── method_reading.py ├── sampler.py └── transforms.py ├── inference ├── cluster.py └── search.py ├── loss ├── __init__.py ├── loss_multilabel.py └── triplet_loss.py ├── main.py ├── model ├── __init__.py ├── model_factory.py └── net │ ├── backbones │ ├── osnet.py │ ├── resnet.py │ ├── vit_pytorch_ssltransreid.py │ └── vit_pytorch_transreid.py │ ├── resnet.py │ ├── transreid.py │ └── transreid_ssl.py ├── solver ├── __init__.py ├── cosine_lr.py ├── lr_scheduler.py ├── make_optimizer.py ├── scheduler.py └── scheduler_factory.py ├── tools ├── __init__.py └── format_print.py └── utils ├── center_loss.py ├── dist_compute.py ├── eval_reid.py ├── logger.py ├── rank.py ├── rank_cylib ├── Makefile ├── __init__.py ├── rank_cy.pyx ├── roc_cy.c ├── roc_cy.pyx ├── setup.py └── test_cython.py └── re_ranking.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/README.md -------------------------------------------------------------------------------- /config/cfg-base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/config/cfg-base.yaml -------------------------------------------------------------------------------- /config/statistics.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/config/statistics.yaml -------------------------------------------------------------------------------- /config/transreid-market-train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/config/transreid-market-train.yaml -------------------------------------------------------------------------------- /config/transreid-msmt17-train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/config/transreid-msmt17-train.yaml -------------------------------------------------------------------------------- /config/transreid-occduke-train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/config/transreid-occduke-train.yaml -------------------------------------------------------------------------------- /config/transreid-occludedreid-test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/config/transreid-occludedreid-test.yaml -------------------------------------------------------------------------------- /config/transreid-partialreid-test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/config/transreid-partialreid-test.yaml -------------------------------------------------------------------------------- /config/transreidssl-fwreid-train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/config/transreidssl-fwreid-train.yaml -------------------------------------------------------------------------------- /config/transreidssl-msmt17-train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/config/transreidssl-msmt17-train.yaml -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/dataset_dir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/dataset/dataset_dir.py -------------------------------------------------------------------------------- /dataset/dataset_duke.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/dataset/dataset_duke.py -------------------------------------------------------------------------------- /dataset/dataset_fwreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/dataset/dataset_fwreid.py -------------------------------------------------------------------------------- /dataset/dataset_market.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/dataset/dataset_market.py -------------------------------------------------------------------------------- /dataset/dataset_mars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/dataset/dataset_mars.py -------------------------------------------------------------------------------- /dataset/dataset_msmt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/dataset/dataset_msmt.py -------------------------------------------------------------------------------- /dataset/dataset_occludedreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/dataset/dataset_occludedreid.py -------------------------------------------------------------------------------- /dataset/dataset_partialreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/dataset/dataset_partialreid.py -------------------------------------------------------------------------------- /dataset/method_reading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/dataset/method_reading.py -------------------------------------------------------------------------------- /dataset/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/dataset/sampler.py -------------------------------------------------------------------------------- /dataset/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/dataset/transforms.py -------------------------------------------------------------------------------- /inference/cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/inference/cluster.py -------------------------------------------------------------------------------- /inference/search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/inference/search.py -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/loss/__init__.py -------------------------------------------------------------------------------- /loss/loss_multilabel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/loss/loss_multilabel.py -------------------------------------------------------------------------------- /loss/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/loss/triplet_loss.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/main.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/model_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/model/model_factory.py -------------------------------------------------------------------------------- /model/net/backbones/osnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/model/net/backbones/osnet.py -------------------------------------------------------------------------------- /model/net/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/model/net/backbones/resnet.py -------------------------------------------------------------------------------- /model/net/backbones/vit_pytorch_ssltransreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/model/net/backbones/vit_pytorch_ssltransreid.py -------------------------------------------------------------------------------- /model/net/backbones/vit_pytorch_transreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/model/net/backbones/vit_pytorch_transreid.py -------------------------------------------------------------------------------- /model/net/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/model/net/resnet.py -------------------------------------------------------------------------------- /model/net/transreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/model/net/transreid.py -------------------------------------------------------------------------------- /model/net/transreid_ssl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/model/net/transreid_ssl.py -------------------------------------------------------------------------------- /solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/solver/__init__.py -------------------------------------------------------------------------------- /solver/cosine_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/solver/cosine_lr.py -------------------------------------------------------------------------------- /solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/solver/lr_scheduler.py -------------------------------------------------------------------------------- /solver/make_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/solver/make_optimizer.py -------------------------------------------------------------------------------- /solver/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/solver/scheduler.py -------------------------------------------------------------------------------- /solver/scheduler_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/solver/scheduler_factory.py -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/tools/__init__.py -------------------------------------------------------------------------------- /tools/format_print.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/tools/format_print.py -------------------------------------------------------------------------------- /utils/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/utils/center_loss.py -------------------------------------------------------------------------------- /utils/dist_compute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/utils/dist_compute.py -------------------------------------------------------------------------------- /utils/eval_reid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/utils/eval_reid.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/rank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/utils/rank.py -------------------------------------------------------------------------------- /utils/rank_cylib/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/utils/rank_cylib/Makefile -------------------------------------------------------------------------------- /utils/rank_cylib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/utils/rank_cylib/__init__.py -------------------------------------------------------------------------------- /utils/rank_cylib/rank_cy.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/utils/rank_cylib/rank_cy.pyx -------------------------------------------------------------------------------- /utils/rank_cylib/roc_cy.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/utils/rank_cylib/roc_cy.c -------------------------------------------------------------------------------- /utils/rank_cylib/roc_cy.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/utils/rank_cylib/roc_cy.pyx -------------------------------------------------------------------------------- /utils/rank_cylib/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/utils/rank_cylib/setup.py -------------------------------------------------------------------------------- /utils/rank_cylib/test_cython.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/utils/rank_cylib/test_cython.py -------------------------------------------------------------------------------- /utils/re_ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZepengW/person_reid_develop/HEAD/utils/re_ranking.py --------------------------------------------------------------------------------