CLIP-ReID初尝试

1. 下载clipreid

Syliz517/CLIP-ReID: Official implementation for "CLIP-ReID: Exploiting Vision-Language Model for Image Re-identification without Concrete Text Labels" (AAAI 2023)

2. 数据集配置

由于想试的是人员ReID，下载了DukeMTMC-reId，修改configs/person/vit_clipreid.yml配置，我在D盘下面放了dukemtmcreid\DukeMTMC-reID，配置对应的是

DATASETS:

  NAMES: ('dukemtmc')
  ROOT_DIR: ('D:/')

ReId常用数据集有：Market1501、DukeMTMC-reID、CUHK-03

3 模型初始化训练

python train_clipreid.py --config_file configs/person/vit_clipreid.yml

消费级显卡还是不太够用

4. 任务微调优化

python train_clipreid.py --config_file configs/person/vit_clipreid.yml MODEL.STAGE 2

5. 测试

python test_clipreid.py --config_file configs/person/vit_clipreid.yml TEST.WEIGHT 'ViT-B-16_60.pth'

Reference

2211.13977 CLIP-ReID: Exploiting Vision-Language Model for Image Re-Identification without Concrete Text Labels

CLIP-ReID：利用视觉-语言模型在无具体文本标签的情况下进行图像再识别 | 论文 | HyperAI超神经