├── Experiment-all_tricks-tri_center-market.sh ├── LICENCE.md ├── README.md ├── Test-all_tricks-tri_center-feat_after_bn-cos-market.sh ├── config ├── __init__.py └── defaults.py ├── configs ├── baseline.yml ├── softmax.yml ├── softmax_triplet.yml └── softmax_triplet_with_center.yml ├── data ├── __init__.py ├── build.py ├── collate_batch.py ├── datasets │ ├── __init__.py │ ├── bases.py │ ├── cuhk03.py │ ├── dataset_loader.py │ ├── dukemtmcreid.py │ ├── eval_reid.py │ ├── market1501.py │ ├── msmt17.py │ ├── nformer.py │ └── veri.py ├── samplers │ ├── __init__.py │ └── triplet_sampler.py └── transforms │ ├── __init__.py │ ├── build.py │ └── transforms.py ├── engine ├── inference.py └── trainer.py ├── layers ├── __init__.py ├── center_loss.py └── triplet_loss.py ├── modeling ├── __init__.py ├── backbones │ ├── __init__.py │ ├── resnet.py │ ├── resnet_ibn_a.py │ └── senet.py ├── baseline.py ├── model.py └── nformer.py ├── pipeline.jpg ├── solver ├── __init__.py ├── build.py └── lr_scheduler.py ├── tools ├── __init__.py ├── nformer_train.py ├── test.py └── train.py └── utils ├── __init__.py ├── iotools.py ├── logger.py ├── re_ranking.py └── reid_metric.py /Experiment-all_tricks-tri_center-market.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/Experiment-all_tricks-tri_center-market.sh -------------------------------------------------------------------------------- /LICENCE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/LICENCE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/README.md -------------------------------------------------------------------------------- /Test-all_tricks-tri_center-feat_after_bn-cos-market.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/Test-all_tricks-tri_center-feat_after_bn-cos-market.sh -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/config/defaults.py -------------------------------------------------------------------------------- /configs/baseline.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/configs/baseline.yml -------------------------------------------------------------------------------- /configs/softmax.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/configs/softmax.yml -------------------------------------------------------------------------------- /configs/softmax_triplet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/configs/softmax_triplet.yml -------------------------------------------------------------------------------- /configs/softmax_triplet_with_center.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/configs/softmax_triplet_with_center.yml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/build.py -------------------------------------------------------------------------------- /data/collate_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/collate_batch.py -------------------------------------------------------------------------------- /data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/datasets/__init__.py -------------------------------------------------------------------------------- /data/datasets/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/datasets/bases.py -------------------------------------------------------------------------------- /data/datasets/cuhk03.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/datasets/cuhk03.py -------------------------------------------------------------------------------- /data/datasets/dataset_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/datasets/dataset_loader.py -------------------------------------------------------------------------------- /data/datasets/dukemtmcreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/datasets/dukemtmcreid.py -------------------------------------------------------------------------------- /data/datasets/eval_reid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/datasets/eval_reid.py -------------------------------------------------------------------------------- /data/datasets/market1501.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/datasets/market1501.py -------------------------------------------------------------------------------- /data/datasets/msmt17.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/datasets/msmt17.py -------------------------------------------------------------------------------- /data/datasets/nformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/datasets/nformer.py -------------------------------------------------------------------------------- /data/datasets/veri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/datasets/veri.py -------------------------------------------------------------------------------- /data/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/samplers/__init__.py -------------------------------------------------------------------------------- /data/samplers/triplet_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/samplers/triplet_sampler.py -------------------------------------------------------------------------------- /data/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/transforms/__init__.py -------------------------------------------------------------------------------- /data/transforms/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/transforms/build.py -------------------------------------------------------------------------------- /data/transforms/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/data/transforms/transforms.py -------------------------------------------------------------------------------- /engine/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/engine/inference.py -------------------------------------------------------------------------------- /engine/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/engine/trainer.py -------------------------------------------------------------------------------- /layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/layers/__init__.py -------------------------------------------------------------------------------- /layers/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/layers/center_loss.py -------------------------------------------------------------------------------- /layers/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/layers/triplet_loss.py -------------------------------------------------------------------------------- /modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/modeling/__init__.py -------------------------------------------------------------------------------- /modeling/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/modeling/backbones/__init__.py -------------------------------------------------------------------------------- /modeling/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/modeling/backbones/resnet.py -------------------------------------------------------------------------------- /modeling/backbones/resnet_ibn_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/modeling/backbones/resnet_ibn_a.py -------------------------------------------------------------------------------- /modeling/backbones/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/modeling/backbones/senet.py -------------------------------------------------------------------------------- /modeling/baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/modeling/baseline.py -------------------------------------------------------------------------------- /modeling/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/modeling/model.py -------------------------------------------------------------------------------- /modeling/nformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/modeling/nformer.py -------------------------------------------------------------------------------- /pipeline.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/pipeline.jpg -------------------------------------------------------------------------------- /solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/solver/__init__.py -------------------------------------------------------------------------------- /solver/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/solver/build.py -------------------------------------------------------------------------------- /solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/solver/lr_scheduler.py -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/tools/__init__.py -------------------------------------------------------------------------------- /tools/nformer_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/tools/nformer_train.py -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/tools/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/iotools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/utils/iotools.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/re_ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/utils/re_ranking.py -------------------------------------------------------------------------------- /utils/reid_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haochenheheda/NFormer/HEAD/utils/reid_metric.py --------------------------------------------------------------------------------