├── LICENSE ├── README.md ├── fast-reid ├── LICENSE ├── README.md ├── configs │ ├── Base-AGW.yml │ ├── Base-MGN.yml │ ├── Base-Strongerbaseline.yml │ ├── Base-bagtricks.yml │ ├── DukeMTMC │ │ ├── AGW_R101-ibn.yml │ │ ├── AGW_R50-ibn.yml │ │ ├── AGW_R50.yml │ │ ├── AGW_S50.yml │ │ ├── bagtricks_R101-ibn.yml │ │ ├── bagtricks_R50-ibn.yml │ │ ├── bagtricks_R50.yml │ │ ├── bagtricks_R50_ReduceHead.yml │ │ ├── bagtricks_S50.yml │ │ ├── mgn_R50-ibn.yml │ │ ├── sbs_R101-ibn.yml │ │ ├── sbs_R50-ibn.yml │ │ ├── sbs_R50.yml │ │ ├── sbs_R50_ReduceHead.yml │ │ └── sbs_S50.yml │ ├── MOT-Strongerbaseline.yml │ ├── MSMT17 │ │ ├── AGW_R101-ibn.yml │ │ ├── AGW_R50-ibn.yml │ │ ├── AGW_R50.yml │ │ ├── AGW_S50.yml │ │ ├── bagtricks_R101-ibn.yml │ │ ├── bagtricks_R18_ReduceHead.yml │ │ ├── bagtricks_R50-ibn.yml │ │ ├── bagtricks_R50.yml │ │ ├── bagtricks_R50_ReduceHead.yml │ │ ├── bagtricks_S50.yml │ │ ├── mgn_R50-ibn.yml │ │ ├── sbs_R101-ibn.yml │ │ ├── sbs_R50-ibn.yml │ │ ├── sbs_R50.yml │ │ ├── sbs_R50_ReduceHead.yml │ │ └── sbs_S50.yml │ ├── Market1501 │ │ ├── AGW_R101-ibn.yml │ │ ├── AGW_R50-ibn.yml │ │ ├── AGW_R50.yml │ │ ├── AGW_S50.yml │ │ ├── bagtricks_R101-ibn.yml │ │ ├── bagtricks_R50-ibn.yml │ │ ├── bagtricks_R50.yml │ │ ├── bagtricks_R50_ReduceHead.yml │ │ ├── bagtricks_S50.yml │ │ ├── mgn_R50-ibn.yml │ │ ├── sbs_R101-ibn.yml │ │ ├── sbs_R50-ibn.yml │ │ ├── sbs_R50.yml │ │ ├── sbs_R50_ReduceHead.yml │ │ └── sbs_S50.yml │ ├── NAIC │ │ ├── bagtricks_R50-ibn.yml │ │ ├── bagtricks_R50.yml │ │ ├── mgn_R50-ibn.yml │ │ ├── sbs_R101-ibn-hzp.yml │ │ ├── sbs_R101-ibn-lgc.yml │ │ ├── sbs_R101-ibn.yml │ │ ├── sbs_R50-cd.yml │ │ ├── sbs_R50-ibn.yml │ │ ├── sbs_R50.yml │ │ └── sbs_R50_p2019.yml │ ├── Product10k │ │ └── product.yml │ ├── VERIWild │ │ └── bagtricks_R50-ibn.yml │ ├── VeRi │ │ └── sbs_R50-ibn.yml │ └── VehicleID │ │ └── bagtricks_R50-ibn.yml ├── demo │ ├── Eval_MOT.py │ ├── README.md │ ├── __pycache__ │ │ ├── Eval_MOT.cpython-37.pyc │ │ └── predictor.cpython-37.pyc │ ├── demo.py │ ├── plot_roc_with_pickle.py │ ├── predictor.py │ ├── run_bjz.sh │ ├── run_demo.sh │ ├── test.json │ ├── test.png │ └── visualize_result.py ├── fastreid │ ├── __init__.py │ ├── __pycache__ │ │ └── __init__.cpython-37.pyc │ ├── config │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-37.pyc │ │ │ ├── config.cpython-37.pyc │ │ │ └── defaults.cpython-37.pyc │ │ ├── config.py │ │ └── defaults.py │ ├── data │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-37.pyc │ │ │ ├── build.cpython-37.pyc │ │ │ ├── common.cpython-37.pyc │ │ │ └── data_utils.cpython-37.pyc │ │ ├── build.py │ │ ├── common.py │ │ ├── data_utils.py │ │ ├── datasets │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-37.pyc │ │ │ │ ├── bases.cpython-37.pyc │ │ │ │ ├── cuhk03.cpython-37.pyc │ │ │ │ ├── dukemtmcreid.cpython-37.pyc │ │ │ │ ├── market1501.cpython-37.pyc │ │ │ │ ├── mot17.cpython-37.pyc │ │ │ │ ├── mot20.cpython-37.pyc │ │ │ │ ├── msmt17.cpython-37.pyc │ │ │ │ ├── naic.cpython-37.pyc │ │ │ │ ├── product10k.cpython-37.pyc │ │ │ │ ├── reid2019.cpython-37.pyc │ │ │ │ ├── vehicleid.cpython-37.pyc │ │ │ │ ├── veri.cpython-37.pyc │ │ │ │ └── veriwild.cpython-37.pyc │ │ │ ├── bases.py │ │ │ ├── cuhk03.py │ │ │ ├── dukemtmcreid.py │ │ │ ├── market1501.py │ │ │ ├── mot17.py │ │ │ ├── mot20.py │ │ │ ├── msmt17.py │ │ │ ├── naic.py │ │ │ ├── product10k.py │ │ │ ├── reid2019.py │ │ │ ├── vehicleid.py │ │ │ ├── veri.py │ │ │ └── veriwild.py │ │ ├── samplers │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-37.pyc │ │ │ │ ├── data_sampler.cpython-37.pyc │ │ │ │ └── triplet_sampler.cpython-37.pyc │ │ │ ├── data_sampler.py │ │ │ └── triplet_sampler.py │ │ └── transforms │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-37.pyc │ │ │ ├── autoaugment.cpython-37.pyc │ │ │ ├── build.cpython-37.pyc │ │ │ ├── functional.cpython-37.pyc │ │ │ └── transforms.cpython-37.pyc │ │ │ ├── autoaugment.py │ │ │ ├── build.py │ │ │ ├── functional.py │ │ │ └── transforms.py │ ├── engine │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-37.pyc │ │ │ ├── defaults.cpython-37.pyc │ │ │ ├── hooks.cpython-37.pyc │ │ │ ├── launch.cpython-37.pyc │ │ │ └── train_loop.cpython-37.pyc │ │ ├── defaults.py │ │ ├── hooks.py │ │ ├── launch.py │ │ └── train_loop.py │ ├── evaluation │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-37.pyc │ │ │ ├── evaluator.cpython-37.pyc │ │ │ ├── query_expansion.cpython-37.pyc │ │ │ ├── rank.cpython-37.pyc │ │ │ ├── reid_evaluation.cpython-37.pyc │ │ │ ├── rerank.cpython-37.pyc │ │ │ ├── roc.cpython-37.pyc │ │ │ └── testing.cpython-37.pyc │ │ ├── evaluator.py │ │ ├── query_expansion.py │ │ ├── rank.py │ │ ├── rank_cylib │ │ │ ├── Makefile │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ └── __init__.cpython-37.pyc │ │ │ ├── rank_cy.c │ │ │ ├── rank_cy.cpython-37m-x86_64-linux-gnu.so │ │ │ ├── rank_cy.pyx │ │ │ ├── roc_cy.c │ │ │ ├── roc_cy.cpython-37m-x86_64-linux-gnu.so │ │ │ ├── roc_cy.pyx │ │ │ ├── setup.py │ │ │ └── test_cython.py │ │ ├── reid_evaluation.py │ │ ├── rerank.py │ │ ├── roc.py │ │ └── testing.py │ ├── layers │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-37.pyc │ │ │ ├── activation.cpython-37.pyc │ │ │ ├── am_softmax.cpython-37.pyc │ │ │ ├── arc_softmax.cpython-37.pyc │ │ │ ├── batch_drop.cpython-37.pyc │ │ │ ├── batch_norm.cpython-37.pyc │ │ │ ├── cd_conv.cpython-37.pyc │ │ │ ├── circle_softmax.cpython-37.pyc │ │ │ ├── context_block.cpython-37.pyc │ │ │ ├── frn.cpython-37.pyc │ │ │ ├── non_local.cpython-37.pyc │ │ │ ├── pooling.cpython-37.pyc │ │ │ ├── se_layer.cpython-37.pyc │ │ │ └── splat.cpython-37.pyc │ │ ├── activation.py │ │ ├── am_softmax.py │ │ ├── arc_softmax.py │ │ ├── batch_drop.py │ │ ├── batch_norm.py │ │ ├── cd_conv.py │ │ ├── circle_softmax.py │ │ ├── context_block.py │ │ ├── frn.py │ │ ├── non_local.py │ │ ├── pooling.py │ │ ├── se_layer.py │ │ └── splat.py │ ├── modeling │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ └── __init__.cpython-37.pyc │ │ ├── backbones │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-37.pyc │ │ │ │ ├── build.cpython-37.pyc │ │ │ │ ├── osnet.cpython-37.pyc │ │ │ │ ├── resnest.cpython-37.pyc │ │ │ │ ├── resnet.cpython-37.pyc │ │ │ │ └── resnext.cpython-37.pyc │ │ │ ├── build.py │ │ │ ├── osnet.py │ │ │ ├── regnet │ │ │ │ ├── __init__.py │ │ │ │ ├── __pycache__ │ │ │ │ │ ├── __init__.cpython-37.pyc │ │ │ │ │ ├── config.cpython-37.pyc │ │ │ │ │ └── regnet.cpython-37.pyc │ │ │ │ ├── config.py │ │ │ │ ├── regnet.py │ │ │ │ ├── regnetx │ │ │ │ │ ├── RegNetX-1.6GF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetX-12GF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetX-16GF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetX-200MF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetX-3.2GF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetX-32GF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetX-4.0GF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetX-400MF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetX-6.4GF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetX-600MF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetX-8.0GF_dds_8gpu.yaml │ │ │ │ │ └── RegNetX-800MF_dds_8gpu.yaml │ │ │ │ └── regnety │ │ │ │ │ ├── RegNetY-1.6GF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetY-12GF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetY-16GF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetY-200MF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetY-3.2GF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetY-32GF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetY-4.0GF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetY-400MF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetY-6.4GF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetY-600MF_dds_8gpu.yaml │ │ │ │ │ ├── RegNetY-8.0GF_dds_8gpu.yaml │ │ │ │ │ └── RegNetY-800MF_dds_8gpu.yaml │ │ │ ├── resnest.py │ │ │ ├── resnet.py │ │ │ └── resnext.py │ │ ├── heads │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-37.pyc │ │ │ │ ├── bnneck_head.cpython-37.pyc │ │ │ │ ├── build.cpython-37.pyc │ │ │ │ ├── linear_head.cpython-37.pyc │ │ │ │ ├── reduction_head.cpython-37.pyc │ │ │ │ └── sigma_head.cpython-37.pyc │ │ │ ├── bnneck_head.py │ │ │ ├── build.py │ │ │ ├── linear_head.py │ │ │ ├── reduction_head.py │ │ │ └── sigma_head.py │ │ ├── losses │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-37.pyc │ │ │ │ ├── circle_loss.cpython-37.pyc │ │ │ │ ├── cross_entroy_loss.cpython-37.pyc │ │ │ │ ├── focal_loss.cpython-37.pyc │ │ │ │ ├── npair_loss.cpython-37.pyc │ │ │ │ ├── triplet_loss.cpython-37.pyc │ │ │ │ └── utils.cpython-37.pyc │ │ │ ├── circle_loss.py │ │ │ ├── cross_entroy_loss.py │ │ │ ├── focal_loss.py │ │ │ ├── loss_kd.py │ │ │ ├── npair_loss.py │ │ │ ├── smooth_ap.py │ │ │ ├── triplet_loss.py │ │ │ └── utils.py │ │ └── meta_arch │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-37.pyc │ │ │ ├── baseline.cpython-37.pyc │ │ │ ├── build.cpython-37.pyc │ │ │ └── mgn.cpython-37.pyc │ │ │ ├── baseline.py │ │ │ ├── build.py │ │ │ ├── iresnet.py │ │ │ └── mgn.py │ ├── solver │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-37.pyc │ │ │ ├── build.cpython-37.pyc │ │ │ └── lr_scheduler.cpython-37.pyc │ │ ├── build.py │ │ ├── lr_scheduler.py │ │ └── optim │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-37.pyc │ │ │ ├── adam.cpython-37.pyc │ │ │ ├── lamb.cpython-37.pyc │ │ │ ├── sgd.cpython-37.pyc │ │ │ └── swa.cpython-37.pyc │ │ │ ├── adam.py │ │ │ ├── lamb.py │ │ │ ├── sgd.py │ │ │ └── swa.py │ └── utils │ │ ├── __init__.py │ │ ├── __pycache__ │ │ ├── __init__.cpython-37.pyc │ │ ├── checkpoint.cpython-37.pyc │ │ ├── collect_env.cpython-37.pyc │ │ ├── comm.cpython-37.pyc │ │ ├── env.cpython-37.pyc │ │ ├── events.cpython-37.pyc │ │ ├── file_io.cpython-37.pyc │ │ ├── history_buffer.cpython-37.pyc │ │ ├── logger.cpython-37.pyc │ │ ├── precision_bn.cpython-37.pyc │ │ ├── registry.cpython-37.pyc │ │ ├── timer.cpython-37.pyc │ │ └── weight_init.cpython-37.pyc │ │ ├── checkpoint.py │ │ ├── cluster_utils.py │ │ ├── collect_env.py │ │ ├── comm.py │ │ ├── env.py │ │ ├── events.py │ │ ├── file_io.py │ │ ├── history_buffer.py │ │ ├── logger.py │ │ ├── precision_bn.py │ │ ├── registry.py │ │ ├── summary.py │ │ ├── timer.py │ │ ├── visualizer.py │ │ └── weight_init.py ├── projects │ ├── Cross-domain-reid │ │ └── README.md │ ├── CrowdReID │ ├── DistillReID │ │ ├── README.md │ │ ├── configs-bagtricks-ibn-dukemtmcreid │ │ │ ├── KD-bot101ibn-bot18.yml │ │ │ ├── KD-bot101ibn-bot18ibn.yml │ │ │ ├── KD-bot101ibn-bot34.yml │ │ │ ├── KD-bot101ibn-bot34ibn.yml │ │ │ ├── KD-bot101ibn-bot50.yml │ │ │ ├── KD-bot101ibn-bot50ibn.yml │ │ │ ├── KD-bot34ibn-bot18ibn.yml │ │ │ ├── KD-bot50-bot18.yml │ │ │ ├── KD-bot50ibn-bot18ibn.yml │ │ │ ├── KD-bot50ibn-bot34ibn.yml │ │ │ ├── bagtricks_R101-ibn.yml │ │ │ ├── bagtricks_R18-ibn.yml │ │ │ ├── bagtricks_R34-ibn.yml │ │ │ └── bagtricks_R50-ibn.yml │ │ ├── configs-bagtricks-ibn-market1501 │ │ │ ├── KD-bot101ibn-bot18.yml │ │ │ ├── KD-bot101ibn-bot18ibn.yml │ │ │ ├── KD-bot101ibn-bot34.yml │ │ │ ├── KD-bot101ibn-bot34ibn.yml │ │ │ ├── KD-bot101ibn-bot50.yml │ │ │ ├── KD-bot101ibn-bot50ibn.yml │ │ │ ├── KD-bot34ibn-bot18ibn.yml │ │ │ ├── KD-bot50-bot18.yml │ │ │ ├── KD-bot50ibn-bot18ibn.yml │ │ │ ├── KD-bot50ibn-bot34ibn.yml │ │ │ ├── bagtricks_R101-ibn.yml │ │ │ ├── bagtricks_R18-ibn.yml │ │ │ ├── bagtricks_R34-ibn.yml │ │ │ └── bagtricks_R50-ibn.yml │ │ ├── configs-bagtricks-ibn-msmt17 │ │ │ ├── KD-bot101ibn-bot18.yml │ │ │ ├── KD-bot101ibn-bot18ibn.yml │ │ │ ├── KD-bot101ibn-bot34.yml │ │ │ ├── KD-bot101ibn-bot34ibn.yml │ │ │ ├── KD-bot101ibn-bot50.yml │ │ │ ├── KD-bot101ibn-bot50ibn.yml │ │ │ ├── KD-bot34ibn-bot18ibn.yml │ │ │ ├── KD-bot50-bot18.yml │ │ │ ├── KD-bot50ibn-bot18ibn.yml │ │ │ ├── KD-bot50ibn-bot34ibn.yml │ │ │ ├── bagtricks_R101-ibn.yml │ │ │ ├── bagtricks_R18-ibn.yml │ │ │ ├── bagtricks_R34-ibn.yml │ │ │ └── bagtricks_R50-ibn.yml │ │ ├── configs-sbs-ibn-dukemtmcreid │ │ │ ├── KD-bot50-bot18-uncertainty.yml │ │ │ ├── KD-bot50-bot18.yml │ │ │ ├── KD-sbs101ibn-sbs18ibn.yml │ │ │ ├── KD-sbs101ibn-sbs34ibn.yml │ │ │ ├── KD-sbs101ibn-sbs50ibn.yml │ │ │ ├── KD-sbs34ibn-sbs18ibn.yml │ │ │ ├── KD-sbs50ibn-sbs18ibn.yml │ │ │ ├── KD-sbs50ibn-sbs34ibn.yml │ │ │ ├── sbs_R101-ibn.yml │ │ │ ├── sbs_R18-ibn.yml │ │ │ ├── sbs_R34-ibn.yml │ │ │ └── sbs_R50-ibn.yml │ │ ├── configs-sbs-ibn-market1501 │ │ │ ├── KD-bot50-bot18-uncertainty.yml │ │ │ ├── KD-bot50-bot18.yml │ │ │ ├── KD-sbs101ibn-sbs18ibn.yml │ │ │ ├── KD-sbs101ibn-sbs34ibn.yml │ │ │ ├── KD-sbs101ibn-sbs50ibn.yml │ │ │ ├── KD-sbs34ibn-sbs18ibn.yml │ │ │ ├── KD-sbs50ibn-sbs18ibn.yml │ │ │ ├── KD-sbs50ibn-sbs34ibn.yml │ │ │ ├── sbs_R101-ibn.yml │ │ │ ├── sbs_R18-ibn.yml │ │ │ ├── sbs_R34-ibn.yml │ │ │ └── sbs_R50-ibn.yml │ │ ├── configs-sbs-ibn-msmt17 │ │ │ ├── KD-bot50-bot18-uncertainty.yml │ │ │ ├── KD-bot50-bot18.yml │ │ │ ├── KD-sbs101ibn-sbs18ibn.yml │ │ │ ├── KD-sbs101ibn-sbs34ibn.yml │ │ │ ├── KD-sbs101ibn-sbs50ibn.yml │ │ │ ├── KD-sbs34ibn-sbs18ibn.yml │ │ │ ├── KD-sbs50ibn-sbs18ibn.yml │ │ │ ├── KD-sbs50ibn-sbs34ibn.yml │ │ │ ├── sbs_R101-ibn.yml │ │ │ ├── sbs_R18-ibn.yml │ │ │ ├── sbs_R34-ibn.yml │ │ │ └── sbs_R50-ibn.yml │ │ ├── kdreid │ │ │ ├── __init__.py │ │ │ ├── config.py │ │ │ ├── kd_trainer.py │ │ │ ├── loss_kd.py │ │ │ └── modeling │ │ │ │ ├── __init__.py │ │ │ │ ├── backbones │ │ │ │ ├── __init__.py │ │ │ │ └── shufflenetv2 │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── blocks.py │ │ │ │ │ └── network.py │ │ │ │ └── sigma.py │ │ ├── run_bagtricks.sh │ │ └── train_net.py │ ├── HAA │ │ └── Readme.md │ ├── NAIC │ │ └── dataset_analysis.py │ ├── PartialReID │ │ ├── README.md │ │ ├── configs │ │ │ └── partial_market.yml │ │ ├── partialreid │ │ │ ├── __init__.py │ │ │ ├── config.py │ │ │ ├── dsr_distance.py │ │ │ ├── dsr_evaluation.py │ │ │ ├── dsr_head.py │ │ │ ├── partial_dataset.py │ │ │ └── partialbaseline.py │ │ └── train_net.py │ └── README.md ├── start.sh ├── start_docker.sh ├── tests │ ├── __init__.py │ ├── dataset_test.py │ ├── feature_align.py │ ├── interp_test.py │ ├── lr_scheduler_test.py │ ├── model_test.py │ └── sampler_test.py └── tools │ ├── NAIC_train.txt │ ├── NAIC_train_val.txt │ ├── NAIC_val_gallery.txt │ ├── NAIC_val_query.txt │ ├── crop_body.py │ ├── deploy │ ├── Caffe │ │ ├── ReadMe.md │ │ ├── __init__.py │ │ ├── caffe.proto │ │ ├── caffe_lmdb.py │ │ ├── caffe_net.py │ │ ├── caffe_pb2.py │ │ ├── layer_param.py │ │ └── net.py │ ├── README.md │ ├── caffe_export.py │ ├── caffe_inference.py │ ├── onnx_export.py │ ├── onnx_inference.py │ ├── pytorch_to_caffe.py │ ├── run_export.sh │ ├── test_data │ │ ├── 0022_c6s1_002976_01.jpg │ │ ├── 0027_c2s2_091032_02.jpg │ │ ├── 0032_c6s1_002851_01.jpg │ │ ├── 0048_c1s1_005351_01.jpg │ │ └── 0065_c6s1_009501_02.jpg │ ├── trt_export.py │ └── trt_inference.py │ ├── ensemble.py │ ├── mot17_test.log │ ├── mot17_train.log │ ├── split_train_eval.py │ └── train_net.py ├── images └── framework.png ├── learnable_proposal_classifier ├── gcn_based_purity_network │ └── dsgcn │ │ ├── __init__.py │ │ ├── datasets │ │ ├── __init__.py │ │ ├── build_dataloader.py │ │ ├── cluster_dataset.py │ │ ├── cluster_det_processor.py │ │ └── sampler.py │ │ ├── models │ │ ├── __init__.py │ │ └── dsgcn.py │ │ └── pipeline_main.py ├── proposal_generation │ ├── cluster_nodes.py │ └── core_function.py ├── proto │ ├── __init__.py │ ├── common.proto │ ├── common_pb2.py │ ├── detection_results.proto │ ├── detection_results_pb2.py │ ├── online_tracking_results.proto │ └── online_tracking_results_pb2.py ├── scripts │ ├── GCN_input_data_generation.py │ ├── ReID_feature_extraction.py │ ├── deoverlapping.py │ ├── main.sh │ ├── mot_metric_evaluation.py │ ├── post_processing.py │ ├── propsal_generation.py │ └── single_view_tracker_mot.py └── svonline │ ├── __init__.py │ ├── algorithms │ ├── __init__.py │ └── simple_online_method.py │ ├── configs │ ├── single_view_online_moving.json │ └── single_view_online_static.json │ ├── core │ ├── __init__.py │ ├── affinity.py │ ├── creator.py │ ├── data_types.py │ ├── graph.py │ ├── models.py │ ├── node.py │ └── utils.py │ └── fileio │ ├── __init__.py │ └── load_save_data.py └── models └── dsgcn_model_iter_30.pth /LICENSE: -------------------------------------------------------------------------------- 1 | MIT License 2 | 3 | Copyright (c) 2020 daip13 4 | 5 | Permission is hereby granted, free of charge, to any person obtaining a copy 6 | of this software and associated documentation files (the "Software"), to deal 7 | in the Software without restriction, including without limitation the rights 8 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 9 | copies of the Software, and to permit persons to whom the Software is 10 | furnished to do so, subject to the following conditions: 11 | 12 | The above copyright notice and this permission notice shall be included in all 13 | copies or substantial portions of the Software. 14 | 15 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 16 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 17 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 18 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 19 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 20 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 21 | SOFTWARE. 22 | -------------------------------------------------------------------------------- /fast-reid/configs/Base-AGW.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_NL: True 6 | 7 | HEADS: 8 | POOL_LAYER: "gempool" 9 | 10 | LOSSES: 11 | NAME: ("CrossEntropyLoss", "TripletLoss") 12 | CE: 13 | EPSILON: 0.1 14 | SCALE: 1.0 15 | 16 | TRI: 17 | MARGIN: 0.0 18 | HARD_MINING: False 19 | SCALE: 1.0 20 | -------------------------------------------------------------------------------- /fast-reid/configs/Base-MGN.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | META_ARCHITECTURE: 'MGN' 5 | 6 | FREEZE_LAYERS: ["backbone", "b1", "b2", "b3",] 7 | 8 | BACKBONE: 9 | WITH_NL: False 10 | 11 | HEADS: 12 | IN_FEAT: 256 13 | 14 | LOSSES: 15 | NAME: ("CrossEntropyLoss", "TripletLoss",) 16 | CE: 17 | EPSILON: 0.1 18 | SCALE: 0.125 19 | 20 | TRI: 21 | MARGIN: 0.0 22 | HARD_MINING: True 23 | NORM_FEAT: False 24 | SCALE: 0.20 25 | 26 | -------------------------------------------------------------------------------- /fast-reid/configs/Base-Strongerbaseline.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "Base-bagtricks.yml" 2 | 3 | MODEL: 4 | FREEZE_LAYERS: ["backbone"] 5 | 6 | BACKBONE: 7 | WITH_NL: True 8 | 9 | HEADS: 10 | NECK_FEAT: "after" 11 | POOL_LAYER: "gempool" 12 | CLS_LAYER: "circleSoftmax" 13 | SCALE: 64 14 | MARGIN: 0.35 15 | 16 | LOSSES: 17 | NAME: ("CrossEntropyLoss", "TripletLoss",) 18 | CE: 19 | EPSILON: 0.1 20 | SCALE: 1.0 21 | TRI: 22 | MARGIN: 0.0 23 | HARD_MINING: True 24 | NORM_FEAT: False 25 | SCALE: 1.0 26 | 27 | INPUT: 28 | SIZE_TRAIN: [384, 128] 29 | SIZE_TEST: [384, 128] 30 | DO_AUTOAUG: True 31 | 32 | DATALOADER: 33 | NUM_INSTANCE: 16 34 | 35 | SOLVER: 36 | OPT: "Adam" 37 | MAX_ITER: 60 38 | BASE_LR: 0.00035 39 | BIAS_LR_FACTOR: 1. 40 | WEIGHT_DECAY: 0.0005 41 | WEIGHT_DECAY_BIAS: 0.0005 42 | IMS_PER_BATCH: 64 43 | 44 | SCHED: "WarmupCosineAnnealingLR" 45 | DELAY_ITERS: 30 46 | ETA_MIN_LR: 0.00000077 47 | 48 | WARMUP_FACTOR: 0.01 49 | WARMUP_ITERS: 10 50 | FREEZE_ITERS: 10 51 | 52 | CHECKPOINT_PERIOD: 30 53 | 54 | TEST: 55 | EVAL_PERIOD: 30 56 | IMS_PER_BATCH: 128 57 | 58 | CUDNN_BENCHMARK: True 59 | 60 | -------------------------------------------------------------------------------- /fast-reid/configs/Base-bagtricks.yml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | META_ARCHITECTURE: "Baseline" 3 | 4 | BACKBONE: 5 | NAME: "build_resnet_backbone" 6 | NORM: "BN" 7 | DEPTH: "50x" 8 | LAST_STRIDE: 1 9 | WITH_IBN: False 10 | PRETRAIN: True 11 | 12 | HEADS: 13 | NAME: "BNneckHead" 14 | NORM: "BN" 15 | POOL_LAYER: "avgpool" 16 | NECK_FEAT: "before" 17 | CLS_LAYER: "linear" 18 | 19 | LOSSES: 20 | NAME: ("CrossEntropyLoss", "TripletLoss",) 21 | 22 | CE: 23 | EPSILON: 0.1 24 | SCALE: 1. 25 | 26 | TRI: 27 | MARGIN: 0.3 28 | HARD_MINING: True 29 | NORM_FEAT: False 30 | SCALE: 1. 31 | 32 | INPUT: 33 | SIZE_TRAIN: [256, 128] 34 | SIZE_TEST: [256, 128] 35 | REA: 36 | ENABLED: True 37 | PROB: 0.5 38 | MEAN: [123.675, 116.28, 103.53] 39 | DO_PAD: True 40 | 41 | DATALOADER: 42 | PK_SAMPLER: True 43 | NAIVE_WAY: True 44 | NUM_INSTANCE: 4 45 | NUM_WORKERS: 8 46 | 47 | SOLVER: 48 | OPT: "Adam" 49 | MAX_ITER: 120 50 | BASE_LR: 0.00035 51 | BIAS_LR_FACTOR: 2. 52 | WEIGHT_DECAY: 0.0005 53 | WEIGHT_DECAY_BIAS: 0.0005 54 | IMS_PER_BATCH: 64 55 | 56 | SCHED: "WarmupMultiStepLR" 57 | STEPS: [40, 90] 58 | GAMMA: 0.1 59 | 60 | WARMUP_FACTOR: 0.01 61 | WARMUP_ITERS: 10 62 | 63 | CHECKPOINT_PERIOD: 60 64 | 65 | TEST: 66 | EVAL_PERIOD: 30 67 | IMS_PER_BATCH: 128 68 | 69 | CUDNN_BENCHMARK: True 70 | 71 | -------------------------------------------------------------------------------- /fast-reid/configs/DukeMTMC/AGW_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-AGW.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "101x" 6 | WITH_IBN: True 7 | 8 | DATASETS: 9 | NAMES: ("DukeMTMC",) 10 | TESTS: ("DukeMTMC",) 11 | 12 | OUTPUT_DIR: "logs/dukemtmc/agw_R101-ibn" 13 | -------------------------------------------------------------------------------- /fast-reid/configs/DukeMTMC/AGW_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-AGW.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_IBN: True 6 | 7 | DATASETS: 8 | NAMES: ("DukeMTMC",) 9 | TESTS: ("DukeMTMC",) 10 | 11 | OUTPUT_DIR: "logs/dukemtmc/agw_R50-ibn" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/DukeMTMC/AGW_R50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-AGW.yml" 2 | 3 | DATASETS: 4 | NAMES: ("DukeMTMC",) 5 | TESTS: ("DukeMTMC",) 6 | 7 | OUTPUT_DIR: "logs/dukemtmc/agw_R50" 8 | -------------------------------------------------------------------------------- /fast-reid/configs/DukeMTMC/AGW_S50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-AGW.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | NAME: "build_resnest_backbone" 6 | 7 | DATASETS: 8 | NAMES: ("DukeMTMC",) 9 | TESTS: ("DukeMTMC",) 10 | 11 | OUTPUT_DIR: "logs/dukemtmc/agw_S50" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/DukeMTMC/bagtricks_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "101x" 6 | WITH_IBN: True 7 | 8 | DATASETS: 9 | NAMES: ("DukeMTMC",) 10 | TESTS: ("DukeMTMC",) 11 | 12 | OUTPUT_DIR: "logs/dukemtmc/bagtricks_R101-ibn" 13 | -------------------------------------------------------------------------------- /fast-reid/configs/DukeMTMC/bagtricks_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_IBN: True 6 | 7 | DATASETS: 8 | NAMES: ("DukeMTMC",) 9 | TESTS: ("DukeMTMC",) 10 | 11 | OUTPUT_DIR: "logs/dukemtmc/bagtricks_R50-ibn" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/DukeMTMC/bagtricks_R50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | DATASETS: 4 | NAMES: ("DukeMTMC",) 5 | TESTS: ("DukeMTMC",) 6 | 7 | OUTPUT_DIR: "logs/dukemtmc/bagtricks_R50" 8 | -------------------------------------------------------------------------------- /fast-reid/configs/DukeMTMC/bagtricks_R50_ReduceHead.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | MODEL: 4 | WEIGHTS: "./model/DukeMTMC/duke_bot_R50.pth" 5 | BACKBONE: 6 | PRETRAIN: False 7 | PRETRAIN_PATH: "" 8 | # NORM: "syncBN" 9 | HEADS: 10 | NAME: "ReductionHead" 11 | REDUCTION_DIM: 256 12 | # NORM: "syncBN" 13 | 14 | DATASETS: 15 | NAMES: ("DukeMTMC",) 16 | TESTS: ("DukeMTMC",) 17 | 18 | OUTPUT_DIR: "logs/dukemtmc/bagtricks_R50_ReduceHead_CD" 19 | -------------------------------------------------------------------------------- /fast-reid/configs/DukeMTMC/bagtricks_S50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | NAME: "build_resnest_backbone" 6 | 7 | DATASETS: 8 | NAMES: ("DukeMTMC",) 9 | TESTS: ("DukeMTMC",) 10 | 11 | OUTPUT_DIR: "logs/dukemtmc/bagtricks_S50" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/DukeMTMC/mgn_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-MGN.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_IBN: True 6 | 7 | DATASETS: 8 | NAMES: ("DukeMTMC",) 9 | TESTS: ("DukeMTMC",) 10 | 11 | OUTPUT_DIR: "logs/dukemtmc/mgn_R50-ibn" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/DukeMTMC/sbs_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "101x" 6 | WITH_IBN: True 7 | 8 | DATASETS: 9 | NAMES: ("DukeMTMC",) 10 | TESTS: ("DukeMTMC",) 11 | 12 | OUTPUT_DIR: "logs/dukemtmc/sbs_R101-ibn" 13 | -------------------------------------------------------------------------------- /fast-reid/configs/DukeMTMC/sbs_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_IBN: True 6 | 7 | DATASETS: 8 | NAMES: ("DukeMTMC",) 9 | TESTS: ("DukeMTMC",) 10 | 11 | OUTPUT_DIR: "logs/dukemtmc/sbs_R50-ibn" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/DukeMTMC/sbs_R50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | DATASETS: 4 | NAMES: ("DukeMTMC",) 5 | TESTS: ("DukeMTMC",) 6 | 7 | OUTPUT_DIR: "logs/dukemtmc/sbs_R50" 8 | -------------------------------------------------------------------------------- /fast-reid/configs/DukeMTMC/sbs_R50_ReduceHead.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | WEIGHTS: "./model/DukeMTMC/duke_sbs_R50.pth" 5 | BACKBONE: 6 | PRETRAIN: False 7 | PRETRAIN_PATH: "" 8 | CD: True 9 | # NORM: "syncBN" 10 | HEADS: 11 | NAME: "ReductionHead" 12 | REDUCTION_DIM: 256 13 | # NORM: "syncBN" 14 | 15 | DATASETS: 16 | NAMES: ("DukeMTMC",) 17 | TESTS: ("DukeMTMC",) 18 | 19 | OUTPUT_DIR: "logs/dukemtmc/sbs_R50_ReduceHead_CD8" 20 | -------------------------------------------------------------------------------- /fast-reid/configs/DukeMTMC/sbs_S50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | NAME: "build_resnest_backbone" 6 | 7 | DATASETS: 8 | NAMES: ("DukeMTMC",) 9 | TESTS: ("DukeMTMC",) 10 | 11 | OUTPUT_DIR: "logs/dukemtmc/sbs_S50" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/AGW_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-AGW.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "101x" 6 | WITH_IBN: True 7 | 8 | DATASETS: 9 | NAMES: ("MSMT17",) 10 | TESTS: ("MSMT17",) 11 | 12 | OUTPUT_DIR: "logs/msmt17/agw_R101-ibn" 13 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/AGW_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-AGW.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_IBN: True 6 | 7 | DATASETS: 8 | NAMES: ("MSMT17",) 9 | TESTS: ("MSMT17",) 10 | 11 | OUTPUT_DIR: "logs/msmt17/agw_R50-ibn" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/AGW_R50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-AGW.yml" 2 | 3 | DATASETS: 4 | NAMES: ("MSMT17",) 5 | TESTS: ("MSMT17",) 6 | 7 | OUTPUT_DIR: "logs/msmt17/agw_R50" 8 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/AGW_S50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-AGW.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | NAME: "build_resnest_backbone" 6 | 7 | DATASETS: 8 | NAMES: ("MSMT17",) 9 | TESTS: ("MSMT17",) 10 | 11 | OUTPUT_DIR: "logs/msmt17/agw_S50" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/bagtricks_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "101x" 6 | WITH_IBN: True 7 | 8 | DATASETS: 9 | NAMES: ("MSMT17",) 10 | TESTS: ("MSMT17",) 11 | 12 | OUTPUT_DIR: "logs/msmt17/bagtricks_R101-ibn" 13 | 14 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/bagtricks_R18_ReduceHead.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | MODEL: 4 | WEIGHTS: "" 5 | BACKBONE: 6 | PRETRAIN: True 7 | PRETRAIN_PATH: "./model/ImageNet/resnet18-5c106cde.pth" 8 | # NORM: "syncBN" 9 | DEPTH: "18x" 10 | HEADS: 11 | IN_FEAT: 512 12 | NAME: "ReductionHead" 13 | REDUCTION_DIM: 256 14 | # NORM: "syncBN" 15 | DATASETS: 16 | NAMES: ("MSMT17",) 17 | TESTS: ("MSMT17",) 18 | 19 | # SOLVER: 20 | # MAX_ITER: 200 21 | # STEPS: [80, 160] 22 | # BASE_LR: 0.0035 23 | 24 | OUTPUT_DIR: "logs/msmt17/bagtricks_R50_ReduceHead" 25 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/bagtricks_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_IBN: True 6 | 7 | DATASETS: 8 | NAMES: ("MSMT17",) 9 | TESTS: ("MSMT17",) 10 | 11 | OUTPUT_DIR: "logs/msmt17/bagtricks_R50-ibn" 12 | 13 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/bagtricks_R50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | DATASETS: 4 | NAMES: ("MSMT17",) 5 | TESTS: ("MSMT17",) 6 | 7 | OUTPUT_DIR: "logs/msmt17/bagtricks_R50" 8 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/bagtricks_R50_ReduceHead.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | MODEL: 4 | WEIGHTS: "./model/MSMT17/msmt_bot_R50.pth" 5 | BACKBONE: 6 | PRETRAIN: False 7 | PRETRAIN_PATH: "" 8 | # NORM: "syncBN" 9 | HEADS: 10 | NAME: "ReductionHead" 11 | REDUCTION_DIM: 256 12 | # NORM: "syncBN" 13 | DATASETS: 14 | NAMES: ("MSMT17",) 15 | TESTS: ("MSMT17",) 16 | 17 | # SOLVER: 18 | # MAX_ITER: 200 19 | # STEPS: [80, 160] 20 | # BASE_LR: 3.5e-5 21 | 22 | OUTPUT_DIR: "logs/msmt17/bagtricks_R50_ReduceHead_CD" 23 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/bagtricks_S50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | NAME: "build_resnest_backbone" 6 | 7 | DATASETS: 8 | NAMES: ("MSMT17",) 9 | TESTS: ("MSMT17",) 10 | 11 | OUTPUT_DIR: "logs/msmt17/bagtricks_S50" 12 | 13 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/mgn_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-MGN.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_IBN: True 6 | 7 | DATASETS: 8 | NAMES: ("MSMT17",) 9 | TESTS: ("MSMT17",) 10 | 11 | OUTPUT_DIR: "logs/msmt17/mgn_R50-ibn" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/sbs_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "101x" 6 | WITH_IBN: True 7 | 8 | DATASETS: 9 | NAMES: ("MSMT17",) 10 | TESTS: ("MSMT17",) 11 | 12 | OUTPUT_DIR: "logs/msmt17/sbs_R101-ibn" 13 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/sbs_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_IBN: True 6 | 7 | DATASETS: 8 | NAMES: ("MSMT17",) 9 | TESTS: ("MSMT17",) 10 | 11 | OUTPUT_DIR: "logs/msmt17/sbs_R50-ibn" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/sbs_R50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | DATASETS: 4 | NAMES: ("MSMT17",) 5 | TESTS: ("MSMT17",) 6 | 7 | OUTPUT_DIR: "logs/msmt17/sbs_R50" 8 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/sbs_R50_ReduceHead.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | WEIGHTS: "./model/MSMT17/msmt_sbs_R50.pth" 5 | BACKBONE: 6 | PRETRAIN: False 7 | PRETRAIN_PATH: "" 8 | CD: True 9 | # NORM: "syncBN" 10 | HEADS: 11 | NAME: "ReductionHead" 12 | REDUCTION_DIM: 256 13 | # NORM: "syncBN" 14 | DATASETS: 15 | NAMES: ("MSMT17",) 16 | TESTS: ("MSMT17",) 17 | 18 | # SOLVER: 19 | # MAX_ITER: 200 20 | # STEPS: [80, 160] 21 | # BASE_LR: 3.5e-5 22 | 23 | OUTPUT_DIR: "logs/msmt17/sbs_R50_ReductionHead_CD8" 24 | -------------------------------------------------------------------------------- /fast-reid/configs/MSMT17/sbs_S50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | NAME: "build_resnest_backbone" 6 | 7 | DATASETS: 8 | NAMES: ("MSMT17",) 9 | TESTS: ("MSMT17",) 10 | 11 | OUTPUT_DIR: "logs/msmt17/sbs_S50" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/Market1501/AGW_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-AGW.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "101x" 6 | WITH_IBN: True 7 | 8 | DATASETS: 9 | NAMES: ("Market1501",) 10 | TESTS: ("Market1501",) 11 | 12 | OUTPUT_DIR: "logs/market1501/agw_R101-ibn" 13 | -------------------------------------------------------------------------------- /fast-reid/configs/Market1501/AGW_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-AGW.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_IBN: True 6 | 7 | DATASETS: 8 | NAMES: ("Market1501",) 9 | TESTS: ("Market1501",) 10 | 11 | OUTPUT_DIR: "logs/market1501/agw_R50-ibn" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/Market1501/AGW_R50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-AGW.yml" 2 | 3 | DATASETS: 4 | NAMES: ("Market1501",) 5 | TESTS: ("Market1501",) 6 | 7 | OUTPUT_DIR: "logs/market1501/agw_R50" 8 | -------------------------------------------------------------------------------- /fast-reid/configs/Market1501/AGW_S50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-AGW.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | NAME: "build_resnest_backbone" 6 | 7 | DATASETS: 8 | NAMES: ("Market1501",) 9 | TESTS: ("Market1501",) 10 | 11 | OUTPUT_DIR: "logs/market1501/agw_S50" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/Market1501/bagtricks_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "101x" 6 | WITH_IBN: True 7 | 8 | DATASETS: 9 | NAMES: ("Market1501",) 10 | TESTS: ("Market1501",) 11 | 12 | OUTPUT_DIR: "logs/market1501/bagtricks_R101-ibn" 13 | -------------------------------------------------------------------------------- /fast-reid/configs/Market1501/bagtricks_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_IBN: True 6 | 7 | DATASETS: 8 | NAMES: ("Market1501",) 9 | TESTS: ("Market1501",) 10 | 11 | OUTPUT_DIR: "logs/market1501/bagtricks_R50-ibn" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/Market1501/bagtricks_R50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | DATASETS: 4 | NAMES: ("Market1501",) 5 | TESTS: ("Market1501",) 6 | 7 | OUTPUT_DIR: "logs/market1501/bagtricks_R50" 8 | -------------------------------------------------------------------------------- /fast-reid/configs/Market1501/bagtricks_R50_ReduceHead.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | MODEL: 4 | WEIGHTS: "./model/Market1501/market_bot_R50.pth" 5 | BACKBONE: 6 | PRETRAIN: False 7 | PRETRAIN_PATH: "" 8 | # NORM: "syncBN" 9 | HEADS: 10 | NAME: "ReductionHead" 11 | REDUCTION_DIM: 256 12 | # NORM: "syncBN" 13 | DATASETS: 14 | NAMES: ("Market1501",) 15 | TESTS: ("Market1501",) 16 | 17 | # SOLVER: 18 | # MAX_ITER: 200 19 | # STEPS: [80, 160] 20 | # BASE_LR: 0.000035 21 | 22 | OUTPUT_DIR: "logs/market1501/bagtricks_R50_ReduceHead_CD" 23 | -------------------------------------------------------------------------------- /fast-reid/configs/Market1501/bagtricks_S50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | NAME: "build_resnest_backbone" 6 | 7 | DATASETS: 8 | NAMES: ("Market1501",) 9 | TESTS: ("Market1501",) 10 | 11 | OUTPUT_DIR: "logs/market1501/bagtricks_S50" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/Market1501/mgn_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-MGN.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_IBN: True 6 | 7 | DATASETS: 8 | NAMES: ("Market1501",) 9 | TESTS: ("Market1501",) 10 | 11 | 12 | OUTPUT_DIR: "logs/market1501/mgn_R50-ibn" 13 | -------------------------------------------------------------------------------- /fast-reid/configs/Market1501/sbs_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "101x" 6 | WITH_IBN: True 7 | 8 | DATASETS: 9 | NAMES: ("Market1501",) 10 | TESTS: ("Market1501",) 11 | 12 | OUTPUT_DIR: "logs/market1501/sbs_R101-ibn" 13 | -------------------------------------------------------------------------------- /fast-reid/configs/Market1501/sbs_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_IBN: True 6 | 7 | DATASETS: 8 | NAMES: ("Market1501",) 9 | TESTS: ("Market1501",) 10 | 11 | OUTPUT_DIR: "logs/market1501/sbs_R50-ibn" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/Market1501/sbs_R50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | DATASETS: 4 | NAMES: ("Market1501",) 5 | TESTS: ("Market1501",) 6 | 7 | OUTPUT_DIR: "logs/market1501/sbs_R50" 8 | -------------------------------------------------------------------------------- /fast-reid/configs/Market1501/sbs_R50_ReduceHead.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | WEIGHTS: "./model/Market1501/market_sbs_R50.pth" 5 | BACKBONE: 6 | PRETRAIN: False 7 | PRETRAIN_PATH: "" 8 | CD: True 9 | # NORM: "syncBN" 10 | HEADS: 11 | NAME: "ReductionHead" 12 | REDUCTION_DIM: 256 13 | # NORM: "syncBN" 14 | DATASETS: 15 | NAMES: ("Market1501",) 16 | TESTS: ("Market1501",) 17 | 18 | # SOLVER: 19 | # MAX_ITER: 200 20 | # STEPS: [80, 160] 21 | # BASE_LR: 0.000035 22 | 23 | OUTPUT_DIR: "logs/market1501/sbs_R50_ReduceHead_CD8" 24 | -------------------------------------------------------------------------------- /fast-reid/configs/Market1501/sbs_S50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | NAME: "build_resnest_backbone" 6 | 7 | DATASETS: 8 | NAMES: ("Market1501",) 9 | TESTS: ("Market1501",) 10 | 11 | OUTPUT_DIR: "logs/market1501/sbs_S50" 12 | -------------------------------------------------------------------------------- /fast-reid/configs/NAIC/bagtricks_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_IBN: True 6 | 7 | DATASETS: 8 | NAMES: ("NAIC",) 9 | TESTS: ("NAIC",) 10 | 11 | TEST: 12 | ENABLED: False 13 | 14 | OUTPUT_DIR: "logs/naic/bagtricks_R50-ibn" 15 | -------------------------------------------------------------------------------- /fast-reid/configs/NAIC/bagtricks_R50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | MODEL: 4 | WEIGHTS: "./model/Market1501/market_bot_R50.pth" 5 | 6 | DATASETS: 7 | NAMES: ("NAIC",) 8 | TESTS: ("NAIC",) 9 | 10 | OUTPUT_DIR: "logs/naic/bagtricks_R50" 11 | -------------------------------------------------------------------------------- /fast-reid/configs/NAIC/sbs_R101-ibn-lgc.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | WEIGHTS: "./logs/naic/sbs_R101_ibn/model_batchsize_128_all_final.pth" 5 | BACKBONE: 6 | DEPTH: "101x" 7 | WITH_IBN: True 8 | PRETRAIN: False 9 | PRETRAIN_PATH: "" 10 | NORM: "syncBN" 11 | HEADS: 12 | NORM: "syncBN" 13 | PIXEL_STD: 14 | - 58.395 15 | - -57.120000000000005 16 | - -57.375 17 | 18 | DATASETS: 19 | NAMES: ("NAIC",) 20 | TESTS: ("NAIC",) 21 | # SPLIT_FILE: "NAIC_train_val.txt" 22 | 23 | DATALOADER: 24 | NUM_INSTANCE: 4 25 | 26 | INPUT: 27 | CJ: 28 | ENABLED: True 29 | PROB: 0.2 30 | BRIGHTNESS: 0.15 31 | CONTRAST: 0.15 32 | SATURATION: 0.1 33 | HUE: 0.1 34 | RPT: 35 | ENABLED: True 36 | PROB: 0.2 37 | 38 | SOLVER: 39 | IMS_PER_BATCH: 128 40 | 41 | TEST: 42 | ENABLED: False 43 | RERANK: 44 | ENABLED: True 45 | K1: 14 46 | K2: 4 47 | LAMBDA: 0.4 48 | JSON_FILE: "sbs_R101-ibn_all_k1_14_k2_4_lambda_0.4_result.json" 49 | 50 | 51 | OUTPUT_DIR: "logs/naic/sbs_R101-ibn" -------------------------------------------------------------------------------- /fast-reid/configs/NAIC/sbs_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | # FREEZE_LAYERS: [""] 5 | WEIGHTS: "./logs/naic/sbs_R101_ibn/model_final.pth" 6 | BACKBONE: 7 | PRETRAIN: False 8 | PRETRAIN_PATH: "" 9 | CD: False 10 | DEPTH: "101x" 11 | WITH_IBN: True 12 | 13 | DATASETS: 14 | NAMES: ("NAIC",) 15 | TESTS: ("NAIC",) 16 | 17 | DATALOADER: 18 | NUM_INSTANCE: 4 19 | 20 | INPUT: 21 | SIZE_TRAIN: [256, 128] 22 | SIZE_TEST: [256, 128] 23 | 24 | # SOLVER: 25 | # MAX_ITER: 20 26 | # BASE_LR: 0.0001 27 | 28 | TEST: 29 | ENABLED: True 30 | EVAL_INIT: False 31 | RERANK: 32 | ENABLED: True 33 | JSON_FILE: 'logs/naic/ensemble/sbs-R50_sbs-R101.json' 34 | 35 | OUTPUT_DIR: "logs/naic/ensemble" 36 | -------------------------------------------------------------------------------- /fast-reid/configs/NAIC/sbs_R50-cd.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | # FREEZE_LAYERS: [""] 5 | WEIGHTS: "./model/NAIC/msmt_sbs_R50.pth" 6 | BACKBONE: 7 | PRETRAIN: False 8 | PRETRAIN_PATH: "" 9 | CD: True 10 | NORM: "syncBN" 11 | 12 | HEADS: 13 | NORM: "syncBN" 14 | 15 | DATASETS: 16 | NAMES: ("NAIC",) 17 | TESTS: ("NAIC",) 18 | 19 | DATALOADER: 20 | NUM_INSTANCE: 4 21 | 22 | INPUT: 23 | SIZE_TRAIN: [512, 256] 24 | SIZE_TEST: [512, 256] 25 | 26 | # SOLVER: 27 | # IMS_PER_BATCH: 512 28 | # MAX_ITER: 20 29 | # BASE_LR: 0.0001 30 | 31 | TEST: 32 | ENABLED: True 33 | EVAL_INIT: False 34 | EVAL_PERIOD: 2 35 | RERANK: 36 | ENABLED: False 37 | JSON_FILE: '' 38 | # IMS_PER_BATCH: 512 39 | 40 | OUTPUT_DIR: "logs/naic/sbs_R50_cd" 41 | -------------------------------------------------------------------------------- /fast-reid/configs/NAIC/sbs_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_IBN: True 6 | 7 | DATASETS: 8 | NAMES: ("NAIC",) 9 | TESTS: ("NAIC",) 10 | ROOT: "" 11 | SPLIT_FILE: "NAIC_train_val.txt" 12 | 13 | OUTPUT_DIR: "logs/naic/sbs_R50-ibn" -------------------------------------------------------------------------------- /fast-reid/configs/NAIC/sbs_R50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | # FREEZE_LAYERS: [""] 5 | WEIGHTS: "logs/naic/sbs_R50_rc/model_final.pth" 6 | BACKBONE: 7 | PRETRAIN: False 8 | PRETRAIN_PATH: "" 9 | CD: False 10 | NORM: "syncBN" 11 | 12 | HEADS: 13 | NORM: "syncBN" 14 | # CLS_LAYER: "" 15 | 16 | # LOSSES: 17 | # NAME: ("NpairLoss", ) 18 | 19 | DATASETS: 20 | NAMES: ("NAIC",) 21 | TESTS: ("NAIC",) 22 | # EXTRA_DATA: 23 | # ROOTS: ["datasets/NAIC2019/1", "datasets/NAIC2019/2"] 24 | # LISTS: ["train_list.txt", "train_list.txt"] 25 | 26 | DATALOADER: 27 | NUM_INSTANCE: 4 28 | 29 | INPUT: 30 | SIZE_TRAIN: [256, 128] 31 | SIZE_TEST: [256, 128] 32 | RC: 33 | ENABLED: True 34 | 35 | SOLVER: 36 | MAX_ITER: 80 37 | IMS_PER_BATCH: 256 38 | # BASE_LR: 0.0001 39 | 40 | TEST: 41 | ENABLED: True 42 | EVAL_INIT: False 43 | EVAL_PERIOD: 2 44 | RERANK: 45 | ENABLED: True 46 | JSON_FILE: 'logs/naic/sbs_R50_rc/sbs_R50_rc.json' 47 | 48 | OUTPUT_DIR: "logs/naic/sbs_R50_rc" 49 | -------------------------------------------------------------------------------- /fast-reid/configs/NAIC/sbs_R50_p2019.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | # FREEZE_LAYERS: [""] 5 | WEIGHTS: "logs/naic/sbs_R50_p2019/model_final.pth" 6 | BACKBONE: 7 | PRETRAIN: False 8 | PRETRAIN_PATH: "" 9 | CD: False 10 | NORM: "syncBN" 11 | 12 | HEADS: 13 | NORM: "syncBN" 14 | # CLS_LAYER: "" 15 | 16 | # LOSSES: 17 | # NAME: ("NpairLoss", ) 18 | 19 | DATASETS: 20 | NAMES: ("NAIC",) 21 | TESTS: ("NAIC",) 22 | EXTRA_DATA: 23 | ROOTS: ["datasets/NAIC2019/1", "datasets/NAIC2019/2"] 24 | LISTS: ["train_list.txt", "train_list.txt"] 25 | 26 | DATALOADER: 27 | NUM_INSTANCE: 4 28 | 29 | INPUT: 30 | SIZE_TRAIN: [256, 128] 31 | SIZE_TEST: [256, 128] 32 | RC: 33 | ENABLED: True 34 | 35 | SOLVER: 36 | IMS_PER_BATCH: 128 37 | MAX_ITER: 80 38 | # BASE_LR: 0.0001 39 | 40 | TEST: 41 | ENABLED: True 42 | EVAL_INIT: False 43 | EVAL_PERIOD: 2 44 | RERANK: 45 | ENABLED: True 46 | JSON_FILE: 'logs/naic/sbs_R50_p2019/sbs_R50_p2019.json' 47 | 48 | OUTPUT_DIR: "logs/naic/sbs_R50_p2019" 49 | -------------------------------------------------------------------------------- /fast-reid/configs/VERIWild/bagtricks_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | INPUT: 4 | SIZE_TRAIN: [256, 256] 5 | SIZE_TEST: [256, 256] 6 | 7 | MODEL: 8 | BACKBONE: 9 | WITH_IBN: True 10 | HEADS: 11 | POOL_LAYER: gempool 12 | LOSSES: 13 | TRI: 14 | HARD_MINING: False 15 | MARGIN: 0.0 16 | 17 | DATASETS: 18 | NAMES: ("VeRiWild",) 19 | TESTS: ("SmallVeRiWild", "MediumVeRiWild", "LargeVeRiWild",) 20 | 21 | SOLVER: 22 | IMS_PER_BATCH: 128 23 | MAX_ITER: 60 24 | STEPS: [30, 50] 25 | WARMUP_ITERS: 10 26 | 27 | CHECKPOINT_PERIOD: 20 28 | 29 | TEST: 30 | EVAL_PERIOD: 20 31 | IMS_PER_BATCH: 128 32 | 33 | OUTPUT_DIR: "logs/veriwild/bagtricks_R50-ibn_4gpu" 34 | -------------------------------------------------------------------------------- /fast-reid/configs/VeRi/sbs_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-Strongerbaseline.yml" 2 | 3 | INPUT: 4 | SIZE_TRAIN: [256, 256] 5 | SIZE_TEST: [256, 256] 6 | 7 | MODEL: 8 | BACKBONE: 9 | WITH_IBN: True 10 | 11 | SOLVER: 12 | OPT: "SGD" 13 | BASE_LR: 0.01 14 | ETA_MIN_LR: 7.7e-5 15 | 16 | IMS_PER_BATCH: 64 17 | MAX_ITER: 60 18 | DELAY_ITERS: 30 19 | WARMUP_ITERS: 10 20 | FREEZE_ITERS: 10 21 | 22 | CHECKPOINT_PERIOD: 20 23 | 24 | DATASETS: 25 | NAMES: ("VeRi",) 26 | TESTS: ("VeRi",) 27 | 28 | TEST: 29 | EVAL_PERIOD: 20 30 | IMS_PER_BATCH: 128 31 | 32 | OUTPUT_DIR: "logs/veri/sbs_R50-ibn" 33 | -------------------------------------------------------------------------------- /fast-reid/configs/VehicleID/bagtricks_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../Base-bagtricks.yml" 2 | 3 | INPUT: 4 | SIZE_TRAIN: [256, 256] 5 | SIZE_TEST: [256, 256] 6 | 7 | MODEL: 8 | BACKBONE: 9 | WITH_IBN: True 10 | HEADS: 11 | POOL_LAYER: gempool 12 | LOSSES: 13 | TRI: 14 | HARD_MINING: False 15 | MARGIN: 0.0 16 | 17 | DATASETS: 18 | NAMES: ("VehicleID",) 19 | TESTS: ("SmallVehicleID", "MediumVehicleID", "LargeVehicleID",) 20 | 21 | SOLVER: 22 | BIAS_LR_FACTOR: 1. 23 | 24 | IMS_PER_BATCH: 512 25 | MAX_ITER: 60 26 | STEPS: [30, 50] 27 | WARMUP_ITERS: 10 28 | 29 | CHECKPOINT_PERIOD: 20 30 | 31 | TEST: 32 | EVAL_PERIOD: 20 33 | IMS_PER_BATCH: 128 34 | 35 | OUTPUT_DIR: "logs/vehicleid/bagtricks_R50-ibn_4gpu" 36 | -------------------------------------------------------------------------------- /fast-reid/demo/README.md: -------------------------------------------------------------------------------- 1 | # FastReID Demo 2 | 3 | We provide a command line tool to run a simple demo of builtin models. 4 | 5 | You can run this command to get cosine similarites between different images 6 | 7 | ```bash 8 | cd demo/ 9 | sh run_demo.sh 10 | ``` -------------------------------------------------------------------------------- /fast-reid/demo/__pycache__/Eval_MOT.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/demo/__pycache__/Eval_MOT.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/demo/__pycache__/predictor.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/demo/__pycache__/predictor.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/demo/plot_roc_with_pickle.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: xingyu liao 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | import matplotlib.pyplot as plt 8 | import sys 9 | 10 | sys.path.append('.') 11 | from fastreid.utils.visualizer import Visualizer 12 | 13 | if __name__ == "__main__": 14 | baseline_res = Visualizer.load_roc_info("logs/duke_vis/roc_info.pickle") 15 | mgn_res = Visualizer.load_roc_info("logs/mgn_duke_vis/roc_info.pickle") 16 | 17 | fig = Visualizer.plot_roc_curve(baseline_res['fpr'], baseline_res['tpr'], name='baseline') 18 | Visualizer.plot_roc_curve(mgn_res['fpr'], mgn_res['tpr'], name='mgn', fig=fig) 19 | plt.savefig('roc.jpg') 20 | 21 | fig = Visualizer.plot_distribution(baseline_res['pos'], baseline_res['neg'], name='baseline') 22 | Visualizer.plot_distribution(mgn_res['pos'], mgn_res['neg'], name='mgn', fig=fig) 23 | plt.savefig('dist.jpg') 24 | -------------------------------------------------------------------------------- /fast-reid/demo/run_bjz.sh: -------------------------------------------------------------------------------- 1 | python demo.py --config-file ../projects/bjzProject/logs/bjz/R50_512dim_circle_bjz_0618_8x3/config.yaml \ 2 | --input \ 3 | "/home/liaoxingyu2/lxy/logs/badcase/seed.jpg" \ 4 | "/home/liaoxingyu2/lxy/logs/badcase/leaf_0.4883.jpg" \ 5 | "/home/liaoxingyu2/lxy/logs/badcase/leaf_0.467.jpg" \ 6 | --output logs/R50_256x128_pytorch_feat_output \ 7 | --opts MODEL.WEIGHTS ../projects/bjzProject/logs/bjz/R50_512dim_circle_bjz_0618_8x32/model_final.pth \ 8 | MODEL.DEVICE "cuda: 2" 9 | 10 | -------------------------------------------------------------------------------- /fast-reid/demo/run_demo.sh: -------------------------------------------------------------------------------- 1 | python demo/visualize_result.py \ 2 | --config-file config/NAIC/sbs_R50-cd.yaml \ 3 | --parallel \ 4 | --vis-label \ 5 | --dataset-name 'NAIC' \ 6 | --output logs/mgn_duke_vis \ 7 | --opts MODEL.WEIGHTS logs/naic/sbs_R50_cd/model_partial.pth 8 | -------------------------------------------------------------------------------- /fast-reid/demo/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/demo/test.png -------------------------------------------------------------------------------- /fast-reid/fastreid/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | 8 | __version__ = "0.1.0" -------------------------------------------------------------------------------- /fast-reid/fastreid/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/config/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: l1aoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from .config import CfgNode, get_cfg 8 | from .defaults import _C as cfg 9 | -------------------------------------------------------------------------------- /fast-reid/fastreid/config/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/config/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/config/__pycache__/config.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/config/__pycache__/config.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/config/__pycache__/defaults.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/config/__pycache__/defaults.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: sherlock 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from .build import build_reid_train_loader, build_reid_test_loader 8 | -------------------------------------------------------------------------------- /fast-reid/fastreid/data/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/__pycache__/build.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/__pycache__/build.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/__pycache__/common.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/__pycache__/common.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/__pycache__/data_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/__pycache__/data_utils.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/common.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | import torch 8 | from torch.utils.data import Dataset 9 | 10 | from .data_utils import read_image 11 | 12 | 13 | class CommDataset(Dataset): 14 | """Image Person ReID Dataset""" 15 | 16 | def __init__(self, img_items, transform=None, relabel=True): 17 | self.img_items = img_items 18 | self.transform = transform 19 | self.relabel = relabel 20 | 21 | self.pid_dict = {} 22 | if self.relabel: 23 | pids = list() 24 | for i, item in enumerate(img_items): 25 | if item[1] in pids: continue 26 | pids.append(item[1]) 27 | self.pids = pids 28 | self.pid_dict = dict([(p, i) for i, p in enumerate(self.pids)]) 29 | 30 | def __len__(self): 31 | return len(self.img_items) 32 | 33 | def __getitem__(self, index): 34 | img_path, pid, camid = self.img_items[index] 35 | img = read_image(img_path) 36 | if self.transform is not None: img = self.transform(img) 37 | if self.relabel: pid = self.pid_dict[pid] 38 | return { 39 | "images": img, 40 | "targets": pid, 41 | "camid": camid, 42 | "img_path": img_path.split('/')[-1] 43 | } 44 | 45 | @property 46 | def num_classes(self): 47 | return len(self.pids) 48 | -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from ...utils.registry import Registry 8 | 9 | DATASET_REGISTRY = Registry("DATASET") 10 | DATASET_REGISTRY.__doc__ = """ 11 | Registry for datasets 12 | It must returns an instance of :class:`Backbone`. 13 | """ 14 | 15 | from .cuhk03 import CUHK03 16 | from .dukemtmcreid import DukeMTMC 17 | from .market1501 import Market1501 18 | from .msmt17 import MSMT17 19 | from .veri import VeRi 20 | from .vehicleid import VehicleID, SmallVehicleID, MediumVehicleID, LargeVehicleID 21 | from .veriwild import VeRiWild, SmallVeRiWild, MediumVeRiWild, LargeVeRiWild 22 | from .naic import NAIC 23 | from .reid2019 import REID2019 24 | from .mot17 import MOT17 25 | from .mot20 import MOT20 26 | from .product10k import Product10k 27 | 28 | __all__ = [k for k in globals().keys() if "builtin" not in k and not k.startswith("_")] 29 | -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/datasets/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/__pycache__/bases.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/datasets/__pycache__/bases.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/__pycache__/cuhk03.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/datasets/__pycache__/cuhk03.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/__pycache__/dukemtmcreid.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/datasets/__pycache__/dukemtmcreid.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/__pycache__/market1501.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/datasets/__pycache__/market1501.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/__pycache__/mot17.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/datasets/__pycache__/mot17.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/__pycache__/mot20.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/datasets/__pycache__/mot20.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/__pycache__/msmt17.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/datasets/__pycache__/msmt17.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/__pycache__/naic.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/datasets/__pycache__/naic.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/__pycache__/product10k.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/datasets/__pycache__/product10k.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/__pycache__/reid2019.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/datasets/__pycache__/reid2019.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/__pycache__/vehicleid.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/datasets/__pycache__/vehicleid.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/__pycache__/veri.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/datasets/__pycache__/veri.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/__pycache__/veriwild.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/datasets/__pycache__/veriwild.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/datasets/reid2019.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: sherlock 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | import glob 8 | import os 9 | import os.path as osp 10 | import re 11 | import warnings 12 | 13 | from .bases import ImageDataset 14 | from ..datasets import DATASET_REGISTRY 15 | 16 | 17 | @DATASET_REGISTRY.register() 18 | class REID2019(ImageDataset): 19 | _junk_pids = [0, -1] 20 | data_dir = 'NAIC/1/A' 21 | dataset_name = "REID2019" 22 | def __init__(self, root='', **kwargs): 23 | self.root = root 24 | self.dataset_dir = osp.join(self.root, self.data_dir, self.dataset_name) 25 | if not osp.isdir(self.dataset_dir): 26 | warnings.warn('The current data dir:%s is not a dir.' % self.dataset_dir) 27 | 28 | self.train_dir = osp.join(self.dataset_dir) 29 | required_files = [ 30 | self.train_dir, 31 | ] 32 | train = self.process_dir(self.train_dir) 33 | query = [] 34 | gallery = [] 35 | super(REID2019, self).__init__(train, query, gallery, **kwargs) 36 | 37 | def process_dir(self, dir_path): 38 | data = [] 39 | num = 0 40 | for pid in os.listdir(dir_path): 41 | for img in glob.glob(osp.join(dir_path, pid, '*.png')): 42 | data.append([img, pid, num]) 43 | num += 1 44 | return data 45 | -------------------------------------------------------------------------------- /fast-reid/fastreid/data/samplers/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from .triplet_sampler import BalancedIdentitySampler, NaiveIdentitySampler 8 | from .data_sampler import TrainingSampler, InferenceSampler 9 | -------------------------------------------------------------------------------- /fast-reid/fastreid/data/samplers/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/samplers/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/samplers/__pycache__/data_sampler.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/samplers/__pycache__/data_sampler.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/samplers/__pycache__/triplet_sampler.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/samplers/__pycache__/triplet_sampler.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/transforms/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: sherlock 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | 8 | from .build import build_transforms 9 | from .transforms import * 10 | from .autoaugment import * 11 | -------------------------------------------------------------------------------- /fast-reid/fastreid/data/transforms/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/transforms/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/transforms/__pycache__/autoaugment.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/transforms/__pycache__/autoaugment.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/transforms/__pycache__/build.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/transforms/__pycache__/build.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/transforms/__pycache__/functional.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/transforms/__pycache__/functional.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/data/transforms/__pycache__/transforms.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/data/transforms/__pycache__/transforms.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/engine/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | from .train_loop import * 7 | 8 | __all__ = [k for k in globals().keys() if not k.startswith("_")] 9 | 10 | 11 | # prefer to let hooks and defaults live in separate namespaces (therefore not in __all__) 12 | # but still make them available here 13 | from .hooks import * 14 | from .defaults import * 15 | from .launch import * 16 | -------------------------------------------------------------------------------- /fast-reid/fastreid/engine/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/engine/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/engine/__pycache__/defaults.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/engine/__pycache__/defaults.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/engine/__pycache__/hooks.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/engine/__pycache__/hooks.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/engine/__pycache__/launch.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/engine/__pycache__/launch.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/engine/__pycache__/train_loop.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/engine/__pycache__/train_loop.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/evaluation/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved 2 | from .evaluator import DatasetEvaluator, inference_context, inference_on_dataset 3 | from .rank import evaluate_rank 4 | from .roc import evaluate_roc 5 | from .reid_evaluation import ReidEvaluator 6 | from .testing import print_csv_format, verify_results 7 | 8 | __all__ = [k for k in globals().keys() if not k.startswith("_")] 9 | -------------------------------------------------------------------------------- /fast-reid/fastreid/evaluation/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/evaluation/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/evaluation/__pycache__/evaluator.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/evaluation/__pycache__/evaluator.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/evaluation/__pycache__/query_expansion.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/evaluation/__pycache__/query_expansion.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/evaluation/__pycache__/rank.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/evaluation/__pycache__/rank.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/evaluation/__pycache__/reid_evaluation.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/evaluation/__pycache__/reid_evaluation.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/evaluation/__pycache__/rerank.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/evaluation/__pycache__/rerank.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/evaluation/__pycache__/roc.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/evaluation/__pycache__/roc.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/evaluation/__pycache__/testing.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/evaluation/__pycache__/testing.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/evaluation/rank_cylib/Makefile: -------------------------------------------------------------------------------- 1 | all: 2 | python3 setup.py build_ext --inplace 3 | rm -rf build 4 | python3 test_cython.py 5 | clean: 6 | rm -rf build 7 | rm -f rank_cy.c *.so 8 | -------------------------------------------------------------------------------- /fast-reid/fastreid/evaluation/rank_cylib/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ -------------------------------------------------------------------------------- /fast-reid/fastreid/evaluation/rank_cylib/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/evaluation/rank_cylib/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/evaluation/rank_cylib/rank_cy.cpython-37m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/evaluation/rank_cylib/rank_cy.cpython-37m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /fast-reid/fastreid/evaluation/rank_cylib/roc_cy.cpython-37m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/evaluation/rank_cylib/roc_cy.cpython-37m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /fast-reid/fastreid/evaluation/rank_cylib/setup.py: -------------------------------------------------------------------------------- 1 | from distutils.core import setup 2 | from distutils.extension import Extension 3 | 4 | import numpy as np 5 | from Cython.Build import cythonize 6 | 7 | 8 | def numpy_include(): 9 | try: 10 | numpy_include = np.get_include() 11 | except AttributeError: 12 | numpy_include = np.get_numpy_include() 13 | return numpy_include 14 | 15 | 16 | ext_modules = [ 17 | Extension( 18 | 'rank_cy', 19 | ['rank_cy.pyx'], 20 | include_dirs=[numpy_include()], 21 | ), 22 | Extension( 23 | 'roc_cy', 24 | ['roc_cy.pyx'], 25 | include_dirs=[numpy_include()], 26 | ) 27 | ] 28 | 29 | setup( 30 | name='Cython-based reid evaluation code', 31 | ext_modules=cythonize(ext_modules) 32 | ) 33 | -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from .activation import * 8 | from .arc_softmax import ArcSoftmax 9 | from .circle_softmax import CircleSoftmax 10 | from .am_softmax import AMSoftmax 11 | from .batch_drop import BatchDrop 12 | from .batch_norm import * 13 | from .context_block import ContextBlock 14 | from .frn import FRN, TLU 15 | from .non_local import Non_local 16 | from .pooling import * 17 | from .se_layer import SELayer 18 | from .splat import SplAtConv2d 19 | from .cd_conv import CDConvBlock 20 | -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/layers/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/__pycache__/activation.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/layers/__pycache__/activation.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/__pycache__/am_softmax.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/layers/__pycache__/am_softmax.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/__pycache__/arc_softmax.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/layers/__pycache__/arc_softmax.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/__pycache__/batch_drop.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/layers/__pycache__/batch_drop.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/__pycache__/batch_norm.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/layers/__pycache__/batch_norm.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/__pycache__/cd_conv.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/layers/__pycache__/cd_conv.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/__pycache__/circle_softmax.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/layers/__pycache__/circle_softmax.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/__pycache__/context_block.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/layers/__pycache__/context_block.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/__pycache__/frn.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/layers/__pycache__/frn.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/__pycache__/non_local.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/layers/__pycache__/non_local.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/__pycache__/pooling.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/layers/__pycache__/pooling.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/__pycache__/se_layer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/layers/__pycache__/se_layer.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/__pycache__/splat.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/layers/__pycache__/splat.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/batch_drop.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | import random 8 | 9 | from torch import nn 10 | 11 | 12 | class BatchDrop(nn.Module): 13 | """ref: https://github.com/daizuozhuo/batch-dropblock-network/blob/master/models/networks.py 14 | batch drop mask 15 | """ 16 | 17 | def __init__(self, h_ratio, w_ratio): 18 | super(BatchDrop, self).__init__() 19 | self.h_ratio = h_ratio 20 | self.w_ratio = w_ratio 21 | 22 | def forward(self, x): 23 | if self.training: 24 | h, w = x.size()[-2:] 25 | rh = round(self.h_ratio * h) 26 | rw = round(self.w_ratio * w) 27 | sx = random.randint(0, h - rh) 28 | sy = random.randint(0, w - rw) 29 | mask = x.new_ones(x.size()) 30 | mask[:, :, sx:sx + rh, sy:sy + rw] = 0 31 | x = x * mask 32 | return x 33 | -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/circle_softmax.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | import math 8 | 9 | import torch 10 | import torch.nn as nn 11 | import torch.nn.functional as F 12 | from torch.nn import Parameter 13 | 14 | 15 | class CircleSoftmax(nn.Module): 16 | def __init__(self, cfg, in_feat, num_classes): 17 | super().__init__() 18 | self.in_feat = in_feat 19 | self._num_classes = num_classes 20 | self._s = cfg.MODEL.HEADS.SCALE 21 | self._m = cfg.MODEL.HEADS.MARGIN 22 | 23 | self.weight = Parameter(torch.Tensor(num_classes, in_feat)) 24 | nn.init.kaiming_uniform_(self.weight, a=math.sqrt(5)) 25 | 26 | def forward(self, features, targets): 27 | sim_mat = F.linear(F.normalize(features), F.normalize(self.weight)) 28 | alpha_p = torch.clamp_min(-sim_mat.detach() + 1 + self._m, min=0.) 29 | alpha_n = torch.clamp_min(sim_mat.detach() + self._m, min=0.) 30 | delta_p = 1 - self._m 31 | delta_n = self._m 32 | 33 | s_p = self._s * alpha_p * (sim_mat - delta_p) 34 | s_n = self._s * alpha_n * (sim_mat - delta_n) 35 | 36 | targets = F.one_hot(targets, num_classes=self._num_classes) 37 | 38 | pred_class_logits = targets * s_p + (1.0 - targets) * s_n 39 | 40 | return pred_class_logits 41 | 42 | def extra_repr(self): 43 | return 'in_features={}, num_classes={}, scale={}, margin={}'.format( 44 | self.in_feat, self._num_classes, self._s, self._m 45 | ) 46 | -------------------------------------------------------------------------------- /fast-reid/fastreid/layers/se_layer.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from torch import nn 8 | 9 | 10 | class SELayer(nn.Module): 11 | def __init__(self, channel, reduction=16): 12 | super(SELayer, self).__init__() 13 | self.avg_pool = nn.AdaptiveAvgPool2d(1) 14 | self.fc = nn.Sequential( 15 | nn.Linear(channel, int(channel / reduction), bias=False), 16 | nn.ReLU(inplace=True), 17 | nn.Linear(int(channel / reduction), channel, bias=False), 18 | nn.Sigmoid() 19 | ) 20 | 21 | def forward(self, x): 22 | b, c, _, _ = x.size() 23 | y = self.avg_pool(x).view(b, c) 24 | y = self.fc(y).view(b, c, 1, 1) 25 | return x * y.expand_as(x) 26 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: sherlock 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from .meta_arch import build_model 8 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from .build import build_backbone, BACKBONE_REGISTRY 8 | 9 | from .resnet import build_resnet_backbone 10 | from .osnet import build_osnet_backbone 11 | from .resnest import build_resnest_backbone 12 | from .resnext import build_resnext_backbone 13 | from .regnet import build_regnet_backbone 14 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/backbones/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/__pycache__/build.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/backbones/__pycache__/build.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/__pycache__/osnet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/backbones/__pycache__/osnet.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/__pycache__/resnest.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/backbones/__pycache__/resnest.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/__pycache__/resnet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/backbones/__pycache__/resnet.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/__pycache__/resnext.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/backbones/__pycache__/resnext.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/build.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from ...utils.registry import Registry 8 | 9 | BACKBONE_REGISTRY = Registry("BACKBONE") 10 | BACKBONE_REGISTRY.__doc__ = """ 11 | Registry for backbones, which extract feature maps from images 12 | The registered object must be a callable that accepts two arguments: 13 | 1. A :class:`detectron2.config.CfgNode` 14 | 2. A :class:`detectron2.layers.ShapeSpec`, which contains the input shape specification. 15 | It must returns an instance of :class:`Backbone`. 16 | """ 17 | 18 | 19 | def build_backbone(cfg): 20 | """ 21 | Build a backbone from `cfg.MODEL.BACKBONE.NAME`. 22 | Returns: 23 | an instance of :class:`Backbone` 24 | """ 25 | 26 | backbone_name = cfg.MODEL.BACKBONE.NAME 27 | backbone = BACKBONE_REGISTRY.get(backbone_name)(cfg) 28 | return backbone 29 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | from .regnet import build_regnet_backbone 4 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/backbones/regnet/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/__pycache__/config.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/backbones/regnet/__pycache__/config.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/__pycache__/regnet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/backbones/regnet/__pycache__/regnet.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-1.6GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | DEPTH: 18 6 | W0: 80 7 | WA: 34.01 8 | WM: 2.25 9 | GROUP_W: 24 10 | OPTIM: 11 | LR_POLICY: cos 12 | BASE_LR: 0.8 13 | MAX_EPOCH: 100 14 | MOMENTUM: 0.9 15 | WEIGHT_DECAY: 5e-5 16 | WARMUP_EPOCHS: 5 17 | TRAIN: 18 | DATASET: imagenet 19 | IM_SIZE: 224 20 | BATCH_SIZE: 1024 21 | TEST: 22 | DATASET: imagenet 23 | IM_SIZE: 256 24 | BATCH_SIZE: 800 25 | NUM_GPUS: 8 26 | OUT_DIR: . 27 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-12GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | DEPTH: 19 6 | W0: 168 7 | WA: 73.36 8 | WM: 2.37 9 | GROUP_W: 112 10 | OPTIM: 11 | LR_POLICY: cos 12 | BASE_LR: 0.4 13 | MAX_EPOCH: 100 14 | MOMENTUM: 0.9 15 | WEIGHT_DECAY: 5e-5 16 | WARMUP_EPOCHS: 5 17 | TRAIN: 18 | DATASET: imagenet 19 | IM_SIZE: 224 20 | BATCH_SIZE: 512 21 | TEST: 22 | DATASET: imagenet 23 | IM_SIZE: 256 24 | BATCH_SIZE: 400 25 | NUM_GPUS: 8 26 | OUT_DIR: . 27 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-16GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | DEPTH: 22 6 | W0: 216 7 | WA: 55.59 8 | WM: 2.1 9 | GROUP_W: 128 10 | OPTIM: 11 | LR_POLICY: cos 12 | BASE_LR: 0.4 13 | MAX_EPOCH: 100 14 | MOMENTUM: 0.9 15 | WEIGHT_DECAY: 5e-5 16 | WARMUP_EPOCHS: 5 17 | TRAIN: 18 | DATASET: imagenet 19 | IM_SIZE: 224 20 | BATCH_SIZE: 512 21 | TEST: 22 | DATASET: imagenet 23 | IM_SIZE: 256 24 | BATCH_SIZE: 400 25 | NUM_GPUS: 8 26 | OUT_DIR: . 27 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-200MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | DEPTH: 13 6 | W0: 24 7 | WA: 36.44 8 | WM: 2.49 9 | GROUP_W: 8 10 | OPTIM: 11 | LR_POLICY: cos 12 | BASE_LR: 0.8 13 | MAX_EPOCH: 100 14 | MOMENTUM: 0.9 15 | WEIGHT_DECAY: 5e-5 16 | WARMUP_EPOCHS: 5 17 | TRAIN: 18 | DATASET: imagenet 19 | IM_SIZE: 224 20 | BATCH_SIZE: 1024 21 | TEST: 22 | DATASET: imagenet 23 | IM_SIZE: 256 24 | BATCH_SIZE: 800 25 | NUM_GPUS: 8 26 | OUT_DIR: . 27 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-3.2GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | DEPTH: 25 6 | W0: 88 7 | WA: 26.31 8 | WM: 2.25 9 | GROUP_W: 48 10 | OPTIM: 11 | LR_POLICY: cos 12 | BASE_LR: 0.4 13 | MAX_EPOCH: 100 14 | MOMENTUM: 0.9 15 | WEIGHT_DECAY: 5e-5 16 | WARMUP_EPOCHS: 5 17 | TRAIN: 18 | DATASET: imagenet 19 | IM_SIZE: 224 20 | BATCH_SIZE: 512 21 | TEST: 22 | DATASET: imagenet 23 | IM_SIZE: 256 24 | BATCH_SIZE: 400 25 | NUM_GPUS: 8 26 | OUT_DIR: . 27 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-32GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | DEPTH: 23 6 | W0: 320 7 | WA: 69.86 8 | WM: 2.0 9 | GROUP_W: 168 10 | OPTIM: 11 | LR_POLICY: cos 12 | BASE_LR: 0.2 13 | MAX_EPOCH: 100 14 | MOMENTUM: 0.9 15 | WEIGHT_DECAY: 5e-5 16 | WARMUP_EPOCHS: 5 17 | TRAIN: 18 | DATASET: imagenet 19 | IM_SIZE: 224 20 | BATCH_SIZE: 256 21 | TEST: 22 | DATASET: imagenet 23 | IM_SIZE: 256 24 | BATCH_SIZE: 200 25 | NUM_GPUS: 8 26 | OUT_DIR: . 27 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-4.0GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | DEPTH: 23 6 | W0: 96 7 | WA: 38.65 8 | WM: 2.43 9 | GROUP_W: 40 10 | OPTIM: 11 | LR_POLICY: cos 12 | BASE_LR: 0.4 13 | MAX_EPOCH: 100 14 | MOMENTUM: 0.9 15 | WEIGHT_DECAY: 5e-5 16 | WARMUP_EPOCHS: 5 17 | TRAIN: 18 | DATASET: imagenet 19 | IM_SIZE: 224 20 | BATCH_SIZE: 512 21 | TEST: 22 | DATASET: imagenet 23 | IM_SIZE: 256 24 | BATCH_SIZE: 400 25 | NUM_GPUS: 8 26 | OUT_DIR: . 27 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-400MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | DEPTH: 22 6 | W0: 24 7 | WA: 24.48 8 | WM: 2.54 9 | GROUP_W: 16 10 | OPTIM: 11 | LR_POLICY: cos 12 | BASE_LR: 0.8 13 | MAX_EPOCH: 100 14 | MOMENTUM: 0.9 15 | WEIGHT_DECAY: 5e-5 16 | WARMUP_EPOCHS: 5 17 | TRAIN: 18 | DATASET: imagenet 19 | IM_SIZE: 224 20 | BATCH_SIZE: 1024 21 | TEST: 22 | DATASET: imagenet 23 | IM_SIZE: 256 24 | BATCH_SIZE: 800 25 | NUM_GPUS: 8 26 | OUT_DIR: . 27 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-6.4GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | DEPTH: 17 6 | W0: 184 7 | WA: 60.83 8 | WM: 2.07 9 | GROUP_W: 56 10 | OPTIM: 11 | LR_POLICY: cos 12 | BASE_LR: 0.4 13 | MAX_EPOCH: 100 14 | MOMENTUM: 0.9 15 | WEIGHT_DECAY: 5e-5 16 | WARMUP_EPOCHS: 5 17 | TRAIN: 18 | DATASET: imagenet 19 | IM_SIZE: 224 20 | BATCH_SIZE: 512 21 | TEST: 22 | DATASET: imagenet 23 | IM_SIZE: 256 24 | BATCH_SIZE: 400 25 | NUM_GPUS: 8 26 | OUT_DIR: . 27 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-600MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | DEPTH: 16 6 | W0: 48 7 | WA: 36.97 8 | WM: 2.24 9 | GROUP_W: 24 10 | OPTIM: 11 | LR_POLICY: cos 12 | BASE_LR: 0.8 13 | MAX_EPOCH: 100 14 | MOMENTUM: 0.9 15 | WEIGHT_DECAY: 5e-5 16 | WARMUP_EPOCHS: 5 17 | TRAIN: 18 | DATASET: imagenet 19 | IM_SIZE: 224 20 | BATCH_SIZE: 1024 21 | TEST: 22 | DATASET: imagenet 23 | IM_SIZE: 256 24 | BATCH_SIZE: 800 25 | NUM_GPUS: 8 26 | OUT_DIR: . 27 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-8.0GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | DEPTH: 23 6 | W0: 80 7 | WA: 49.56 8 | WM: 2.88 9 | GROUP_W: 120 10 | OPTIM: 11 | LR_POLICY: cos 12 | BASE_LR: 0.4 13 | MAX_EPOCH: 100 14 | MOMENTUM: 0.9 15 | WEIGHT_DECAY: 5e-5 16 | WARMUP_EPOCHS: 5 17 | TRAIN: 18 | DATASET: imagenet 19 | IM_SIZE: 224 20 | BATCH_SIZE: 512 21 | TEST: 22 | DATASET: imagenet 23 | IM_SIZE: 256 24 | BATCH_SIZE: 400 25 | NUM_GPUS: 8 26 | OUT_DIR: . 27 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnetx/RegNetX-800MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | DEPTH: 16 6 | W0: 56 7 | WA: 35.73 8 | WM: 2.28 9 | GROUP_W: 16 10 | OPTIM: 11 | LR_POLICY: cos 12 | BASE_LR: 0.8 13 | MAX_EPOCH: 100 14 | MOMENTUM: 0.9 15 | WEIGHT_DECAY: 5e-5 16 | WARMUP_EPOCHS: 5 17 | TRAIN: 18 | DATASET: imagenet 19 | IM_SIZE: 224 20 | BATCH_SIZE: 1024 21 | TEST: 22 | DATASET: imagenet 23 | IM_SIZE: 256 24 | BATCH_SIZE: 800 25 | NUM_GPUS: 8 26 | OUT_DIR: . 27 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-1.6GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | SE_ON: True 6 | DEPTH: 27 7 | W0: 48 8 | WA: 20.71 9 | WM: 2.65 10 | GROUP_W: 24 11 | OPTIM: 12 | LR_POLICY: cos 13 | BASE_LR: 0.8 14 | MAX_EPOCH: 100 15 | MOMENTUM: 0.9 16 | WEIGHT_DECAY: 5e-5 17 | WARMUP_EPOCHS: 5 18 | TRAIN: 19 | DATASET: imagenet 20 | IM_SIZE: 224 21 | BATCH_SIZE: 1024 22 | TEST: 23 | DATASET: imagenet 24 | IM_SIZE: 256 25 | BATCH_SIZE: 800 26 | NUM_GPUS: 8 27 | OUT_DIR: . 28 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-12GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | SE_ON: True 6 | DEPTH: 19 7 | W0: 168 8 | WA: 73.36 9 | WM: 2.37 10 | GROUP_W: 112 11 | OPTIM: 12 | LR_POLICY: cos 13 | BASE_LR: 0.4 14 | MAX_EPOCH: 100 15 | MOMENTUM: 0.9 16 | WEIGHT_DECAY: 5e-5 17 | WARMUP_EPOCHS: 5 18 | TRAIN: 19 | DATASET: imagenet 20 | IM_SIZE: 224 21 | BATCH_SIZE: 512 22 | TEST: 23 | DATASET: imagenet 24 | IM_SIZE: 256 25 | BATCH_SIZE: 400 26 | NUM_GPUS: 8 27 | OUT_DIR: . 28 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-16GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | SE_ON: True 6 | DEPTH: 18 7 | W0: 200 8 | WA: 106.23 9 | WM: 2.48 10 | GROUP_W: 112 11 | OPTIM: 12 | LR_POLICY: cos 13 | BASE_LR: 0.2 14 | MAX_EPOCH: 100 15 | MOMENTUM: 0.9 16 | WEIGHT_DECAY: 5e-5 17 | WARMUP_EPOCHS: 5 18 | TRAIN: 19 | DATASET: imagenet 20 | IM_SIZE: 224 21 | BATCH_SIZE: 256 22 | TEST: 23 | DATASET: imagenet 24 | IM_SIZE: 256 25 | BATCH_SIZE: 200 26 | NUM_GPUS: 8 27 | OUT_DIR: . 28 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-200MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | SE_ON: True 6 | DEPTH: 13 7 | W0: 24 8 | WA: 36.44 9 | WM: 2.49 10 | GROUP_W: 8 11 | OPTIM: 12 | LR_POLICY: cos 13 | BASE_LR: 0.8 14 | MAX_EPOCH: 100 15 | MOMENTUM: 0.9 16 | WEIGHT_DECAY: 5e-5 17 | WARMUP_EPOCHS: 5 18 | TRAIN: 19 | DATASET: imagenet 20 | IM_SIZE: 224 21 | BATCH_SIZE: 1024 22 | TEST: 23 | DATASET: imagenet 24 | IM_SIZE: 256 25 | BATCH_SIZE: 800 26 | NUM_GPUS: 8 27 | OUT_DIR: . 28 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-3.2GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | SE_ON: True 6 | DEPTH: 21 7 | W0: 80 8 | WA: 42.63 9 | WM: 2.66 10 | GROUP_W: 24 11 | OPTIM: 12 | LR_POLICY: cos 13 | BASE_LR: 0.4 14 | MAX_EPOCH: 100 15 | MOMENTUM: 0.9 16 | WEIGHT_DECAY: 5e-5 17 | WARMUP_EPOCHS: 5 18 | TRAIN: 19 | DATASET: imagenet 20 | IM_SIZE: 224 21 | BATCH_SIZE: 512 22 | TEST: 23 | DATASET: imagenet 24 | IM_SIZE: 256 25 | BATCH_SIZE: 400 26 | NUM_GPUS: 8 27 | OUT_DIR: . 28 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-32GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | SE_ON: True 6 | DEPTH: 20 7 | W0: 232 8 | WA: 115.89 9 | WM: 2.53 10 | GROUP_W: 232 11 | OPTIM: 12 | LR_POLICY: cos 13 | BASE_LR: 0.2 14 | MAX_EPOCH: 100 15 | MOMENTUM: 0.9 16 | WEIGHT_DECAY: 5e-5 17 | WARMUP_EPOCHS: 5 18 | TRAIN: 19 | DATASET: imagenet 20 | IM_SIZE: 224 21 | BATCH_SIZE: 256 22 | TEST: 23 | DATASET: imagenet 24 | IM_SIZE: 256 25 | BATCH_SIZE: 200 26 | NUM_GPUS: 8 27 | OUT_DIR: . 28 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-4.0GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | SE_ON: True 6 | DEPTH: 22 7 | W0: 96 8 | WA: 31.41 9 | WM: 2.24 10 | GROUP_W: 64 11 | OPTIM: 12 | LR_POLICY: cos 13 | BASE_LR: 0.4 14 | MAX_EPOCH: 100 15 | MOMENTUM: 0.9 16 | WEIGHT_DECAY: 5e-5 17 | WARMUP_EPOCHS: 5 18 | TRAIN: 19 | DATASET: imagenet 20 | IM_SIZE: 224 21 | BATCH_SIZE: 512 22 | TEST: 23 | DATASET: imagenet 24 | IM_SIZE: 256 25 | BATCH_SIZE: 400 26 | NUM_GPUS: 8 27 | OUT_DIR: . 28 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-400MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | SE_ON: True 6 | DEPTH: 16 7 | W0: 48 8 | WA: 27.89 9 | WM: 2.09 10 | GROUP_W: 8 11 | OPTIM: 12 | LR_POLICY: cos 13 | BASE_LR: 0.8 14 | MAX_EPOCH: 100 15 | MOMENTUM: 0.9 16 | WEIGHT_DECAY: 5e-5 17 | WARMUP_EPOCHS: 5 18 | TRAIN: 19 | DATASET: imagenet 20 | IM_SIZE: 224 21 | BATCH_SIZE: 1024 22 | TEST: 23 | DATASET: imagenet 24 | IM_SIZE: 256 25 | BATCH_SIZE: 800 26 | NUM_GPUS: 8 27 | OUT_DIR: . 28 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-6.4GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | SE_ON: True 6 | DEPTH: 25 7 | W0: 112 8 | WA: 33.22 9 | WM: 2.27 10 | GROUP_W: 72 11 | OPTIM: 12 | LR_POLICY: cos 13 | BASE_LR: 0.4 14 | MAX_EPOCH: 100 15 | MOMENTUM: 0.9 16 | WEIGHT_DECAY: 5e-5 17 | WARMUP_EPOCHS: 5 18 | TRAIN: 19 | DATASET: imagenet 20 | IM_SIZE: 224 21 | BATCH_SIZE: 512 22 | TEST: 23 | DATASET: imagenet 24 | IM_SIZE: 256 25 | BATCH_SIZE: 400 26 | NUM_GPUS: 8 27 | OUT_DIR: . 28 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-600MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | SE_ON: True 6 | DEPTH: 15 7 | W0: 48 8 | WA: 32.54 9 | WM: 2.32 10 | GROUP_W: 16 11 | OPTIM: 12 | LR_POLICY: cos 13 | BASE_LR: 0.8 14 | MAX_EPOCH: 100 15 | MOMENTUM: 0.9 16 | WEIGHT_DECAY: 5e-5 17 | WARMUP_EPOCHS: 5 18 | TRAIN: 19 | DATASET: imagenet 20 | IM_SIZE: 224 21 | BATCH_SIZE: 1024 22 | TEST: 23 | DATASET: imagenet 24 | IM_SIZE: 256 25 | BATCH_SIZE: 800 26 | NUM_GPUS: 8 27 | OUT_DIR: . 28 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-8.0GF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | SE_ON: true 6 | DEPTH: 17 7 | W0: 192 8 | WA: 76.82 9 | WM: 2.19 10 | GROUP_W: 56 11 | OPTIM: 12 | LR_POLICY: cos 13 | BASE_LR: 0.4 14 | MAX_EPOCH: 100 15 | MOMENTUM: 0.9 16 | WEIGHT_DECAY: 5e-5 17 | WARMUP_EPOCHS: 5 18 | TRAIN: 19 | DATASET: imagenet 20 | IM_SIZE: 224 21 | BATCH_SIZE: 512 22 | TEST: 23 | DATASET: imagenet 24 | IM_SIZE: 256 25 | BATCH_SIZE: 400 26 | NUM_GPUS: 8 27 | OUT_DIR: . 28 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/backbones/regnet/regnety/RegNetY-800MF_dds_8gpu.yaml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | TYPE: regnet 3 | NUM_CLASSES: 1000 4 | REGNET: 5 | SE_ON: True 6 | DEPTH: 14 7 | W0: 56 8 | WA: 38.84 9 | WM: 2.4 10 | GROUP_W: 16 11 | OPTIM: 12 | LR_POLICY: cos 13 | BASE_LR: 0.8 14 | MAX_EPOCH: 100 15 | MOMENTUM: 0.9 16 | WEIGHT_DECAY: 5e-5 17 | WARMUP_EPOCHS: 5 18 | TRAIN: 19 | DATASET: imagenet 20 | IM_SIZE: 224 21 | BATCH_SIZE: 1024 22 | TEST: 23 | DATASET: imagenet 24 | IM_SIZE: 256 25 | BATCH_SIZE: 800 26 | NUM_GPUS: 8 27 | OUT_DIR: . 28 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/heads/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from .build import REID_HEADS_REGISTRY, build_reid_heads 8 | 9 | # import all the meta_arch, so they will be registered 10 | from .linear_head import LinearHead 11 | from .bnneck_head import BNneckHead 12 | from .reduction_head import ReductionHead 13 | from .sigma_head import SigmaHead 14 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/heads/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/heads/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/heads/__pycache__/bnneck_head.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/heads/__pycache__/bnneck_head.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/heads/__pycache__/build.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/heads/__pycache__/build.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/heads/__pycache__/linear_head.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/heads/__pycache__/linear_head.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/heads/__pycache__/reduction_head.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/heads/__pycache__/reduction_head.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/heads/__pycache__/sigma_head.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/heads/__pycache__/sigma_head.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/heads/build.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from ...utils.registry import Registry 8 | 9 | REID_HEADS_REGISTRY = Registry("HEADS") 10 | REID_HEADS_REGISTRY.__doc__ = """ 11 | Registry for ROI heads in a generalized R-CNN model. 12 | ROIHeads take feature maps and region proposals, and 13 | perform per-region computation. 14 | The registered object will be called with `obj(cfg, input_shape)`. 15 | The call is expected to return an :class:`ROIHeads`. 16 | """ 17 | 18 | 19 | def build_reid_heads(cfg, in_feat, num_classes, pool_layer): 20 | """ 21 | Build REIDHeads defined by `cfg.MODEL.REID_HEADS.NAME`. 22 | """ 23 | head = cfg.MODEL.HEADS.NAME 24 | return REID_HEADS_REGISTRY.get(head)(cfg, in_feat, num_classes, pool_layer) 25 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/losses/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: l1aoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from .cross_entroy_loss import CrossEntropyLoss 8 | from .focal_loss import FocalLoss 9 | from .triplet_loss import TripletLoss 10 | from .circle_loss import CircleLoss 11 | from .npair_loss import NpairLoss 12 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/losses/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/losses/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/losses/__pycache__/circle_loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/losses/__pycache__/circle_loss.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/losses/__pycache__/cross_entroy_loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/losses/__pycache__/cross_entroy_loss.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/losses/__pycache__/focal_loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/losses/__pycache__/focal_loss.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/losses/__pycache__/npair_loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/losses/__pycache__/npair_loss.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/losses/__pycache__/triplet_loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/losses/__pycache__/triplet_loss.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/losses/__pycache__/utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/losses/__pycache__/utils.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/meta_arch/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from .build import META_ARCH_REGISTRY, build_model 8 | 9 | 10 | # import all the meta_arch, so they will be registered 11 | from .baseline import Baseline 12 | from .mgn import MGN 13 | -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/meta_arch/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/meta_arch/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/meta_arch/__pycache__/baseline.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/meta_arch/__pycache__/baseline.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/meta_arch/__pycache__/build.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/meta_arch/__pycache__/build.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/meta_arch/__pycache__/mgn.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/modeling/meta_arch/__pycache__/mgn.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/modeling/meta_arch/build.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | import torch 7 | 8 | from fastreid.utils.registry import Registry 9 | 10 | META_ARCH_REGISTRY = Registry("META_ARCH") # noqa F401 isort:skip 11 | META_ARCH_REGISTRY.__doc__ = """ 12 | Registry for meta-architectures, i.e. the whole model. 13 | The registered object will be called with `obj(cfg)` 14 | and expected to return a `nn.Module` object. 15 | """ 16 | 17 | 18 | def build_model(cfg): 19 | """ 20 | Build the whole model architecture, defined by ``cfg.MODEL.META_ARCHITECTURE``. 21 | Note that it does not load any weights from ``cfg``. 22 | """ 23 | meta_arch = cfg.MODEL.META_ARCHITECTURE 24 | model = META_ARCH_REGISTRY.get(meta_arch)(cfg) 25 | model.to(torch.device(cfg.MODEL.DEVICE)) 26 | return model 27 | -------------------------------------------------------------------------------- /fast-reid/fastreid/solver/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | 8 | from .build import build_lr_scheduler, build_optimizer -------------------------------------------------------------------------------- /fast-reid/fastreid/solver/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/solver/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/solver/__pycache__/build.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/solver/__pycache__/build.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/solver/__pycache__/lr_scheduler.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/solver/__pycache__/lr_scheduler.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/solver/optim/__init__.py: -------------------------------------------------------------------------------- 1 | from .lamb import Lamb 2 | from .swa import SWA 3 | from .adam import Adam 4 | from .sgd import SGD 5 | 6 | -------------------------------------------------------------------------------- /fast-reid/fastreid/solver/optim/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/solver/optim/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/solver/optim/__pycache__/adam.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/solver/optim/__pycache__/adam.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/solver/optim/__pycache__/lamb.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/solver/optim/__pycache__/lamb.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/solver/optim/__pycache__/sgd.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/solver/optim/__pycache__/sgd.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/solver/optim/__pycache__/swa.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/solver/optim/__pycache__/swa.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: sherlock 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | -------------------------------------------------------------------------------- /fast-reid/fastreid/utils/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/utils/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/utils/__pycache__/checkpoint.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/utils/__pycache__/checkpoint.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/utils/__pycache__/collect_env.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/utils/__pycache__/collect_env.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/utils/__pycache__/comm.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/utils/__pycache__/comm.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/utils/__pycache__/env.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/utils/__pycache__/env.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/utils/__pycache__/events.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/utils/__pycache__/events.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/utils/__pycache__/file_io.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/utils/__pycache__/file_io.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/utils/__pycache__/history_buffer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/utils/__pycache__/history_buffer.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/utils/__pycache__/logger.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/utils/__pycache__/logger.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/utils/__pycache__/precision_bn.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/utils/__pycache__/precision_bn.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/utils/__pycache__/registry.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/utils/__pycache__/registry.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/utils/__pycache__/timer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/utils/__pycache__/timer.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/utils/__pycache__/weight_init.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/fastreid/utils/__pycache__/weight_init.cpython-37.pyc -------------------------------------------------------------------------------- /fast-reid/fastreid/utils/weight_init.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: xingyu liao 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | import math 8 | from torch import nn 9 | 10 | __all__ = [ 11 | 'weights_init_classifier', 12 | 'weights_init_kaiming', 13 | ] 14 | 15 | 16 | def weights_init_kaiming(m): 17 | classname = m.__class__.__name__ 18 | if classname.find('Linear') != -1: 19 | nn.init.normal_(m.weight, 0, 0.01) 20 | if m.bias is not None: 21 | nn.init.constant_(m.bias, 0.0) 22 | elif classname.find('Conv') != -1: 23 | nn.init.kaiming_normal_(m.weight, mode='fan_out', nonlinearity='relu') 24 | if m.bias is not None: 25 | nn.init.constant_(m.bias, 0.0) 26 | elif classname.find('BatchNorm') != -1: 27 | if m.affine: 28 | nn.init.normal_(m.weight, 1.0, 0.02) 29 | nn.init.constant_(m.bias, 0.0) 30 | 31 | 32 | def weights_init_classifier(m): 33 | classname = m.__class__.__name__ 34 | if classname.find('Linear') != -1: 35 | nn.init.normal_(m.weight, std=0.001) 36 | if m.bias is not None: 37 | nn.init.constant_(m.bias, 0.0) 38 | -------------------------------------------------------------------------------- /fast-reid/projects/CrowdReID: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot18.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "18x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: False 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("DukeMTMC",) 48 | TESTS: ("DukeMTMC",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/dukemtmc/bot101ibn-kd-bot18" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "18x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("DukeMTMC",) 48 | TESTS: ("DukeMTMC",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/dukemtmc/bot101ibn-kd-bot18ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot34.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "34x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: False 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("DukeMTMC",) 48 | TESTS: ("DukeMTMC",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/dukemtmc/bot101ibn-kd-bot34" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot34ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "34x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("DukeMTMC",) 48 | TESTS: ("DukeMTMC",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/dukemtmc/bot101ibn-kd-bot34ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "50x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: False 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 2048 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("DukeMTMC",) 48 | TESTS: ("DukeMTMC",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/dukemtmc/bot101ibn-kd-bot50" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot50ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "50x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 2048 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("DukeMTMC",) 48 | TESTS: ("DukeMTMC",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/dukemtmc/bot101ibn-kd-bot50ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot34ibn-bot18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "34x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 512 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "18x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R34-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("DukeMTMC",) 48 | TESTS: ("DukeMTMC",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/dukemtmc/bot34ibn-kd-bot18ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50-bot18.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: False 15 | PRETRAIN: False 16 | 17 | HEADS: 18 | NAME: "ReductionHead" 19 | IN_FEAT: 2048 20 | REDUCTION_DIM: 256 21 | NORM: "BN" 22 | POOL_LAYER: "avgpool" 23 | NECK_FEAT: "before" 24 | CLS_LAYER: "" 25 | 26 | MODEL: 27 | META_ARCHITECTURE: "Baseline" 28 | 29 | BACKBONE: 30 | NAME: "build_resnet_backbone" 31 | NORM: "BN" 32 | DEPTH: "18x" 33 | LAST_STRIDE: 1 34 | WITH_IBN: False 35 | PRETRAIN: True 36 | PRETRAIN_PATH: "./model/ImageNet/resnet18-5c106cde.pth" 37 | 38 | HEADS: 39 | NAME: "ReductionHead" 40 | IN_FEAT: 512 41 | NORM: "BN" 42 | REDUCTION_DIM: 256 43 | POOL_LAYER: "avgpool" 44 | NECK_FEAT: "before" 45 | CLS_LAYER: "" 46 | 47 | STUDENT_WEIGHTS: "" 48 | TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R50_ReduceHead/teacher.pth" 49 | DATASETS: 50 | NAMES: ("DukeMTMC",) 51 | TESTS: ("DukeMTMC",) 52 | 53 | DATALOADER: 54 | NUM_WORKERS: 8 55 | 56 | SOLVER: 57 | BASE_LR: 0.00035 58 | 59 | OUTPUT_DIR: "logs/dukemtmc/bot50-bot18-l2" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50ibn-bot18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "18x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R50-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("DukeMTMC",) 48 | TESTS: ("DukeMTMC",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/dukemtmc/bot50ibn-kd-bot18ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50ibn-bot34ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "34x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R50-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("DukeMTMC",) 48 | TESTS: ("DukeMTMC",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/dukemtmc/bot50ibn-kd-bot34ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: '101x' 6 | WITH_IBN: True 7 | 8 | DATASETS: 9 | NAMES: ("DukeMTMC",) 10 | TESTS: ("DukeMTMC",) 11 | 12 | OUTPUT_DIR: "logs/dukemtmc/bagtricks_R101-ibn" 13 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R18-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "18x" 6 | WITH_IBN: True 7 | HEADS: 8 | IN_FEAT: 512 9 | 10 | DATASETS: 11 | NAMES: ("DukeMTMC",) 12 | TESTS: ("DukeMTMC",) 13 | 14 | OUTPUT_DIR: "logs/dukemtmc/bagtricks_R18-ibn" 15 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R34-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "34x" 6 | WITH_IBN: True 7 | HEADS: 8 | IN_FEAT: 512 9 | 10 | DATASETS: 11 | NAMES: ("DukeMTMC",) 12 | TESTS: ("DukeMTMC",) 13 | 14 | OUTPUT_DIR: "logs/dukemtmc/bagtricks_R34-ibn" 15 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "50x" 6 | WITH_IBN: True 7 | 8 | DATASETS: 9 | NAMES: ("DukeMTMC",) 10 | TESTS: ("DukeMTMC",) 11 | 12 | OUTPUT_DIR: "logs/dukemtmc/bagtricks_R50-ibn" 13 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot18.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "18x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: False 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("Market1501",) 48 | TESTS: ("Market1501",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/market1501/bot101ibn-kd-bot18" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "18x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("Market1501",) 48 | TESTS: ("Market1501",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/market1501/bot101ibn-kd-bot18ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot34.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "34x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: False 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("Market1501",) 48 | TESTS: ("Market1501",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/market1501/bot101ibn-kd-bot34" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot34ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "34x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("Market1501",) 48 | TESTS: ("Market1501",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/market1501/bot101ibn-kd-bot34ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "50x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: False 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 2048 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("Market1501",) 48 | TESTS: ("Market1501",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/market1501/bot101ibn-kd-bot50" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot50ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "50x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 2048 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("Market1501",) 48 | TESTS: ("Market1501",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/market1501/bot101ibn-kd-bot50ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot34ibn-bot18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "34x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 512 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "18x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R34-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("Market1501",) 48 | TESTS: ("Market1501",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/market1501/bot34ibn-kd-bot18ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot50-bot18.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: False 15 | PRETRAIN: False 16 | 17 | HEADS: 18 | NAME: "ReductionHead" 19 | IN_FEAT: 2048 20 | REDUCTION_DIM: 256 21 | NORM: "BN" 22 | POOL_LAYER: "avgpool" 23 | NECK_FEAT: "before" 24 | CLS_LAYER: "" 25 | 26 | MODEL: 27 | META_ARCHITECTURE: "Baseline" 28 | WEIGHTS: "./logs/market1501/bot50-bot18-l2/model_final.pth" 29 | 30 | BACKBONE: 31 | NAME: "build_resnet_backbone" 32 | NORM: "BN" 33 | DEPTH: "18x" 34 | LAST_STRIDE: 1 35 | WITH_IBN: False 36 | PRETRAIN: True 37 | PRETRAIN_PATH: "./model/ImageNet/resnet18-5c106cde.pth" 38 | 39 | HEADS: 40 | NAME: "ReductionHead" 41 | IN_FEAT: 512 42 | NORM: "BN" 43 | REDUCTION_DIM: 256 44 | POOL_LAYER: "avgpool" 45 | NECK_FEAT: "before" 46 | CLS_LAYER: "" 47 | 48 | STUDENT_WEIGHTS: "" 49 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R50_ReduceHead/teacher.pth" 50 | DATASETS: 51 | NAMES: ("Market1501",) 52 | TESTS: ("Market1501",) 53 | 54 | DATALOADER: 55 | NUM_WORKERS: 8 56 | NUM_INSTANCE: 8 57 | 58 | SOLVER: 59 | BASE_LR: 3.5e-5 60 | IMS_PER_BATCH: 128 61 | MAX_ITER: 240 62 | STEPS: [80, 180] 63 | 64 | OUTPUT_DIR: "logs/market1501/bot50-bot18-l2-fdm" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot50ibn-bot18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "18x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R50-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("Market1501",) 48 | TESTS: ("Market1501",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/market1501/bot50ibn-kd-bot18ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot50ibn-bot34ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "34x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R50-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("Market1501",) 48 | TESTS: ("Market1501",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/market1501/bot50ibn-kd-bot34ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: '101x' 6 | WITH_IBN: True 7 | 8 | DATASETS: 9 | NAMES: ("Market1501",) 10 | TESTS: ("Market1501",) 11 | 12 | OUTPUT_DIR: "logs/market1501/bagtricks_R101-ibn" 13 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R18-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "18x" 6 | WITH_IBN: True 7 | HEADS: 8 | IN_FEAT: 512 9 | 10 | DATASETS: 11 | NAMES: ("Market1501",) 12 | TESTS: ("Market1501",) 13 | 14 | OUTPUT_DIR: "logs/market1501/bagtricks_R18-ibn" 15 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R34-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "34x" 6 | WITH_IBN: True 7 | HEADS: 8 | IN_FEAT: 512 9 | 10 | DATASETS: 11 | NAMES: ("Market1501",) 12 | TESTS: ("Market1501",) 13 | 14 | OUTPUT_DIR: "logs/market1501/bagtricks_R34-ibn" 15 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "50x" 6 | WITH_IBN: True 7 | 8 | DATASETS: 9 | NAMES: ("Market1501",) 10 | TESTS: ("Market1501",) 11 | 12 | OUTPUT_DIR: "logs/market1501/bagtricks_R50-ibn" 13 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot18.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "18x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: False 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("MSMT17",) 48 | TESTS: ("MSMT17",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/msmt17/bot101ibn-kd-bot18" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "18x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("MSMT17",) 48 | TESTS: ("MSMT17",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/msmt17/bot101ibn-kd-bot18ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot34.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "34x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: False 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("MSMT17",) 48 | TESTS: ("MSMT17",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/msmt17/bot101ibn-kd-bot34" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot34ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "34x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("MSMT17",) 48 | TESTS: ("MSMT17",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/msmt17/bot101ibn-kd-bot34ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot50.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "50x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: False 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 2048 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("MSMT17",) 48 | TESTS: ("MSMT17",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/msmt17/bot101ibn-kd-bot50" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot50ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "50x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 2048 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("MSMT17",) 48 | TESTS: ("MSMT17",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/msmt17/bot101ibn-kd-bot50ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot34ibn-bot18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "34x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 512 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "18x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R34-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("MSMT17",) 48 | TESTS: ("MSMT17",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/msmt17/bot34ibn-kd-bot18ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot50-bot18.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: False 15 | PRETRAIN: False 16 | 17 | HEADS: 18 | NAME: "ReductionHead" 19 | IN_FEAT: 2048 20 | REDUCTION_DIM: 256 21 | NORM: "BN" 22 | POOL_LAYER: "avgpool" 23 | NECK_FEAT: "before" 24 | CLS_LAYER: "" 25 | 26 | MODEL: 27 | META_ARCHITECTURE: "Baseline" 28 | 29 | BACKBONE: 30 | NAME: "build_resnet_backbone" 31 | NORM: "BN" 32 | DEPTH: "18x" 33 | LAST_STRIDE: 1 34 | WITH_IBN: False 35 | PRETRAIN: True 36 | PRETRAIN_PATH: "./model/ImageNet/resnet18-5c106cde.pth" 37 | 38 | HEADS: 39 | NAME: "ReductionHead" 40 | IN_FEAT: 512 41 | NORM: "BN" 42 | REDUCTION_DIM: 256 43 | POOL_LAYER: "avgpool" 44 | NECK_FEAT: "before" 45 | CLS_LAYER: "" 46 | 47 | STUDENT_WEIGHTS: "" 48 | TEACHER_WEIGHTS: "./logs/msmt17/bagtricks_R50_ReduceHead/teacher.pth" 49 | DATASETS: 50 | NAMES: ("MSMT17",) 51 | TESTS: ("MSMT17",) 52 | 53 | DATALOADER: 54 | NUM_WORKERS: 8 55 | 56 | SOLVER: 57 | BASE_LR: 0.00035 58 | 59 | OUTPUT_DIR: "logs/msmt17/bot50-bot18-fdm" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot50ibn-bot18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "18x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R50-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("MSMT17",) 48 | TESTS: ("MSMT17",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/msmt17/bot50ibn-kd-bot18ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot50ibn-bot34ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: True 15 | PRETRAIN: True 16 | 17 | HEADS: 18 | NAME: "BNneckHead" 19 | IN_FEAT: 2048 20 | NORM: "BN" 21 | POOL_LAYER: "avgpool" 22 | NECK_FEAT: "before" 23 | CLS_LAYER: "linear" 24 | 25 | MODEL: 26 | META_ARCHITECTURE: "Baseline" 27 | 28 | BACKBONE: 29 | NAME: "build_resnet_backbone" 30 | NORM: "BN" 31 | DEPTH: "34x" 32 | LAST_STRIDE: 1 33 | WITH_IBN: True 34 | PRETRAIN: True 35 | 36 | HEADS: 37 | NAME: "BNneckHead" 38 | IN_FEAT: 512 39 | NORM: "BN" 40 | POOL_LAYER: "avgpool" 41 | NECK_FEAT: "before" 42 | CLS_LAYER: "linear" 43 | 44 | STUDENT_WEIGHTS: "" 45 | TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R50-ibn/model_final.pth" 46 | DATASETS: 47 | NAMES: ("MSMT17",) 48 | TESTS: ("MSMT17",) 49 | 50 | DATALOADER: 51 | NUM_WORKERS: 8 52 | 53 | OUTPUT_DIR: "logs/msmt17/bot50ibn-kd-bot34ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: '101x' 6 | WITH_IBN: True 7 | 8 | DATASETS: 9 | NAMES: ("MSMT17",) 10 | TESTS: ("MSMT17",) 11 | 12 | OUTPUT_DIR: "logs/msmt17/bagtricks_R101-ibn" 13 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R18-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "18x" 6 | WITH_IBN: True 7 | HEADS: 8 | IN_FEAT: 512 9 | 10 | DATASETS: 11 | NAMES: ("MSMT17",) 12 | TESTS: ("MSMT17",) 13 | 14 | OUTPUT_DIR: "logs/msmt17/bagtricks_R18-ibn" 15 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R34-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "34x" 6 | WITH_IBN: True 7 | HEADS: 8 | IN_FEAT: 512 9 | 10 | DATASETS: 11 | NAMES: ("MSMT17",) 12 | TESTS: ("MSMT17",) 13 | 14 | OUTPUT_DIR: "logs/msmt17/bagtricks_R34-ibn" 15 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-bagtricks.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "50x" 6 | WITH_IBN: True 7 | 8 | DATASETS: 9 | NAMES: ("MSMT17",) 10 | TESTS: ("MSMT17",) 11 | 12 | OUTPUT_DIR: "logs/msmt17/bagtricks_R50-ibn" 13 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 2048 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | MODEL: 29 | META_ARCHITECTURE: "Baseline" 30 | 31 | BACKBONE: 32 | WITH_NL: False 33 | DEPTH: "18x" 34 | WITH_IBN: True 35 | HEADS: 36 | IN_FEAT: 512 37 | CLS_LAYER: "circleSoftmax" 38 | 39 | STUDENT_WEIGHTS: "" 40 | TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R101-ibn/model_final.pth" 41 | 42 | 43 | DATASETS: 44 | NAMES: ("DukeMTMC",) 45 | TESTS: ("DukeMTMC",) 46 | 47 | DATALOADER: 48 | NUM_WORKERS: 8 49 | 50 | OUTPUT_DIR: "logs/dukemtmcreid/sbs101ibn-kd-sbs18ibn" 51 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs34ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 2048 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | 29 | MODEL: 30 | META_ARCHITECTURE: "Baseline" 31 | 32 | BACKBONE: 33 | WITH_NL: False 34 | DEPTH: "34x" 35 | WITH_IBN: True 36 | HEADS: 37 | IN_FEAT: 512 38 | CLS_LAYER: "circleSoftmax" 39 | 40 | STUDENT_WEIGHTS: "" 41 | TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R101-ibn/model_final.pth" 42 | 43 | 44 | DATASETS: 45 | NAMES: ("DukeMTMC",) 46 | TESTS: ("DukeMTMC",) 47 | 48 | DATALOADER: 49 | NUM_WORKERS: 8 50 | 51 | OUTPUT_DIR: "logs/dukemtmcreid/sbs101ibn-kd-sbs34ibn" 52 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs50ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 2048 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | 29 | MODEL: 30 | META_ARCHITECTURE: "Baseline" 31 | 32 | BACKBONE: 33 | WITH_NL: False 34 | DEPTH: "50x" 35 | WITH_IBN: True 36 | HEADS: 37 | IN_FEAT: 2048 38 | CLS_LAYER: "circleSoftmax" 39 | 40 | STUDENT_WEIGHTS: "" 41 | TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R101-ibn/model_final.pth" 42 | 43 | 44 | DATASETS: 45 | NAMES: ("DukeMTMC",) 46 | TESTS: ("DukeMTMC",) 47 | 48 | DATALOADER: 49 | NUM_WORKERS: 8 50 | 51 | OUTPUT_DIR: "logs/dukemtmcreid/sbs101ibn-kd-sbs50ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs34ibn-sbs18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "34x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 512 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | MODEL: 29 | META_ARCHITECTURE: "Baseline" 30 | 31 | BACKBONE: 32 | WITH_NL: False 33 | DEPTH: "18x" 34 | WITH_IBN: True 35 | HEADS: 36 | IN_FEAT: 512 37 | CLS_LAYER: "circleSoftmax" 38 | 39 | STUDENT_WEIGHTS: "" 40 | TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R34-ibn/model_final.pth" 41 | 42 | 43 | DATASETS: 44 | NAMES: ("DukeMTMC",) 45 | TESTS: ("DukeMTMC",) 46 | 47 | DATALOADER: 48 | NUM_WORKERS: 8 49 | 50 | OUTPUT_DIR: "logs/dukemtmcreid/sbs34ibn-kd-sbs18ibn" 51 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs50ibn-sbs18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 2048 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | MODEL: 29 | META_ARCHITECTURE: "Baseline" 30 | 31 | BACKBONE: 32 | WITH_NL: False 33 | DEPTH: "18x" 34 | WITH_IBN: True 35 | HEADS: 36 | IN_FEAT: 512 37 | CLS_LAYER: "circleSoftmax" 38 | 39 | STUDENT_WEIGHTS: "" 40 | TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R50-ibn/model_final.pth" 41 | 42 | 43 | DATASETS: 44 | NAMES: ("DukeMTMC",) 45 | TESTS: ("DukeMTMC",) 46 | 47 | DATALOADER: 48 | NUM_WORKERS: 8 49 | 50 | OUTPUT_DIR: "logs/dukemtmcreid/sbs50ibn-kd-sbs18ibn" 51 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs50ibn-sbs34ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 2048 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | 29 | MODEL: 30 | META_ARCHITECTURE: "Baseline" 31 | 32 | BACKBONE: 33 | WITH_NL: False 34 | DEPTH: "34x" 35 | WITH_IBN: True 36 | HEADS: 37 | IN_FEAT: 512 38 | CLS_LAYER: "circleSoftmax" 39 | 40 | STUDENT_WEIGHTS: "" 41 | TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R50-ibn/model_final.pth" 42 | 43 | 44 | DATASETS: 45 | NAMES: ("DukeMTMC",) 46 | TESTS: ("DukeMTMC",) 47 | 48 | DATALOADER: 49 | NUM_WORKERS: 8 50 | 51 | OUTPUT_DIR: "logs/dukemtmcreid/sbs50ibn-kd-sbs34ibn" 52 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "101x" 6 | WITH_IBN: True 7 | HEADS: 8 | CLS_LAYER: "circleSoftmax" 9 | 10 | DATASETS: 11 | NAMES: ("DukeMTMC",) 12 | TESTS: ("DukeMTMC",) 13 | 14 | OUTPUT_DIR: "logs/dukemtmcreid/sbs_R101-ibn" 15 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R18-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_NL: False 6 | DEPTH: "18x" 7 | WITH_IBN: True 8 | HEADS: 9 | IN_FEAT: 512 10 | CLS_LAYER: "circleSoftmax" 11 | 12 | DATASETS: 13 | NAMES: ("DukeMTMC",) 14 | TESTS: ("DukeMTMC",) 15 | 16 | OUTPUT_DIR: "logs/dukemtmcreid/sbs_R18-ibn" 17 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R34-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_NL: False 6 | DEPTH: "34x" 7 | WITH_IBN: True 8 | HEADS: 9 | IN_FEAT: 512 10 | CLS_LAYER: "circleSoftmax" 11 | 12 | DATASETS: 13 | NAMES: ("DukeMTMC",) 14 | TESTS: ("DukeMTMC",) 15 | 16 | OUTPUT_DIR: "logs/dukemtmcreid/sbs_R34-ibn" 17 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_NL: False 6 | DEPTH: "50x" 7 | WITH_IBN: True 8 | HEADS: 9 | IN_FEAT: 2048 10 | CLS_LAYER: "circleSoftmax" 11 | 12 | DATASETS: 13 | NAMES: ("DukeMTMC",) 14 | TESTS: ("DukeMTMC",) 15 | 16 | OUTPUT_DIR: "logs/dukemtmcreid/sbs_R50-ibn" 17 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 2048 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | MODEL: 29 | META_ARCHITECTURE: "Baseline" 30 | 31 | BACKBONE: 32 | WITH_NL: False 33 | DEPTH: "18x" 34 | WITH_IBN: True 35 | HEADS: 36 | IN_FEAT: 512 37 | CLS_LAYER: "circleSoftmax" 38 | 39 | STUDENT_WEIGHTS: "" 40 | TEACHER_WEIGHTS: "./logs/market1501/sbs_R101-ibn/model_final.pth" 41 | 42 | 43 | DATASETS: 44 | NAMES: ("Market1501",) 45 | TESTS: ("Market1501",) 46 | 47 | DATALOADER: 48 | NUM_WORKERS: 8 49 | 50 | OUTPUT_DIR: "logs/market1501/sbs101ibn-kd-sbs18ibn" 51 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs34ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 2048 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | 29 | MODEL: 30 | META_ARCHITECTURE: "Baseline" 31 | 32 | BACKBONE: 33 | WITH_NL: False 34 | DEPTH: "34x" 35 | WITH_IBN: True 36 | HEADS: 37 | IN_FEAT: 512 38 | CLS_LAYER: "circleSoftmax" 39 | 40 | STUDENT_WEIGHTS: "" 41 | TEACHER_WEIGHTS: "./logs/market1501/sbs_R101-ibn/model_final.pth" 42 | 43 | 44 | DATASETS: 45 | NAMES: ("Market1501",) 46 | TESTS: ("Market1501",) 47 | 48 | DATALOADER: 49 | NUM_WORKERS: 8 50 | 51 | OUTPUT_DIR: "logs/market1501/sbs101ibn-kd-sbs34ibn" 52 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs50ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 2048 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | 29 | MODEL: 30 | META_ARCHITECTURE: "Baseline" 31 | 32 | BACKBONE: 33 | WITH_NL: False 34 | DEPTH: "50x" 35 | WITH_IBN: True 36 | HEADS: 37 | IN_FEAT: 2048 38 | CLS_LAYER: "circleSoftmax" 39 | 40 | STUDENT_WEIGHTS: "" 41 | TEACHER_WEIGHTS: "./logs/market1501/sbs_R101-ibn/model_final.pth" 42 | 43 | 44 | DATASETS: 45 | NAMES: ("Market1501",) 46 | TESTS: ("Market1501",) 47 | 48 | DATALOADER: 49 | NUM_WORKERS: 8 50 | 51 | OUTPUT_DIR: "logs/market1501/sbs101ibn-kd-sbs50ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs34ibn-sbs18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "34x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 512 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | MODEL: 29 | META_ARCHITECTURE: "Baseline" 30 | 31 | BACKBONE: 32 | WITH_NL: False 33 | DEPTH: "18x" 34 | WITH_IBN: True 35 | HEADS: 36 | IN_FEAT: 512 37 | CLS_LAYER: "circleSoftmax" 38 | 39 | STUDENT_WEIGHTS: "" 40 | TEACHER_WEIGHTS: "./logs/market1501/sbs_R34-ibn/model_final.pth" 41 | 42 | 43 | DATASETS: 44 | NAMES: ("Market1501",) 45 | TESTS: ("Market1501",) 46 | 47 | DATALOADER: 48 | NUM_WORKERS: 8 49 | 50 | OUTPUT_DIR: "logs/market1501/sbs34ibn-kd-sbs18ibn" 51 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs50ibn-sbs18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 2048 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | MODEL: 29 | META_ARCHITECTURE: "Baseline" 30 | 31 | BACKBONE: 32 | WITH_NL: False 33 | DEPTH: "18x" 34 | WITH_IBN: True 35 | HEADS: 36 | IN_FEAT: 512 37 | CLS_LAYER: "circleSoftmax" 38 | 39 | STUDENT_WEIGHTS: "" 40 | TEACHER_WEIGHTS: "./logs/market1501/sbs_R50-ibn/model_final.pth" 41 | 42 | 43 | DATASETS: 44 | NAMES: ("Market1501",) 45 | TESTS: ("Market1501",) 46 | 47 | DATALOADER: 48 | NUM_WORKERS: 8 49 | 50 | OUTPUT_DIR: "logs/market1501/sbs50ibn-kd-sbs18ibn" 51 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs50ibn-sbs34ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 2048 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | 29 | MODEL: 30 | META_ARCHITECTURE: "Baseline" 31 | 32 | BACKBONE: 33 | WITH_NL: False 34 | DEPTH: "34x" 35 | WITH_IBN: True 36 | HEADS: 37 | IN_FEAT: 512 38 | CLS_LAYER: "circleSoftmax" 39 | 40 | STUDENT_WEIGHTS: "" 41 | TEACHER_WEIGHTS: "./logs/market1501/sbs_R50-ibn/model_final.pth" 42 | 43 | 44 | DATASETS: 45 | NAMES: ("Market1501",) 46 | TESTS: ("Market1501",) 47 | 48 | DATALOADER: 49 | NUM_WORKERS: 8 50 | 51 | OUTPUT_DIR: "logs/market1501/sbs50ibn-kd-sbs34ibn" 52 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "101x" 6 | WITH_IBN: True 7 | HEADS: 8 | CLS_LAYER: "circleSoftmax" 9 | 10 | DATASETS: 11 | NAMES: ("Market1501",) 12 | TESTS: ("Market1501",) 13 | 14 | OUTPUT_DIR: "logs/market1501/sbs_R101-ibn" 15 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R18-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_NL: False 6 | DEPTH: "18x" 7 | WITH_IBN: True 8 | HEADS: 9 | IN_FEAT: 512 10 | CLS_LAYER: "circleSoftmax" 11 | 12 | DATASETS: 13 | NAMES: ("Market1501",) 14 | TESTS: ("Market1501",) 15 | 16 | OUTPUT_DIR: "logs/market1501/sbs_R18-ibn" 17 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R34-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_NL: False 6 | DEPTH: "34x" 7 | WITH_IBN: True 8 | HEADS: 9 | IN_FEAT: 512 10 | CLS_LAYER: "circleSoftmax" 11 | 12 | DATASETS: 13 | NAMES: ("Market1501",) 14 | TESTS: ("Market1501",) 15 | 16 | OUTPUT_DIR: "logs/market1501/sbs_R34-ibn" 17 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_NL: False 6 | DEPTH: "50x" 7 | WITH_IBN: True 8 | HEADS: 9 | IN_FEAT: 2048 10 | CLS_LAYER: "circleSoftmax" 11 | 12 | DATASETS: 13 | NAMES: ("Market1501",) 14 | TESTS: ("Market1501",) 15 | 16 | OUTPUT_DIR: "logs/market1501/sbs_R50-ibn" 17 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-msmt17/KD-bot50-bot18.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_IBN: False 15 | WITH_NL: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "ReductionHead" 20 | IN_FEAT: 2048 21 | REDUCTION_DIM: 256 22 | NORM: "BN" 23 | POOL_LAYER: "gempool" 24 | NECK_FEAT: "after" 25 | CLS_LAYER: "" 26 | 27 | MODEL: 28 | META_ARCHITECTURE: "Baseline" 29 | WEIGHTS: "logs/msmt17/bot50-bot18-l2/model_final.pth" 30 | 31 | BACKBONE: 32 | NAME: "build_resnet_backbone" 33 | NORM: "BN" 34 | DEPTH: "18x" 35 | LAST_STRIDE: 1 36 | WITH_IBN: False 37 | WITH_NL: True 38 | PRETRAIN: True 39 | PRETRAIN_PATH: "./model/ImageNet/resnet18-5c106cde.pth" 40 | 41 | HEADS: 42 | NAME: "ReductionHead" 43 | IN_FEAT: 512 44 | NORM: "BN" 45 | REDUCTION_DIM: 256 46 | POOL_LAYER: "gempool" 47 | NECK_FEAT: "after" 48 | CLS_LAYER: "" 49 | 50 | STUDENT_WEIGHTS: "" 51 | TEACHER_WEIGHTS: "./logs/msmt17/sbs_R50_ReductionHead/teacher.pth" 52 | DATASETS: 53 | NAMES: ("MSMT17",) 54 | TESTS: ("MSMT17",) 55 | 56 | DATALOADER: 57 | NUM_WORKERS: 8 58 | NUM_INSTANCE: 8 59 | 60 | SOLVER: 61 | BASE_LR: 0.000035 62 | # IMS_PER_BATCH: 128 63 | # MAX_ITER: 240 64 | # STEPS: [80, 180] 65 | 66 | OUTPUT_DIR: "logs/msmt17/bot50-bot18-fdm" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 2048 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | MODEL: 29 | META_ARCHITECTURE: "Baseline" 30 | 31 | BACKBONE: 32 | WITH_NL: False 33 | DEPTH: "18x" 34 | WITH_IBN: True 35 | HEADS: 36 | IN_FEAT: 512 37 | CLS_LAYER: "circleSoftmax" 38 | 39 | STUDENT_WEIGHTS: "" 40 | TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R101-ibn/model_final.pth" 41 | 42 | 43 | DATASETS: 44 | NAMES: ("MSMT17",) 45 | TESTS: ("MSMT17",) 46 | 47 | DATALOADER: 48 | NUM_WORKERS: 8 49 | 50 | OUTPUT_DIR: "logs/dukemtmcreid/sbs101ibn-kd-sbs18ibn" 51 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs34ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 2048 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | 29 | MODEL: 30 | META_ARCHITECTURE: "Baseline" 31 | 32 | BACKBONE: 33 | WITH_NL: False 34 | DEPTH: "34x" 35 | WITH_IBN: True 36 | HEADS: 37 | IN_FEAT: 512 38 | CLS_LAYER: "circleSoftmax" 39 | 40 | STUDENT_WEIGHTS: "" 41 | TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R101-ibn/model_final.pth" 42 | 43 | 44 | DATASETS: 45 | NAMES: ("MSMT17",) 46 | TESTS: ("MSMT17",) 47 | 48 | DATALOADER: 49 | NUM_WORKERS: 8 50 | 51 | OUTPUT_DIR: "logs/dukemtmcreid/sbs101ibn-kd-sbs34ibn" 52 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs50ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "101x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 2048 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | 29 | MODEL: 30 | META_ARCHITECTURE: "Baseline" 31 | 32 | BACKBONE: 33 | WITH_NL: False 34 | DEPTH: "50x" 35 | WITH_IBN: True 36 | HEADS: 37 | IN_FEAT: 2048 38 | CLS_LAYER: "circleSoftmax" 39 | 40 | STUDENT_WEIGHTS: "" 41 | TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R101-ibn/model_final.pth" 42 | 43 | 44 | DATASETS: 45 | NAMES: ("MSMT17",) 46 | TESTS: ("MSMT17",) 47 | 48 | DATALOADER: 49 | NUM_WORKERS: 8 50 | 51 | OUTPUT_DIR: "logs/dukemtmcreid/sbs101ibn-kd-sbs50ibn" -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs34ibn-sbs18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "34x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 512 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | MODEL: 29 | META_ARCHITECTURE: "Baseline" 30 | 31 | BACKBONE: 32 | WITH_NL: False 33 | DEPTH: "18x" 34 | WITH_IBN: True 35 | HEADS: 36 | IN_FEAT: 512 37 | CLS_LAYER: "circleSoftmax" 38 | 39 | STUDENT_WEIGHTS: "" 40 | TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R34-ibn/model_final.pth" 41 | 42 | 43 | DATASETS: 44 | NAMES: ("MSMT17",) 45 | TESTS: ("MSMT17",) 46 | 47 | DATALOADER: 48 | NUM_WORKERS: 8 49 | 50 | OUTPUT_DIR: "logs/dukemtmcreid/sbs34ibn-kd-sbs18ibn" 51 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs50ibn-sbs18ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 2048 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | MODEL: 29 | META_ARCHITECTURE: "Baseline" 30 | 31 | BACKBONE: 32 | WITH_NL: False 33 | DEPTH: "18x" 34 | WITH_IBN: True 35 | HEADS: 36 | IN_FEAT: 512 37 | CLS_LAYER: "circleSoftmax" 38 | 39 | STUDENT_WEIGHTS: "" 40 | TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R50-ibn/model_final.pth" 41 | 42 | 43 | DATASETS: 44 | NAMES: ("MSMT17",) 45 | TESTS: ("MSMT17",) 46 | 47 | DATALOADER: 48 | NUM_WORKERS: 8 49 | 50 | OUTPUT_DIR: "logs/dukemtmcreid/sbs50ibn-kd-sbs18ibn" 51 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs50ibn-sbs34ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | SOLVER: 4 | IMS_PER_BATCH: 64 5 | 6 | MODEL_TEACHER: 7 | META_ARCHITECTURE: "Baseline" 8 | 9 | BACKBONE: 10 | NAME: "build_resnet_backbone" 11 | NORM: "BN" 12 | DEPTH: "50x" 13 | LAST_STRIDE: 1 14 | WITH_NL: True 15 | WITH_IBN: True 16 | PRETRAIN: False 17 | 18 | HEADS: 19 | NAME: "BNneckHead" 20 | IN_FEAT: 2048 21 | NORM: "BN" 22 | NECK_FEAT: "after" 23 | POOL_LAYER: "gempool" 24 | CLS_LAYER: "circle" 25 | SCALE: 64 26 | MARGIN: 0.35 27 | 28 | 29 | MODEL: 30 | META_ARCHITECTURE: "Baseline" 31 | 32 | BACKBONE: 33 | WITH_NL: False 34 | DEPTH: "34x" 35 | WITH_IBN: True 36 | HEADS: 37 | IN_FEAT: 512 38 | CLS_LAYER: "circleSoftmax" 39 | 40 | STUDENT_WEIGHTS: "" 41 | TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R50-ibn/model_final.pth" 42 | 43 | 44 | DATASETS: 45 | NAMES: ("MSMT17",) 46 | TESTS: ("MSMT17",) 47 | 48 | DATALOADER: 49 | NUM_WORKERS: 8 50 | 51 | OUTPUT_DIR: "logs/dukemtmcreid/sbs50ibn-kd-sbs34ibn" 52 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R101-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | DEPTH: "101x" 6 | WITH_IBN: True 7 | HEADS: 8 | CLS_LAYER: "circleSoftmax" 9 | 10 | DATASETS: 11 | NAMES: ("MSMT17",) 12 | TESTS: ("MSMT17",) 13 | 14 | OUTPUT_DIR: "logs/dukemtmcreid/sbs_R101-ibn" 15 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R18-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_NL: False 6 | DEPTH: "18x" 7 | WITH_IBN: True 8 | HEADS: 9 | IN_FEAT: 512 10 | CLS_LAYER: "circleSoftmax" 11 | 12 | DATASETS: 13 | NAMES: ("MSMT17",) 14 | TESTS: ("MSMT17",) 15 | 16 | OUTPUT_DIR: "logs/dukemtmcreid/sbs_R18-ibn" 17 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R34-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_NL: False 6 | DEPTH: "34x" 7 | WITH_IBN: True 8 | HEADS: 9 | IN_FEAT: 512 10 | CLS_LAYER: "circleSoftmax" 11 | 12 | DATASETS: 13 | NAMES: ("MSMT17",) 14 | TESTS: ("MSMT17",) 15 | 16 | OUTPUT_DIR: "logs/dukemtmcreid/sbs_R34-ibn" 17 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R50-ibn.yml: -------------------------------------------------------------------------------- 1 | _BASE_: "../../../configs/Base-Strongerbaseline.yml" 2 | 3 | MODEL: 4 | BACKBONE: 5 | WITH_NL: False 6 | DEPTH: "50x" 7 | WITH_IBN: True 8 | HEADS: 9 | IN_FEAT: 2048 10 | CLS_LAYER: "circleSoftmax" 11 | 12 | DATASETS: 13 | NAMES: ("MSMT17",) 14 | TESTS: ("MSMT17",) 15 | 16 | OUTPUT_DIR: "logs/dukemtmcreid/sbs_R50-ibn" 17 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/kdreid/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: l1aoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from .config import add_kdreid_config, add_shufflenet_config 8 | from .kd_trainer import KDTrainer 9 | from .modeling import build_shufflenetv2_backbone -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/kdreid/modeling/__init__.py: -------------------------------------------------------------------------------- 1 | from .backbones import build_shufflenetv2_backbone -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/kdreid/modeling/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | from .shufflenetv2 import build_shufflenetv2_backbone -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/kdreid/modeling/backbones/shufflenetv2/__init__.py: -------------------------------------------------------------------------------- 1 | import torch 2 | import torch.nn as nn 3 | from collections import OrderedDict 4 | 5 | from fastreid.modeling.backbones.build import BACKBONE_REGISTRY 6 | from .network import ShuffleNetV2 7 | 8 | 9 | __all__ = ['build_shufflenetv2_backbone'] 10 | 11 | 12 | @BACKBONE_REGISTRY.register() 13 | def build_shufflenetv2_backbone(cfg): 14 | 15 | pretrain = cfg.MODEL.BACKBONE.PRETRAIN 16 | pretrain_path = cfg.MODEL.BACKBONE.PRETRAIN_PATH 17 | model_size = cfg.MODEL.BACKBONE.MODEL_SIZE 18 | 19 | return ShuffleNetV2Backbone(model_size=model_size, pretrained=pretrain, pretrain_path=pretrain_path) 20 | 21 | 22 | class ShuffleNetV2Backbone(nn.Module): 23 | 24 | def __init__(self, model_size, pretrained=False, pretrain_path=''): 25 | super(ShuffleNetV2Backbone, self).__init__() 26 | 27 | model = ShuffleNetV2(model_size=model_size) 28 | if pretrained: 29 | new_state_dict = OrderedDict() 30 | state_dict = torch.load(pretrain_path)['state_dict'] 31 | for k, v in state_dict.items(): 32 | if k[:7] == 'module.': 33 | k = k[7:] 34 | new_state_dict[k] = v 35 | model.load_state_dict(new_state_dict, strict=True) 36 | 37 | self.backbone = nn.Sequential( 38 | model.first_conv, model.maxpool, model.features, model.conv_last) 39 | 40 | def forward(self, x): 41 | return self.backbone(x) 42 | 43 | 44 | -------------------------------------------------------------------------------- /fast-reid/projects/DistillReID/kdreid/modeling/sigma.py: -------------------------------------------------------------------------------- 1 | ''' 2 | Branch to estimate uncertainty of samples 3 | -------------------------------------- 4 | Author: Yang Qian 5 | Email: yqian@aibee.com 6 | ''' 7 | import torch.nn as nn 8 | import torch 9 | 10 | class Sigma(nn.Module): 11 | def __init__(self, n_input, n_output): 12 | super().__init__() 13 | self.model = nn.Sequential( 14 | nn.Linear(n_input, n_input), 15 | nn.LeakyReLU(inplace=True), 16 | nn.Linear(n_input, n_output), 17 | nn.BatchNorm1d(n_output) 18 | ) 19 | 20 | for m in self.modules(): 21 | if isinstance(m, nn.Linear): 22 | nn.init.normal_(m.weight.data, 0.0, 1e-4) 23 | nn.init.constant_(m.bias.data, 0.0) 24 | 25 | def forward(self, s, t, l2_norm=True): 26 | if l2_norm: 27 | s = torch.nn.functional.normalize(s) 28 | t = torch.nn.functional.normalize(t) 29 | return self.model((s-t)*(s-t)) -------------------------------------------------------------------------------- /fast-reid/projects/NAIC/dataset_analysis.py: -------------------------------------------------------------------------------- 1 | ''' 2 | analysis color distribution of NAIC datasets 3 | --------------- 4 | Author: Yang Qian 5 | Email: yqian@aibee.com 6 | ''' 7 | 8 | import numpy as np 9 | import cv2 10 | import os 11 | 12 | if __name__ == "__main__": 13 | dataset_dir = 'D:\\document\\NAIC\\NAIC2019\\1\\train_set' 14 | 15 | im_names = os.listdir(dataset_dir) 16 | axis = [0, 1, 2] 17 | for im_name in im_names: 18 | im = cv2.imread(os.path.join(dataset_dir, im_name)) 19 | np.random.shuffle(axis) 20 | im_new = im[:, :, axis] 21 | cv2.imshow('original', im) 22 | cv2.imshow('new', im_new) 23 | cv2.waitKey() -------------------------------------------------------------------------------- /fast-reid/projects/PartialReID/configs/partial_market.yml: -------------------------------------------------------------------------------- 1 | MODEL: 2 | META_ARCHITECTURE: 'PartialBaseline' 3 | 4 | BACKBONE: 5 | NAME: "build_resnet_backbone" 6 | DEPTH: "50x" 7 | NORM: "BN" 8 | LAST_STRIDE: 1 9 | WITH_IBN: True 10 | 11 | HEADS: 12 | NAME: "DSRHead" 13 | NORM: "BN" 14 | POOL_LAYER: "avgpool" 15 | NECK_FEAT: "before" 16 | CLS_LAYER: "linear" 17 | 18 | LOSSES: 19 | NAME: ("CrossEntropyLoss", "TripletLoss") 20 | CE: 21 | EPSILON: 0.1 22 | SCALE: 1. 23 | TRI: 24 | MARGIN: 0.3 25 | HARD_MINING: False 26 | SCALE: 1. 27 | 28 | DATASETS: 29 | NAMES: ("Market1501",) 30 | TESTS: ("PartialREID", "PartialiLIDS","OccludedREID",) 31 | 32 | INPUT: 33 | SIZE_TRAIN: [384, 128] 34 | SIZE_TEST: [384, 128] 35 | REA: 36 | ENABLED: False 37 | DO_PAD: False 38 | 39 | DATALOADER: 40 | PK_SAMPLER: True 41 | NAIVE_WAY: True 42 | NUM_INSTANCE: 4 43 | NUM_WORKERS: 8 44 | 45 | SOLVER: 46 | OPT: "Adam" 47 | MAX_ITER: 30 48 | BASE_LR: 0.00035 49 | BIAS_LR_FACTOR: 2. 50 | WEIGHT_DECAY: 0.0005 51 | WEIGHT_DECAY_BIAS: 0.0 52 | IMS_PER_BATCH: 64 53 | 54 | SCHED: "WarmupMultiStepLR" 55 | STEPS: [15, 25] 56 | GAMMA: 0.1 57 | 58 | WARMUP_FACTOR: 0.01 59 | WARMUP_ITERS: 5 60 | 61 | CHECKPOINT_PERIOD: 10 62 | 63 | TEST: 64 | EVAL_PERIOD: 5 65 | IMS_PER_BATCH: 128 66 | 67 | CUDNN_BENCHMARK: True 68 | 69 | OUTPUT_DIR: "projects/PartialReID/logs/test_partial" 70 | -------------------------------------------------------------------------------- /fast-reid/projects/PartialReID/partialreid/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: xingyu liao 4 | @contact: liaoxingyu5@jd.com 5 | """ 6 | 7 | from .partial_dataset import * 8 | from .partialbaseline import PartialBaseline 9 | from .dsr_head import DSRHead 10 | from .config import add_partialreid_config 11 | from .dsr_evaluation import DsrEvaluator 12 | -------------------------------------------------------------------------------- /fast-reid/projects/PartialReID/partialreid/config.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: l1aoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from fastreid.config import CfgNode as CN 8 | 9 | 10 | def add_partialreid_config(cfg): 11 | _C = cfg 12 | 13 | _C.TEST.DSR = CN() 14 | _C.TEST.DSR.ENABLED = True 15 | 16 | -------------------------------------------------------------------------------- /fast-reid/projects/PartialReID/partialreid/partialbaseline.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @authorr: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | from fastreid.modeling.losses import * 8 | from fastreid.modeling.meta_arch import Baseline 9 | from fastreid.modeling.meta_arch.build import META_ARCH_REGISTRY 10 | 11 | 12 | @META_ARCH_REGISTRY.register() 13 | class PartialBaseline(Baseline): 14 | 15 | def losses(self, outputs, gt_labels): 16 | cls_outputs, fore_cls_outputs, pred_class_logits, global_feat, fore_feat = outputs 17 | loss_dict = {} 18 | loss_names = self._cfg.MODEL.LOSSES.NAME 19 | 20 | # Log prediction accuracy 21 | CrossEntropyLoss.log_accuracy(pred_class_logits, gt_labels) 22 | 23 | if "CrossEntropyLoss" in loss_names: 24 | loss_dict['loss_avg_branch_cls'] = CrossEntropyLoss(self._cfg)(cls_outputs, gt_labels) 25 | loss_dict['loss_fore_branch_cls'] = CrossEntropyLoss(self._cfg)(fore_cls_outputs, gt_labels) 26 | 27 | if "TripletLoss" in loss_names: 28 | loss_dict['loss_avg_branch_triplet'] = TripletLoss(self._cfg)(global_feat, gt_labels) 29 | loss_dict['loss_fore_branch_triplet'] = TripletLoss(self._cfg)(fore_feat, gt_labels) 30 | 31 | return loss_dict 32 | -------------------------------------------------------------------------------- /fast-reid/projects/README.md: -------------------------------------------------------------------------------- 1 | 2 | Here are a few projects that are built on fastreid. 3 | They are examples of how to use fastrei as a library, to make your projects more maintainable. 4 | 5 | # Projects by JDAI 6 | 7 | Note that these are research projects, and therefore may not have the same level of support or stability of fastreid. 8 | 9 | - [Deep Spatial Feature Reconstruction for Partial Person Re-identification](https://github.com/JDAI-CV/fast-reid/tree/master/projects/PartialReID) 10 | 11 | # External Projects 12 | 13 | External projects in the community that use fastreid: 14 | 15 | -------------------------------------------------------------------------------- /fast-reid/start.sh: -------------------------------------------------------------------------------- 1 | CKER_ADDRESS=registry.aibee.cn/aibee/torchpp:1.4.2 2 | docker pull $DOCKER_ADDRESS 3 | docker run --shm-size=128gb -it -d \ 4 | --network=host \ 5 | -e COLUMNS=`tput cols` \ 6 | -e LINES=`tput lines` \ 7 | -v /etc/localtime:/etc/localtime:ro \ 8 | -v /ssd:/ssd \ 9 | -v /mnt:/mnt \ 10 | -v $PWD/..:/workspace \ 11 | $DOCKER_ADDRESS \ 12 | bash 13 | 14 | -------------------------------------------------------------------------------- /fast-reid/start_docker.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | 3 | DOCKER_ADDRESS=registry.aibee.cn/aibee/torchpp:1.4.2 4 | docker pull $DOCKER_ADDRESS 5 | nvidia-docker run --shm-size=128gb -it --rm -d \ 6 | --network=host \ 7 | -e COLUMNS=`tput cols` \ 8 | -e LINES=`tput lines` \ 9 | -v /etc/localtime:/etc/localtime:ro \ 10 | -v /ssd:/ssd \ 11 | -v /mnt:/mnt \ 12 | -v $PWD/..:/workspace \ 13 | -p 12355:12355 \ 14 | $DOCKER_ADDRESS \ 15 | bash 16 | -------------------------------------------------------------------------------- /fast-reid/tests/__init__.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: sherlock 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | -------------------------------------------------------------------------------- /fast-reid/tests/dataset_test.py: -------------------------------------------------------------------------------- 1 | # encoding: utf-8 2 | """ 3 | @author: liaoxingyu 4 | @contact: sherlockliao01@gmail.com 5 | """ 6 | 7 | import sys 8 | sys.path.append('.') 9 | from data import get_dataloader 10 | from config import cfg 11 | import argparse 12 | from data.datasets import init_dataset 13 | # cfg.DATALOADER.SAMPLER = 'triplet' 14 | cfg.DATASETS.NAMES = ("market1501", "dukemtmc", "cuhk03", "msmt17",) 15 | 16 | 17 | if __name__ == '__main__': 18 | parser = argparse.ArgumentParser(description="ReID Baseline Training") 19 | parser.add_argument( 20 | '-cfg', "--config_file", 21 | default="", 22 | metavar="FILE", 23 | help="path to config file", 24 | type=str 25 | ) 26 | # parser.add_argument("--local_rank", type=int, default=0) 27 | parser.add_argument("opts", help="Modify config options using the command-line", default=None, 28 | nargs=argparse.REMAINDER) 29 | args = parser.parse_args() 30 | cfg.merge_from_list(args.opts) 31 | 32 | # dataset = init_dataset('msmt17', combineall=True) 33 | get_dataloader(cfg) 34 | # tng_dataloader, val_dataloader, num_classes, num_query = get_dataloader(cfg) 35 | # def get_ex(): return open_image('datasets/beijingStation/query/000245_c10s2_1561732033722.000000.jpg') 36 | # im = get_ex() 37 | # print(data.train_ds[0]) 38 | # print(data.test_ds[0]) 39 | # a = next(iter(data.train_dl)) 40 | # from IPython import embed; embed() 41 | # from ipdb import set_trace; set_trace() 42 | # im.apply_tfms(crop_pad(size=(300, 300))) 43 | -------------------------------------------------------------------------------- /fast-reid/tests/feature_align.py: -------------------------------------------------------------------------------- 1 | import unittest 2 | import numpy as np 3 | import os 4 | from glob import glob 5 | 6 | 7 | class TestFeatureAlign(unittest.TestCase): 8 | def test_caffe_pytorch_feat_align(self): 9 | caffe_feat_path = "/export/home/lxy/cvpalgo-fast-reid/tools/deploy/caffe_R50_output" 10 | pytorch_feat_path = "/export/home/lxy/cvpalgo-fast-reid/demo/logs/R50_256x128_pytorch_feat_output" 11 | feat_filenames = os.listdir(caffe_feat_path) 12 | for feat_name in feat_filenames: 13 | caffe_feat = np.load(os.path.join(caffe_feat_path, feat_name)) 14 | pytorch_feat = np.load(os.path.join(pytorch_feat_path, feat_name)) 15 | sim = np.dot(caffe_feat, pytorch_feat.transpose())[0][0] 16 | assert sim > 0.97, f"Got similarity {sim} and feature of {feat_name} is not aligned" 17 | 18 | def test_model_performance(self): 19 | caffe_feat_path = "/export/home/lxy/cvpalgo-fast-reid/tools/deploy/caffe_R50_output" 20 | feat_filenames = os.listdir(caffe_feat_path) 21 | feats = [] 22 | for feat_name in feat_filenames: 23 | caffe_feat = np.load(os.path.join(caffe_feat_path, feat_name)) 24 | feats.append(caffe_feat) 25 | from ipdb import set_trace; set_trace() 26 | 27 | 28 | 29 | if __name__ == '__main__': 30 | unittest.main() 31 | -------------------------------------------------------------------------------- /fast-reid/tests/interp_test.py: -------------------------------------------------------------------------------- 1 | import torch 2 | from fastai.vision import * 3 | from fastai.basic_data import * 4 | from fastai.layers import * 5 | 6 | import sys 7 | sys.path.append('.') 8 | from engine.interpreter import ReidInterpretation 9 | 10 | from data import get_data_bunch 11 | from modeling import build_model 12 | from config import cfg 13 | cfg.DATASETS.NAMES = ('market1501',) 14 | cfg.DATASETS.TEST_NAMES = 'market1501' 15 | cfg.MODEL.BACKBONE = 'resnet50' 16 | 17 | data_bunch, test_labels, num_query = get_data_bunch(cfg) 18 | 19 | model = build_model(cfg, 10) 20 | model.load_params_wo_fc(torch.load('logs/2019.8.14/market/baseline/models/model_149.pth')['model']) 21 | learn = Learner(data_bunch, model) 22 | 23 | feats, _ = learn.get_preds(DatasetType.Test, activ=Lambda(lambda x: x)) -------------------------------------------------------------------------------- /fast-reid/tests/lr_scheduler_test.py: -------------------------------------------------------------------------------- 1 | import sys 2 | import unittest 3 | 4 | import torch 5 | from torch import nn 6 | 7 | sys.path.append('.') 8 | from solver.lr_scheduler import WarmupMultiStepLR 9 | from solver.build import make_optimizer 10 | from config import cfg 11 | 12 | 13 | class MyTestCase(unittest.TestCase): 14 | def test_something(self): 15 | net = nn.Linear(10, 10) 16 | optimizer = make_optimizer(cfg, net) 17 | lr_scheduler = WarmupMultiStepLR(optimizer, [20, 40], warmup_iters=10) 18 | for i in range(50): 19 | lr_scheduler.step() 20 | for j in range(3): 21 | print(i, lr_scheduler.get_lr()[0]) 22 | optimizer.step() 23 | 24 | 25 | if __name__ == '__main__': 26 | unittest.main() 27 | -------------------------------------------------------------------------------- /fast-reid/tests/model_test.py: -------------------------------------------------------------------------------- 1 | import unittest 2 | 3 | import torch 4 | 5 | import sys 6 | sys.path.append('.') 7 | from fastreid.config import cfg 8 | from fastreid.modeling.backbones import build_resnet_backbone 9 | from fastreid.modeling.backbones.resnet_ibn_a import se_resnet101_ibn_a 10 | from torch import nn 11 | 12 | 13 | class MyTestCase(unittest.TestCase): 14 | def test_se_resnet101(self): 15 | cfg.MODEL.BACKBONE.NAME = 'resnet101' 16 | cfg.MODEL.BACKBONE.DEPTH = 101 17 | cfg.MODEL.BACKBONE.WITH_IBN = True 18 | cfg.MODEL.BACKBONE.WITH_SE = True 19 | cfg.MODEL.BACKBONE.PRETRAIN_PATH = '/export/home/lxy/.cache/torch/checkpoints/se_resnet101_ibn_a.pth.tar' 20 | 21 | net1 = build_resnet_backbone(cfg) 22 | net1.cuda() 23 | net2 = nn.DataParallel(se_resnet101_ibn_a()) 24 | res = net2.load_state_dict(torch.load(cfg.MODEL.BACKBONE.PRETRAIN_PATH)['state_dict'], strict=False) 25 | net2.cuda() 26 | x = torch.randn(10, 3, 256, 128).cuda() 27 | y1 = net1(x) 28 | y2 = net2(x) 29 | assert y1.sum() == y2.sum(), 'train mode problem' 30 | net1.eval() 31 | net2.eval() 32 | y1 = net1(x) 33 | y2 = net2(x) 34 | assert y1.sum() == y2.sum(), 'eval mode problem' 35 | 36 | 37 | if __name__ == '__main__': 38 | unittest.main() 39 | -------------------------------------------------------------------------------- /fast-reid/tests/sampler_test.py: -------------------------------------------------------------------------------- 1 | import unittest 2 | import sys 3 | sys.path.append('.') 4 | from fastreid.data.samplers import TrainingSampler 5 | 6 | 7 | class SamplerTestCase(unittest.TestCase): 8 | def test_training_sampler(self): 9 | sampler = TrainingSampler(5) 10 | for i in sampler: 11 | from ipdb import set_trace; set_trace() 12 | print(i) 13 | 14 | 15 | if __name__ == '__main__': 16 | unittest.main() 17 | -------------------------------------------------------------------------------- /fast-reid/tools/deploy/Caffe/ReadMe.md: -------------------------------------------------------------------------------- 1 | # The Caffe in nn_tools Provides some convenient API 2 | If there are some problem in parse your prototxt or caffemodel, Please replace 3 | the caffe.proto with your own version and compile it with command 4 | `protoc --python_out ./ caffe.proto` 5 | 6 | ## caffe_net.py 7 | Using `from nn_tools.Caffe import caffe_net` to import this model 8 | ### Prototxt 9 | + `net=caffe_net.Prototxt(file_name)` to open a prototxt file 10 | + `net.init_caffemodel(caffe_cmd_path='caffe')` to generate a caffemodel file in the current work directory \ 11 | if your `caffe` cmd not in the $PATH, specify your caffe cmd path by the `caffe_cmd_path` kwargs. 12 | ### Caffemodel 13 | + `net=caffe_net.Caffemodel(file_name)` to open a caffemodel 14 | + `net.save_prototxt(path)` to save the caffemodel to a prototxt file (not containing the weight data) 15 | + `net.get_layer_data(layer_name)` return the numpy ndarray data of the layer 16 | + `net.set_layer_date(layer_name, datas)` specify the data of one layer in the caffemodel .`datas` is normally a list of numpy ndarray `[weights,bias]` 17 | + `net.save(path)` save the changed caffemodel 18 | ### Functions for both Prototxt and Caffemodel 19 | + `net.add_layer(layer_params,before='',after='')` add a new layer with `Layer_Param` object 20 | + `net.remove_layer_by_name(layer_name)` 21 | + `net.get_layer_by_name(layer_name)` or `net.layer(layer_name)` get the raw Layer object defined in caffe_pb2 22 | -------------------------------------------------------------------------------- /fast-reid/tools/deploy/Caffe/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/tools/deploy/Caffe/__init__.py -------------------------------------------------------------------------------- /fast-reid/tools/deploy/Caffe/caffe_lmdb.py: -------------------------------------------------------------------------------- 1 | import lmdb 2 | from Caffe import caffe_pb2 as pb2 3 | import numpy as np 4 | 5 | class Read_Caffe_LMDB(): 6 | def __init__(self,path,dtype=np.uint8): 7 | 8 | self.env=lmdb.open(path, readonly=True) 9 | self.dtype=dtype 10 | self.txn=self.env.begin() 11 | self.cursor=self.txn.cursor() 12 | 13 | @staticmethod 14 | def to_numpy(value,dtype=np.uint8): 15 | datum = pb2.Datum() 16 | datum.ParseFromString(value) 17 | flat_x = np.fromstring(datum.data, dtype=dtype) 18 | data = flat_x.reshape(datum.channels, datum.height, datum.width) 19 | label=flat_x = datum.label 20 | return data,label 21 | 22 | def iterator(self): 23 | while True: 24 | key,value=self.cursor.key(),self.cursor.value() 25 | yield self.to_numpy(value,self.dtype) 26 | if not self.cursor.next(): 27 | return 28 | 29 | def __iter__(self): 30 | self.cursor.first() 31 | it = self.iterator() 32 | return it 33 | 34 | def __len__(self): 35 | return int(self.env.stat()['entries']) 36 | -------------------------------------------------------------------------------- /fast-reid/tools/deploy/Caffe/net.py: -------------------------------------------------------------------------------- 1 | raise ImportError,'the nn_tools.Caffe.net is no longer used, please use nn_tools.Caffe.caffe_net' -------------------------------------------------------------------------------- /fast-reid/tools/deploy/run_export.sh: -------------------------------------------------------------------------------- 1 | python caffe_export.py --config-file /export/home/lxy/cvpalgo-fast-reid/logs/dukemtmc/R34/config.yaml \ 2 | --name "baseline_R34" \ 3 | --output logs/caffe_R34 \ 4 | --opts MODEL.WEIGHTS /export/home/lxy/cvpalgo-fast-reid/logs/dukemtmc/R34/model_final.pth 5 | -------------------------------------------------------------------------------- /fast-reid/tools/deploy/test_data/0022_c6s1_002976_01.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/tools/deploy/test_data/0022_c6s1_002976_01.jpg -------------------------------------------------------------------------------- /fast-reid/tools/deploy/test_data/0027_c2s2_091032_02.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/tools/deploy/test_data/0027_c2s2_091032_02.jpg -------------------------------------------------------------------------------- /fast-reid/tools/deploy/test_data/0032_c6s1_002851_01.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/tools/deploy/test_data/0032_c6s1_002851_01.jpg -------------------------------------------------------------------------------- /fast-reid/tools/deploy/test_data/0048_c1s1_005351_01.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/tools/deploy/test_data/0048_c1s1_005351_01.jpg -------------------------------------------------------------------------------- /fast-reid/tools/deploy/test_data/0065_c6s1_009501_02.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/fast-reid/tools/deploy/test_data/0065_c6s1_009501_02.jpg -------------------------------------------------------------------------------- /fast-reid/tools/mot17_test.log: -------------------------------------------------------------------------------- 1 | 2020-09-03 19:01:19,591 - root - INFO - start processing ch01005_20200609090000.mp4.cut.mp4 2 | 2020-09-03 19:01:51,146 - root - INFO - processing ch01005_20200609090000.mp4.cut.mp4 done. 3 | 2020-09-03 19:01:51,352 - root - INFO - start processing ch01001_20200609090000.mp4.cut.mp4 4 | 2020-09-03 19:01:53,871 - root - INFO - processing ch01001_20200609090000.mp4.cut.mp4 done. 5 | 2020-09-03 19:01:53,883 - root - INFO - start processing ch01003_20200609090000.mp4.cut.mp4 6 | 2020-09-03 19:02:09,003 - root - INFO - processing ch01003_20200609090000.mp4.cut.mp4 done. 7 | 2020-09-03 19:02:09,112 - root - INFO - start processing ch01002_20200609090000.mp4.cut.mp4 8 | 2020-09-03 19:02:25,516 - root - INFO - processing ch01002_20200609090000.mp4.cut.mp4 done. 9 | 2020-09-03 19:02:25,556 - root - INFO - All done, total collected 2104 pids, 4459 imgs 10 | -------------------------------------------------------------------------------- /fast-reid/tools/mot17_train.log: -------------------------------------------------------------------------------- 1 | 2020-09-02 19:15:59,171 - root - INFO - start processing ch01710_20200616090000.mp4.cut.mp4 2 | 2020-09-02 19:16:02,357 - root - INFO - processing ch01710_20200616090000.mp4.cut.mp4 done. 3 | 2020-09-02 19:16:02,377 - root - INFO - start processing ch01702_20200616090000.mp4.cut.mp4 4 | 2020-09-02 19:16:05,374 - root - INFO - processing ch01702_20200616090000.mp4.cut.mp4 done. 5 | 2020-09-02 19:16:05,388 - root - INFO - start processing ch01704_20200616090000.mp4.cut.mp4 6 | 2020-09-02 19:16:09,996 - root - INFO - processing ch01704_20200616090000.mp4.cut.mp4 done. 7 | 2020-09-02 19:16:10,017 - root - INFO - start processing ch01705_20200616090000.mp4.cut.mp4 8 | 2020-09-02 19:16:11,127 - root - INFO - processing ch01705_20200616090000.mp4.cut.mp4 done. 9 | 2020-09-02 19:16:11,133 - root - INFO - start processing ch01709_20200616090000.mp4.cut.mp4 10 | 2020-09-02 19:16:13,985 - root - INFO - processing ch01709_20200616090000.mp4.cut.mp4 done. 11 | 2020-09-02 19:16:14,006 - root - INFO - start processing ch01713_20200616090000.mp4.cut.mp4 12 | 2020-09-02 19:16:17,115 - root - INFO - processing ch01713_20200616090000.mp4.cut.mp4 done. 13 | 2020-09-02 19:16:17,140 - root - INFO - start processing ch01711_20200616090000.mp4.cut.mp4 14 | 2020-09-02 19:16:21,891 - root - INFO - processing ch01711_20200616090000.mp4.cut.mp4 done. 15 | 2020-09-02 19:16:21,902 - root - INFO - All done, total collected 472 pids, 2389 imgs 16 | -------------------------------------------------------------------------------- /fast-reid/tools/split_train_eval.py: -------------------------------------------------------------------------------- 1 | import os 2 | import os.path as osp 3 | import sys 4 | import random 5 | from collections import defaultdict 6 | 7 | label_file = '/ssd/zphe/data/NAIC/1/A/train/label.txt' 8 | val_ratio = 0.2 9 | pid_imgs = defaultdict(list) 10 | 11 | with open(label_file, 'r') as rf: 12 | for line in rf: 13 | items = line.strip().split(':') 14 | imgname = items[0] 15 | pid = items[1] 16 | pid_imgs[pid].append(imgname) 17 | 18 | total_pids = list(pid_imgs.keys()) 19 | num = len(total_pids) 20 | val_num = int(num * val_ratio) 21 | 22 | val_pids = random.sample(total_pids, val_num) 23 | train_pids = list(set(total_pids) - set(val_pids)) 24 | 25 | wf = open('NAIC_train_val.txt', 'w') 26 | for pid in train_pids: 27 | for img in pid_imgs[pid]: 28 | line = img + ' ' + pid + ' 0\n' 29 | wf.write(line) 30 | 31 | for pid in val_pids: 32 | imgs = pid_imgs[pid] 33 | img_num = len(imgs) 34 | qry_num = int(img_num * 0.5) 35 | random.shuffle(imgs) 36 | for idx, img in enumerate(imgs): 37 | if idx < qry_num: 38 | line = img + ' ' + pid + ' 1\n' 39 | else: 40 | line = img + ' ' + pid + ' 2\n' 41 | wf.write(line) -------------------------------------------------------------------------------- /images/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/images/framework.png -------------------------------------------------------------------------------- /learnable_proposal_classifier/gcn_based_purity_network/dsgcn/__init__.py: -------------------------------------------------------------------------------- 1 | #from .test_cluster_det import test_cluster_det 2 | #from .train_cluster_det import train_cluster_det 3 | #from .debug_cluster_det import debug_cluster_det 4 | #from .get_cluster_det import get_cluster_det 5 | 6 | 7 | __factory__ = { } 8 | 9 | def build_handler(phase, stage): 10 | key_handler = '{}_{}'.format(phase, stage) 11 | if key_handler not in __factory__: 12 | raise KeyError("Unknown op:", key_handler) 13 | return __factory__[key_handler] 14 | -------------------------------------------------------------------------------- /learnable_proposal_classifier/gcn_based_purity_network/dsgcn/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | from .cluster_dataset import ClusterDataset_pipeline 2 | from .cluster_det_processor import ClusterDetProcessor_pipeline 3 | from .build_dataloader import build_dataloader 4 | 5 | 6 | __factory__ = { 7 | 'pipeline': ClusterDetProcessor_pipeline 8 | } 9 | 10 | 11 | def build_dataset(cfg): 12 | return ClusterDataset(cfg) 13 | 14 | def build_dataset_pipeline(cfg): 15 | return ClusterDataset_pipeline(cfg) 16 | 17 | 18 | def build_processor(name): 19 | if name not in __factory__: 20 | raise KeyError("Unknown processor:", name) 21 | return __factory__[name] 22 | -------------------------------------------------------------------------------- /learnable_proposal_classifier/gcn_based_purity_network/dsgcn/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .dsgcn import dsgcn 2 | __factory__ = { 3 | 'dsgcn': dsgcn, 4 | } 5 | 6 | 7 | def build_model(name, *args, **kwargs): 8 | if name not in __factory__: 9 | raise KeyError("Unknown model:", name) 10 | return __factory__[name](*args, **kwargs) 11 | -------------------------------------------------------------------------------- /learnable_proposal_classifier/proto/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/learnable_proposal_classifier/proto/__init__.py -------------------------------------------------------------------------------- /learnable_proposal_classifier/proto/common.proto: -------------------------------------------------------------------------------- 1 | /* Proto definition for common components. */ 2 | 3 | syntax = "proto3"; 4 | 5 | enum DetectionType { 6 | DETECTION_TYPE_SDP = 0; 7 | DETECTION_TYPE_DPM = 1; 8 | DETECTION_TYPE_FRCNN = 2; 9 | } 10 | 11 | message TrackingDetection { 12 | uint32 frame_index = 1; 13 | uint32 detection_id = 2; 14 | float detection_score = 3; 15 | int32 box_x = 4; 16 | int32 box_y = 5; 17 | int32 box_width = 6; 18 | int32 box_height = 7; 19 | AppearanceFeatures features = 8; 20 | } 21 | 22 | message AppearanceFeatures { 23 | repeated Appearance features = 1; 24 | uint32 dim = 2; 25 | } 26 | 27 | message Appearance { 28 | repeated float feats = 1; 29 | } 30 | -------------------------------------------------------------------------------- /learnable_proposal_classifier/proto/detection_results.proto: -------------------------------------------------------------------------------- 1 | /* Proto definition for detection results. */ 2 | 3 | syntax = "proto3"; 4 | 5 | import "common.proto"; 6 | 7 | message Detections { 8 | repeated TrackingDetection tracked_detections = 1; 9 | DetectionType detection_type = 2; 10 | string appearance_model_url = 3; 11 | } 12 | -------------------------------------------------------------------------------- /learnable_proposal_classifier/proto/online_tracking_results.proto: -------------------------------------------------------------------------------- 1 | /* Proto definition for tracking results. */ 2 | 3 | syntax = "proto3"; 4 | 5 | import "common.proto"; 6 | 7 | message Tracks { 8 | repeated Track tracks = 1; 9 | DetectionType detection_type = 2; 10 | string appearance_model_url = 3; 11 | } 12 | 13 | 14 | message Track { 15 | float confidence = 1; 16 | string track_id = 2; 17 | AppearanceFeatures features = 3; 18 | repeated TrackingDetection tracked_detections = 4; 19 | uint32 tracklet_id = 5; 20 | } 21 | -------------------------------------------------------------------------------- /learnable_proposal_classifier/svonline/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/learnable_proposal_classifier/svonline/__init__.py -------------------------------------------------------------------------------- /learnable_proposal_classifier/svonline/algorithms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/learnable_proposal_classifier/svonline/algorithms/__init__.py -------------------------------------------------------------------------------- /learnable_proposal_classifier/svonline/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/learnable_proposal_classifier/svonline/core/__init__.py -------------------------------------------------------------------------------- /learnable_proposal_classifier/svonline/core/data_types.py: -------------------------------------------------------------------------------- 1 | from enum import Enum 2 | import numpy as np 3 | 4 | class NodeTyep(Enum): 5 | Detection = 0 6 | Tracklet2D = 1 7 | 8 | class FeatureTypes(Enum): 9 | Box2D = "box2D" 10 | OrigBox2D = "OrigBox2D" 11 | DetectionScore = "score" 12 | ReID = "ReID" 13 | ID = "DetectionIDs" # it is not really a feature.. but convenient to use existing framework.. 14 | 15 | class AffinityTypes(Enum): 16 | DirectedAffinity = "directed" 17 | UndirectedAffinity = "undirected" 18 | Unknown = "unknown" 19 | 20 | class AffinityOutputType(Enum): 21 | Dense = 0 22 | Sparse = 1 # very fast when it filter out a lot of trivial cases (e.g., temporal gap, camera) 23 | 24 | class ModelTypes(Enum): 25 | MedianAppearance = "MedianAppearance" 26 | MovingAverage3D = "MovingAverage3D" 27 | Unknown = "unknown" 28 | 29 | class NodeAttributeTypes(Enum): 30 | VideoFileName = "VideoFileName" 31 | TrackletType = "TrackletType" 32 | SVID = "SVID" 33 | DeteectedID = "DetectedID" 34 | Unknown = "unknown" 35 | 36 | MAX_AFFINITY_VAL = 1e10 37 | EPS = 1e-10 38 | 39 | class IDContainer(object): 40 | ''' 41 | Object containing all the IDs over the pipeline steps 42 | ''' 43 | single_view_id = -1 # assigend after any single_view_pipeline 44 | labeled_pid = -1 # GT labeled PID 45 | frame_index = -1 # frame index in original video 46 | -------------------------------------------------------------------------------- /learnable_proposal_classifier/svonline/fileio/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/learnable_proposal_classifier/svonline/fileio/__init__.py -------------------------------------------------------------------------------- /models/dsgcn_model_iter_30.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daip13/LPC_MOT/2fb7657278b33eb82a537537c2534de68c3d0277/models/dsgcn_model_iter_30.pth --------------------------------------------------------------------------------