├── .gitignore ├── README.md ├── figs └── pipeline-cop.png ├── multi_continual.py ├── reid ├── datasets │ ├── __init__.py │ ├── cuhk02.py │ ├── cuhk03.py │ ├── cuhksysu.py │ ├── dukemtmcreid.py │ ├── grid.py │ ├── ilids.py │ ├── market1501.py │ ├── msmt17.py │ ├── personx.py │ └── prid.py ├── evaluation_metrics │ ├── __init__.py │ ├── classification.py │ └── ranking.py ├── evaluators.py ├── loss │ ├── __init__.py │ ├── crossentropy.py │ └── triplet.py ├── models │ ├── __init__.py │ ├── cm.py │ ├── gem_pooling.py │ ├── prompter.py │ └── resnet.py ├── trainers.py └── utils │ ├── __init__.py │ ├── color_transformer.py │ ├── data │ ├── __init__.py │ ├── base_dataset.py │ ├── preprocessor.py │ ├── sampler.py │ └── transforms.py │ ├── faiss_rerank.py │ ├── faiss_utils.py │ ├── logging.py │ ├── lr_scheduler.py │ ├── meters.py │ ├── osutils.py │ ├── rerank.py │ ├── serialization.py │ └── train_utils.py ├── scripts ├── multi_continual.sh ├── pretrain_market.sh └── single_continue.sh ├── single_continual.py └── source_pretrain.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/README.md -------------------------------------------------------------------------------- /figs/pipeline-cop.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/figs/pipeline-cop.png -------------------------------------------------------------------------------- /multi_continual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/multi_continual.py -------------------------------------------------------------------------------- /reid/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/datasets/__init__.py -------------------------------------------------------------------------------- /reid/datasets/cuhk02.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/datasets/cuhk02.py -------------------------------------------------------------------------------- /reid/datasets/cuhk03.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/datasets/cuhk03.py -------------------------------------------------------------------------------- /reid/datasets/cuhksysu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/datasets/cuhksysu.py -------------------------------------------------------------------------------- /reid/datasets/dukemtmcreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/datasets/dukemtmcreid.py -------------------------------------------------------------------------------- /reid/datasets/grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/datasets/grid.py -------------------------------------------------------------------------------- /reid/datasets/ilids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/datasets/ilids.py -------------------------------------------------------------------------------- /reid/datasets/market1501.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/datasets/market1501.py -------------------------------------------------------------------------------- /reid/datasets/msmt17.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/datasets/msmt17.py -------------------------------------------------------------------------------- /reid/datasets/personx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/datasets/personx.py -------------------------------------------------------------------------------- /reid/datasets/prid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/datasets/prid.py -------------------------------------------------------------------------------- /reid/evaluation_metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/evaluation_metrics/__init__.py -------------------------------------------------------------------------------- /reid/evaluation_metrics/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/evaluation_metrics/classification.py -------------------------------------------------------------------------------- /reid/evaluation_metrics/ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/evaluation_metrics/ranking.py -------------------------------------------------------------------------------- /reid/evaluators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/evaluators.py -------------------------------------------------------------------------------- /reid/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/loss/__init__.py -------------------------------------------------------------------------------- /reid/loss/crossentropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/loss/crossentropy.py -------------------------------------------------------------------------------- /reid/loss/triplet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/loss/triplet.py -------------------------------------------------------------------------------- /reid/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/models/__init__.py -------------------------------------------------------------------------------- /reid/models/cm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/models/cm.py -------------------------------------------------------------------------------- /reid/models/gem_pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/models/gem_pooling.py -------------------------------------------------------------------------------- /reid/models/prompter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/models/prompter.py -------------------------------------------------------------------------------- /reid/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/models/resnet.py -------------------------------------------------------------------------------- /reid/trainers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/trainers.py -------------------------------------------------------------------------------- /reid/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/__init__.py -------------------------------------------------------------------------------- /reid/utils/color_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/color_transformer.py -------------------------------------------------------------------------------- /reid/utils/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/data/__init__.py -------------------------------------------------------------------------------- /reid/utils/data/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/data/base_dataset.py -------------------------------------------------------------------------------- /reid/utils/data/preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/data/preprocessor.py -------------------------------------------------------------------------------- /reid/utils/data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/data/sampler.py -------------------------------------------------------------------------------- /reid/utils/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/data/transforms.py -------------------------------------------------------------------------------- /reid/utils/faiss_rerank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/faiss_rerank.py -------------------------------------------------------------------------------- /reid/utils/faiss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/faiss_utils.py -------------------------------------------------------------------------------- /reid/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/logging.py -------------------------------------------------------------------------------- /reid/utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/lr_scheduler.py -------------------------------------------------------------------------------- /reid/utils/meters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/meters.py -------------------------------------------------------------------------------- /reid/utils/osutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/osutils.py -------------------------------------------------------------------------------- /reid/utils/rerank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/rerank.py -------------------------------------------------------------------------------- /reid/utils/serialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/serialization.py -------------------------------------------------------------------------------- /reid/utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/reid/utils/train_utils.py -------------------------------------------------------------------------------- /scripts/multi_continual.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/scripts/multi_continual.sh -------------------------------------------------------------------------------- /scripts/pretrain_market.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/scripts/pretrain_market.sh -------------------------------------------------------------------------------- /scripts/single_continue.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/scripts/single_continue.sh -------------------------------------------------------------------------------- /single_continual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/single_continual.py -------------------------------------------------------------------------------- /source_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/ColorPromptReID/HEAD/source_pretrain.py --------------------------------------------------------------------------------