TransReID模型运行

模型源码可通过下方链接进行下载，相应的数据集也可以通过下方链接找到并下载。

https://github.com/damo-cv/TransReID?tab=readme-ov-file

1.环境配置

pip install -r requirements.txt

通过以上指令安装好所需要的包。

2.预训练模型下载

在源码下载页面，找到图片所示位置进行下载（文件可下载至任何位置，但需要记住其绝对路径，以便后续使用）。

3.调整文件位置

将需要使用的数据集放置于‘data’文件夹（自己建立）中，以下通过Market-1501数据集为例。

4.运行模型

根据数据集找到对应的指令进行运行，以下以Market为例。

python train.py --config_file configs/Market/vit_transreid_stride.yml MODEL.DEVICE_ID "('0')"

 找到指令中的yml文件，并进行如下修改：

在PRETRAIN_PATH一项中，将路径修改为你预训练模型所在位置即可。

再找到vit_pytorch.py文件，将下述代码进行替换。

# 原代码
from torch._six import container_abcs

# 替换后代码
import collections.abc as container_abcs
int_classes = int
string_classes = str

完成上述操作后，执行上述指令，即可运行模型。

5.评估

# DukeMTMC
python test.py --config_file configs/DukeMTMC/vit_transreid_stride.yml MODEL.DEVICE_ID "('0')"  TEST.WEIGHT '../logs/duke_vit_transreid_stride/transformer_120.pth'
# MSMT17
python test.py --config_file configs/MSMT17/vit_transreid_stride.yml MODEL.DEVICE_ID "('0')" TEST.WEIGHT '../logs/msmt17_vit_transreid_stride/transformer_120.pth'
# OCC_Duke
python test.py --config_file configs/OCC_Duke/vit_transreid_stride.yml MODEL.DEVICE_ID "('0')" TEST.WEIGHT '../logs/occ_duke_vit_transreid_stride/transformer_120.pth'
# Market
python test.py --config_file configs/Market/vit_transreid_stride.yml MODEL.DEVICE_ID "('0')"  TEST.WEIGHT '../logs/market_vit_transreid_stride/transformer_120.pth'
# VeRi
python test.py --config_file configs/VeRi/vit_transreid_stride.yml MODEL.DEVICE_ID "('0')" TEST.WEIGHT '../logs/veri_vit_transreid_stride/transformer_120.pth'

# VehicleID (We test 10 times and get the final average score to avoid randomness)
python test.py --config_file configs/VehicleID/vit_transreid_stride.yml MODEL.DEVICE_ID "('0')" TEST.WEIGHT '../logs/vehicleID_vit_transreid_stride/transformer_120.pth'

同理，根据数据集选择下方对应指令进行执行。运行结果如下：