代码参考:https://github.com/heshuting555/TransReID
论文参考:https://arxiv.org/abs/2102.04378
1.环境
ubuntu16.04
python3.6
cuda110
torch==1.7.0+cu110
torchvision==0.8.0
timm
yacs
opencv-python==4.1.0.25
2.模型准备
https://www.kaggle.com/abhinand05/vit-base-models-pretrained-pytorch
3.修改
将下面路径修改为你自己的路径:
1)预训练模型路径
2)数据路径
3)