├── .gitignore ├── LICENSE ├── README.md ├── config.py ├── configs ├── debug.yml ├── debug_multi-gpu.yml ├── market2partilibds.yml ├── veri.yml └── veriwild.yml ├── dataset ├── __init__.py ├── collate_batch.py ├── data.py ├── transform.py └── triplet_sampler.py ├── evaluate ├── __init__.py ├── eval_cylib │ ├── Makefile │ ├── __init__.py │ ├── eval_metrics_cy.c │ ├── eval_metrics_cy.pyx │ ├── setup.py │ └── test_cython.py ├── eval_reid.py └── re_ranking.py ├── losses ├── __init__.py ├── center_loss.py ├── cluster_loss.py ├── range_loss.py └── triplet_loss.py ├── main.py ├── model ├── __init__.py ├── backbones │ ├── __init__.py │ └── resnet.py ├── baseline.py └── sync_bn │ ├── __init__.py │ ├── batchnorm.py │ ├── batchnorm_reimpl.py │ ├── comm.py │ ├── replicate.py │ └── unittest.py ├── optim.py ├── outputs ├── debug │ ├── log.txt │ └── log_eval.txt ├── debug_fp16 │ ├── log.txt │ └── log_eval.txt ├── debug_multi-gpu │ ├── log.txt │ └── log_eval.txt ├── market2partilibds │ ├── log.txt │ └── log_eval.txt ├── veri │ ├── log.txt │ └── log_eval.txt └── veriwild │ ├── log.txt │ └── log_eval.txt ├── trainer.py └── utils ├── __init__.py ├── file_op.py ├── logging.py ├── model.py ├── utils.py └── vistools.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/config.py -------------------------------------------------------------------------------- /configs/debug.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/configs/debug.yml -------------------------------------------------------------------------------- /configs/debug_multi-gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/configs/debug_multi-gpu.yml -------------------------------------------------------------------------------- /configs/market2partilibds.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/configs/market2partilibds.yml -------------------------------------------------------------------------------- /configs/veri.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/configs/veri.yml -------------------------------------------------------------------------------- /configs/veriwild.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/configs/veriwild.yml -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/collate_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/dataset/collate_batch.py -------------------------------------------------------------------------------- /dataset/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/dataset/data.py -------------------------------------------------------------------------------- /dataset/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/dataset/transform.py -------------------------------------------------------------------------------- /dataset/triplet_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/dataset/triplet_sampler.py -------------------------------------------------------------------------------- /evaluate/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/evaluate/__init__.py -------------------------------------------------------------------------------- /evaluate/eval_cylib/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/evaluate/eval_cylib/Makefile -------------------------------------------------------------------------------- /evaluate/eval_cylib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /evaluate/eval_cylib/eval_metrics_cy.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/evaluate/eval_cylib/eval_metrics_cy.c -------------------------------------------------------------------------------- /evaluate/eval_cylib/eval_metrics_cy.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/evaluate/eval_cylib/eval_metrics_cy.pyx -------------------------------------------------------------------------------- /evaluate/eval_cylib/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/evaluate/eval_cylib/setup.py -------------------------------------------------------------------------------- /evaluate/eval_cylib/test_cython.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/evaluate/eval_cylib/test_cython.py -------------------------------------------------------------------------------- /evaluate/eval_reid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/evaluate/eval_reid.py -------------------------------------------------------------------------------- /evaluate/re_ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/evaluate/re_ranking.py -------------------------------------------------------------------------------- /losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/losses/__init__.py -------------------------------------------------------------------------------- /losses/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/losses/center_loss.py -------------------------------------------------------------------------------- /losses/cluster_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/losses/cluster_loss.py -------------------------------------------------------------------------------- /losses/range_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/losses/range_loss.py -------------------------------------------------------------------------------- /losses/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/losses/triplet_loss.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/main.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /model/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/model/backbones/resnet.py -------------------------------------------------------------------------------- /model/baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/model/baseline.py -------------------------------------------------------------------------------- /model/sync_bn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/model/sync_bn/__init__.py -------------------------------------------------------------------------------- /model/sync_bn/batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/model/sync_bn/batchnorm.py -------------------------------------------------------------------------------- /model/sync_bn/batchnorm_reimpl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/model/sync_bn/batchnorm_reimpl.py -------------------------------------------------------------------------------- /model/sync_bn/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/model/sync_bn/comm.py -------------------------------------------------------------------------------- /model/sync_bn/replicate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/model/sync_bn/replicate.py -------------------------------------------------------------------------------- /model/sync_bn/unittest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/model/sync_bn/unittest.py -------------------------------------------------------------------------------- /optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/optim.py -------------------------------------------------------------------------------- /outputs/debug/log.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/outputs/debug/log.txt -------------------------------------------------------------------------------- /outputs/debug/log_eval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/outputs/debug/log_eval.txt -------------------------------------------------------------------------------- /outputs/debug_fp16/log.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/outputs/debug_fp16/log.txt -------------------------------------------------------------------------------- /outputs/debug_fp16/log_eval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/outputs/debug_fp16/log_eval.txt -------------------------------------------------------------------------------- /outputs/debug_multi-gpu/log.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/outputs/debug_multi-gpu/log.txt -------------------------------------------------------------------------------- /outputs/debug_multi-gpu/log_eval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/outputs/debug_multi-gpu/log_eval.txt -------------------------------------------------------------------------------- /outputs/market2partilibds/log.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/outputs/market2partilibds/log.txt -------------------------------------------------------------------------------- /outputs/market2partilibds/log_eval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/outputs/market2partilibds/log_eval.txt -------------------------------------------------------------------------------- /outputs/veri/log.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/outputs/veri/log.txt -------------------------------------------------------------------------------- /outputs/veri/log_eval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/outputs/veri/log_eval.txt -------------------------------------------------------------------------------- /outputs/veriwild/log.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/outputs/veriwild/log.txt -------------------------------------------------------------------------------- /outputs/veriwild/log_eval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/outputs/veriwild/log_eval.txt -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/trainer.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/file_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/utils/file_op.py -------------------------------------------------------------------------------- /utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/utils/logging.py -------------------------------------------------------------------------------- /utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/utils/model.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/utils/utils.py -------------------------------------------------------------------------------- /utils/vistools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DTennant/reid_baseline_with_syncbn/HEAD/utils/vistools.py --------------------------------------------------------------------------------