TransReid

xiaolin03332025-11-06 11:13

环境配置

先安装一个conda环境：

bash 复制代码

conda create -n transreid python=3.8

激活conda环境：

bash 复制代码

conda activate transreid

安装其余环境：

bash 复制代码

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pip install timm -i https://pypi.doubanio.com/simple/
pip install yacs -i https://pypi.doubanio.com/simple/

可能会报错：

这是因为：torch._six 是 PyTorch 内部使用的兼容性模块，在 PyTorch 1.9+ 版本中，这个模块被移除了。
> 找到vit_pytorch.py文件，注释from torch._six import container_abcs，并添加：import collections.abc as container_abcs

添加数据集

新建一个data文件夹，把数据集放在这个文件夹下，记得更改配置文件里的路径。

添加预训练文件

新建一个checkpoints文件夹，下载预训练文件后放在这个文件夹下，这个也要记得去配置文件里修改路径。

运行

bash 复制代码

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