├── .gitignore ├── LICENSE ├── README.md ├── config ├── __init__.py ├── defaults.py └── defaults_base.py ├── configs ├── VehicleID │ ├── cnn_base.yml │ ├── cnn_clipreid.yml │ ├── vit_base.yml │ └── vit_clipreid.yml ├── person │ ├── cnn_base.yml │ ├── cnn_clipreid.yml │ ├── vit_base.yml │ └── vit_clipreid.yml └── veri │ ├── cnn_base.yml │ ├── cnn_prom.yml │ ├── vit_base.yml │ └── vit_prom.yml ├── datasets ├── __init__.py ├── bases.py ├── dukemtmcreid.py ├── keypoint_test.txt ├── keypoint_train.txt ├── make_dataloader.py ├── make_dataloader_clipreid.py ├── market1501.py ├── msmt17.py ├── occ_duke.py ├── preprocessing.py ├── sampler.py ├── sampler_ddp.py ├── vehicleid.py └── veri.py ├── fig └── method.png ├── loss ├── __init__.py ├── arcface.py ├── center_loss.py ├── make_loss.py ├── metric_learning.py ├── softmax_loss.py ├── supcontrast.py └── triplet_loss.py ├── model ├── clip │ ├── __init__.py │ ├── bpe_simple_vocab_16e6.txt.gz │ ├── clip.py │ ├── model.py │ └── simple_tokenizer.py ├── make_model.py └── make_model_clipreid.py ├── processor ├── __init__.py ├── processor.py ├── processor_clipreid_stage1.py └── processor_clipreid_stage2.py ├── solver ├── __init__.py ├── cosine_lr.py ├── lr_scheduler.py ├── make_optimizer.py ├── make_optimizer_prompt.py ├── scheduler.py └── scheduler_factory.py ├── test.py ├── test_clipreid.py ├── train.py ├── train_clipreid.py └── utils ├── __init__.py ├── iotools.py ├── logger.py ├── meter.py ├── metrics.py └── reranking.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/config/__init__.py -------------------------------------------------------------------------------- /config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/config/defaults.py -------------------------------------------------------------------------------- /config/defaults_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/config/defaults_base.py -------------------------------------------------------------------------------- /configs/VehicleID/cnn_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/configs/VehicleID/cnn_base.yml -------------------------------------------------------------------------------- /configs/VehicleID/cnn_clipreid.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/configs/VehicleID/cnn_clipreid.yml -------------------------------------------------------------------------------- /configs/VehicleID/vit_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/configs/VehicleID/vit_base.yml -------------------------------------------------------------------------------- /configs/VehicleID/vit_clipreid.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/configs/VehicleID/vit_clipreid.yml -------------------------------------------------------------------------------- /configs/person/cnn_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/configs/person/cnn_base.yml -------------------------------------------------------------------------------- /configs/person/cnn_clipreid.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/configs/person/cnn_clipreid.yml -------------------------------------------------------------------------------- /configs/person/vit_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/configs/person/vit_base.yml -------------------------------------------------------------------------------- /configs/person/vit_clipreid.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/configs/person/vit_clipreid.yml -------------------------------------------------------------------------------- /configs/veri/cnn_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/configs/veri/cnn_base.yml -------------------------------------------------------------------------------- /configs/veri/cnn_prom.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/configs/veri/cnn_prom.yml -------------------------------------------------------------------------------- /configs/veri/vit_base.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/configs/veri/vit_base.yml -------------------------------------------------------------------------------- /configs/veri/vit_prom.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/configs/veri/vit_prom.yml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/datasets/bases.py -------------------------------------------------------------------------------- /datasets/dukemtmcreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/datasets/dukemtmcreid.py -------------------------------------------------------------------------------- /datasets/keypoint_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/datasets/keypoint_test.txt -------------------------------------------------------------------------------- /datasets/keypoint_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/datasets/keypoint_train.txt -------------------------------------------------------------------------------- /datasets/make_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/datasets/make_dataloader.py -------------------------------------------------------------------------------- /datasets/make_dataloader_clipreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/datasets/make_dataloader_clipreid.py -------------------------------------------------------------------------------- /datasets/market1501.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/datasets/market1501.py -------------------------------------------------------------------------------- /datasets/msmt17.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/datasets/msmt17.py -------------------------------------------------------------------------------- /datasets/occ_duke.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/datasets/occ_duke.py -------------------------------------------------------------------------------- /datasets/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/datasets/preprocessing.py -------------------------------------------------------------------------------- /datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/datasets/sampler.py -------------------------------------------------------------------------------- /datasets/sampler_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/datasets/sampler_ddp.py -------------------------------------------------------------------------------- /datasets/vehicleid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/datasets/vehicleid.py -------------------------------------------------------------------------------- /datasets/veri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/datasets/veri.py -------------------------------------------------------------------------------- /fig/method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/fig/method.png -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- 1 | from .arcface import ArcFace -------------------------------------------------------------------------------- /loss/arcface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/loss/arcface.py -------------------------------------------------------------------------------- /loss/center_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/loss/center_loss.py -------------------------------------------------------------------------------- /loss/make_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/loss/make_loss.py -------------------------------------------------------------------------------- /loss/metric_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/loss/metric_learning.py -------------------------------------------------------------------------------- /loss/softmax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/loss/softmax_loss.py -------------------------------------------------------------------------------- /loss/supcontrast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/loss/supcontrast.py -------------------------------------------------------------------------------- /loss/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/loss/triplet_loss.py -------------------------------------------------------------------------------- /model/clip/__init__.py: -------------------------------------------------------------------------------- 1 | from .clip import * 2 | -------------------------------------------------------------------------------- /model/clip/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/model/clip/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /model/clip/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/model/clip/clip.py -------------------------------------------------------------------------------- /model/clip/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/model/clip/model.py -------------------------------------------------------------------------------- /model/clip/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/model/clip/simple_tokenizer.py -------------------------------------------------------------------------------- /model/make_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/model/make_model.py -------------------------------------------------------------------------------- /model/make_model_clipreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/model/make_model_clipreid.py -------------------------------------------------------------------------------- /processor/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /processor/processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/processor/processor.py -------------------------------------------------------------------------------- /processor/processor_clipreid_stage1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/processor/processor_clipreid_stage1.py -------------------------------------------------------------------------------- /processor/processor_clipreid_stage2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/processor/processor_clipreid_stage2.py -------------------------------------------------------------------------------- /solver/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /solver/cosine_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/solver/cosine_lr.py -------------------------------------------------------------------------------- /solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/solver/lr_scheduler.py -------------------------------------------------------------------------------- /solver/make_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/solver/make_optimizer.py -------------------------------------------------------------------------------- /solver/make_optimizer_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/solver/make_optimizer_prompt.py -------------------------------------------------------------------------------- /solver/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/solver/scheduler.py -------------------------------------------------------------------------------- /solver/scheduler_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/solver/scheduler_factory.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/test.py -------------------------------------------------------------------------------- /test_clipreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/test_clipreid.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/train.py -------------------------------------------------------------------------------- /train_clipreid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/train_clipreid.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/iotools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/utils/iotools.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/utils/meter.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/reranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Syliz517/CLIP-ReID/HEAD/utils/reranking.py --------------------------------------------------------------------------------