├── .gitignore ├── README.md ├── config ├── __init__.py └── defaults.py ├── configs ├── naic_round2_model_a.yml ├── naic_round2_model_b.yml └── naic_round2_model_se.yml ├── datasets ├── __init__.py ├── bases.py ├── make_dataloader.py ├── naic.py ├── preprocessing.py ├── sampler.py └── veri.py ├── divided_dataset.py ├── ensemble_dist.py ├── loss ├── __init__.py ├── center_loss.py ├── make_loss.py ├── metric_learning.py ├── softmax_loss.py └── triplet_loss.py ├── model ├── __init__.py ├── backbones │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── resnet.cpython-36.pyc │ │ ├── resnet_ibn_a.cpython-36.pyc │ │ ├── se_module.cpython-36.pyc │ │ └── se_resnet_ibn_a.cpython-36.pyc │ ├── resnet.py │ ├── resnet_ibn_a.py │ ├── resnet_ibn_b.py │ ├── se_module.py │ └── se_resnet_ibn_a.py └── make_model.py ├── processor ├── __init__.py └── processor.py ├── run.sh ├── solver ├── __init__.py ├── lr_scheduler.py ├── make_optimizer.py └── ranger.py ├── test.py ├── train.py ├── train_UDA.py └── utils ├── __init__.py ├── iotools.py ├── logger.py ├── meter.py ├── metrics.py └── reranking.py /.gitignore: -------------------------------------------------------------------------------- 1 | /.idea/ 2 | */__pycache__ -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/config/defaults.py -------------------------------------------------------------------------------- /configs/naic_round2_model_a.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/configs/naic_round2_model_a.yml -------------------------------------------------------------------------------- /configs/naic_round2_model_b.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/configs/naic_round2_model_b.yml -------------------------------------------------------------------------------- /configs/naic_round2_model_se.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/configs/naic_round2_model_se.yml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/datasets/bases.py -------------------------------------------------------------------------------- /datasets/make_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/datasets/make_dataloader.py -------------------------------------------------------------------------------- /datasets/naic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/datasets/naic.py -------------------------------------------------------------------------------- /datasets/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/datasets/preprocessing.py -------------------------------------------------------------------------------- /datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/datasets/sampler.py -------------------------------------------------------------------------------- /datasets/veri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/datasets/veri.py -------------------------------------------------------------------------------- /divided_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/divided_dataset.py -------------------------------------------------------------------------------- /ensemble_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/ensemble_dist.py -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/loss/__init__.py -------------------------------------------------------------------------------- /loss/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/loss/center_loss.py -------------------------------------------------------------------------------- /loss/make_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/loss/make_loss.py -------------------------------------------------------------------------------- /loss/metric_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/loss/metric_learning.py -------------------------------------------------------------------------------- /loss/softmax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/loss/softmax_loss.py -------------------------------------------------------------------------------- /loss/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/loss/triplet_loss.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/backbones/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/model/backbones/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /model/backbones/__pycache__/resnet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/model/backbones/__pycache__/resnet.cpython-36.pyc -------------------------------------------------------------------------------- /model/backbones/__pycache__/resnet_ibn_a.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/model/backbones/__pycache__/resnet_ibn_a.cpython-36.pyc -------------------------------------------------------------------------------- /model/backbones/__pycache__/se_module.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/model/backbones/__pycache__/se_module.cpython-36.pyc -------------------------------------------------------------------------------- /model/backbones/__pycache__/se_resnet_ibn_a.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/model/backbones/__pycache__/se_resnet_ibn_a.cpython-36.pyc -------------------------------------------------------------------------------- /model/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/model/backbones/resnet.py -------------------------------------------------------------------------------- /model/backbones/resnet_ibn_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/model/backbones/resnet_ibn_a.py -------------------------------------------------------------------------------- /model/backbones/resnet_ibn_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/model/backbones/resnet_ibn_b.py -------------------------------------------------------------------------------- /model/backbones/se_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/model/backbones/se_module.py -------------------------------------------------------------------------------- /model/backbones/se_resnet_ibn_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/model/backbones/se_resnet_ibn_a.py -------------------------------------------------------------------------------- /model/make_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/model/make_model.py -------------------------------------------------------------------------------- /processor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/processor/__init__.py -------------------------------------------------------------------------------- /processor/processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/processor/processor.py -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/run.sh -------------------------------------------------------------------------------- /solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/solver/__init__.py -------------------------------------------------------------------------------- /solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/solver/lr_scheduler.py -------------------------------------------------------------------------------- /solver/make_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/solver/make_optimizer.py -------------------------------------------------------------------------------- /solver/ranger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/solver/ranger.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/train.py -------------------------------------------------------------------------------- /train_UDA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/train_UDA.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/iotools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/utils/iotools.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/utils/meter.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/reranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heshuting555/NAIC_Person_ReID_DMT/HEAD/utils/reranking.py --------------------------------------------------------------------------------