├── README.md ├── Rename.py ├── config ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ ├── __init__.cpython-38.pyc │ ├── defaults.cpython-37.pyc │ └── defaults.cpython-38.pyc └── defaults.py ├── configs ├── DukeMTMC │ └── dpm.yml ├── Market │ └── dpm.yml └── OCC_Duke │ └── dpm.yml ├── datasets ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ ├── __init__.cpython-38.pyc │ ├── bases.cpython-37.pyc │ ├── dukemtmcreid.cpython-37.pyc │ ├── make_dataloader.cpython-37.pyc │ ├── make_dataloader.cpython-38.pyc │ ├── market1501.cpython-37.pyc │ ├── msmt17.cpython-37.pyc │ ├── occ_duke.cpython-37.pyc │ ├── sampler.cpython-37.pyc │ ├── sampler_ddp.cpython-37.pyc │ ├── vehicleid.cpython-37.pyc │ └── veri.cpython-37.pyc ├── bases.py ├── dukemtmcreid.py ├── make_dataloader.py ├── market1501.py ├── occ_duke.py ├── preprocessing.py ├── sampler.py └── sampler_ddp.py ├── loss ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ ├── arcface.cpython-37.pyc │ ├── center_loss.cpython-37.pyc │ ├── make_loss.cpython-37.pyc │ ├── metric_learning.cpython-37.pyc │ ├── softmax_loss.cpython-37.pyc │ └── triplet_loss.cpython-37.pyc ├── arcface.py ├── center_loss.py ├── make_loss.py ├── metric_learning.py ├── softmax_loss.py └── triplet_loss.py ├── model ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ └── make_model.cpython-37.pyc ├── backbones │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-37.pyc │ │ ├── resnet.cpython-37.pyc │ │ └── vit_pytorch.cpython-37.pyc │ ├── resnet.py │ └── vit_pytorch.py └── make_model.py ├── processor ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ └── processor.cpython-37.pyc └── processor.py ├── requirements.txt ├── solver ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ ├── cosine_lr.cpython-37.pyc │ ├── lr_scheduler.cpython-37.pyc │ ├── make_optimizer.cpython-37.pyc │ ├── scheduler.cpython-37.pyc │ └── scheduler_factory.cpython-37.pyc ├── cosine_lr.py ├── lr_scheduler.py ├── make_optimizer.py ├── scheduler.py └── scheduler_factory.py ├── test.py ├── train.py └── utils ├── __init__.py ├── __pycache__ ├── __init__.cpython-37.pyc ├── __init__.cpython-38.pyc ├── iotools.cpython-37.pyc ├── logger.cpython-37.pyc ├── logger.cpython-38.pyc ├── meter.cpython-37.pyc ├── metrics.cpython-37.pyc └── reranking.cpython-37.pyc ├── iotools.py ├── logger.py ├── meter.py ├── metrics.py └── reranking.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/README.md -------------------------------------------------------------------------------- /Rename.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/Rename.py -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/config/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /config/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/config/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /config/__pycache__/defaults.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/config/__pycache__/defaults.cpython-37.pyc -------------------------------------------------------------------------------- /config/__pycache__/defaults.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/config/__pycache__/defaults.cpython-38.pyc -------------------------------------------------------------------------------- /config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/config/defaults.py -------------------------------------------------------------------------------- /configs/DukeMTMC/dpm.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/configs/DukeMTMC/dpm.yml -------------------------------------------------------------------------------- /configs/Market/dpm.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/configs/Market/dpm.yml -------------------------------------------------------------------------------- /configs/OCC_Duke/dpm.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/configs/OCC_Duke/dpm.yml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/bases.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/__pycache__/bases.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/dukemtmcreid.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/__pycache__/dukemtmcreid.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/make_dataloader.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/__pycache__/make_dataloader.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/make_dataloader.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/__pycache__/make_dataloader.cpython-38.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/market1501.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/__pycache__/market1501.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/msmt17.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/__pycache__/msmt17.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/occ_duke.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/__pycache__/occ_duke.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/sampler.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/__pycache__/sampler.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/sampler_ddp.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/__pycache__/sampler_ddp.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/vehicleid.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/__pycache__/vehicleid.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/veri.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/__pycache__/veri.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/bases.py -------------------------------------------------------------------------------- /datasets/dukemtmcreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/dukemtmcreid.py -------------------------------------------------------------------------------- /datasets/make_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/make_dataloader.py -------------------------------------------------------------------------------- /datasets/market1501.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/market1501.py -------------------------------------------------------------------------------- /datasets/occ_duke.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/occ_duke.py -------------------------------------------------------------------------------- /datasets/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/preprocessing.py -------------------------------------------------------------------------------- /datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/sampler.py -------------------------------------------------------------------------------- /datasets/sampler_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/datasets/sampler_ddp.py -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/loss/__init__.py -------------------------------------------------------------------------------- /loss/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/loss/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /loss/__pycache__/arcface.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/loss/__pycache__/arcface.cpython-37.pyc -------------------------------------------------------------------------------- /loss/__pycache__/center_loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/loss/__pycache__/center_loss.cpython-37.pyc -------------------------------------------------------------------------------- /loss/__pycache__/make_loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/loss/__pycache__/make_loss.cpython-37.pyc -------------------------------------------------------------------------------- /loss/__pycache__/metric_learning.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/loss/__pycache__/metric_learning.cpython-37.pyc -------------------------------------------------------------------------------- /loss/__pycache__/softmax_loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/loss/__pycache__/softmax_loss.cpython-37.pyc -------------------------------------------------------------------------------- /loss/__pycache__/triplet_loss.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/loss/__pycache__/triplet_loss.cpython-37.pyc -------------------------------------------------------------------------------- /loss/arcface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/loss/arcface.py -------------------------------------------------------------------------------- /loss/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/loss/center_loss.py -------------------------------------------------------------------------------- /loss/make_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/loss/make_loss.py -------------------------------------------------------------------------------- /loss/metric_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/loss/metric_learning.py -------------------------------------------------------------------------------- /loss/softmax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/loss/softmax_loss.py -------------------------------------------------------------------------------- /loss/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/loss/triplet_loss.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/model/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /model/__pycache__/make_model.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/model/__pycache__/make_model.cpython-37.pyc -------------------------------------------------------------------------------- /model/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/backbones/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/model/backbones/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /model/backbones/__pycache__/resnet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/model/backbones/__pycache__/resnet.cpython-37.pyc -------------------------------------------------------------------------------- /model/backbones/__pycache__/vit_pytorch.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/model/backbones/__pycache__/vit_pytorch.cpython-37.pyc -------------------------------------------------------------------------------- /model/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/model/backbones/resnet.py -------------------------------------------------------------------------------- /model/backbones/vit_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/model/backbones/vit_pytorch.py -------------------------------------------------------------------------------- /model/make_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/model/make_model.py -------------------------------------------------------------------------------- /processor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/processor/__init__.py -------------------------------------------------------------------------------- /processor/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/processor/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /processor/__pycache__/processor.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/processor/__pycache__/processor.cpython-37.pyc -------------------------------------------------------------------------------- /processor/processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/processor/processor.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/requirements.txt -------------------------------------------------------------------------------- /solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/solver/__init__.py -------------------------------------------------------------------------------- /solver/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/solver/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /solver/__pycache__/cosine_lr.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/solver/__pycache__/cosine_lr.cpython-37.pyc -------------------------------------------------------------------------------- /solver/__pycache__/lr_scheduler.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/solver/__pycache__/lr_scheduler.cpython-37.pyc -------------------------------------------------------------------------------- /solver/__pycache__/make_optimizer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/solver/__pycache__/make_optimizer.cpython-37.pyc -------------------------------------------------------------------------------- /solver/__pycache__/scheduler.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/solver/__pycache__/scheduler.cpython-37.pyc -------------------------------------------------------------------------------- /solver/__pycache__/scheduler_factory.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/solver/__pycache__/scheduler_factory.cpython-37.pyc -------------------------------------------------------------------------------- /solver/cosine_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/solver/cosine_lr.py -------------------------------------------------------------------------------- /solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/solver/lr_scheduler.py -------------------------------------------------------------------------------- /solver/make_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/solver/make_optimizer.py -------------------------------------------------------------------------------- /solver/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/solver/scheduler.py -------------------------------------------------------------------------------- /solver/scheduler_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/solver/scheduler_factory.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/utils/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/utils/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/iotools.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/utils/__pycache__/iotools.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/logger.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/utils/__pycache__/logger.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/logger.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/utils/__pycache__/logger.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/meter.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/utils/__pycache__/meter.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/metrics.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/utils/__pycache__/metrics.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/reranking.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/utils/__pycache__/reranking.cpython-37.pyc -------------------------------------------------------------------------------- /utils/iotools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/utils/iotools.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/utils/meter.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/reranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stone96123/DPM/HEAD/utils/reranking.py --------------------------------------------------------------------------------