赞
踩
代码参考:https://github.com/heshuting555/TransReID
论文参考:https://arxiv.org/abs/2102.04378
- ubuntu16.04
- python3.6
- cuda110
-
- torch==1.7.0+cu110
- torchvision==0.8.0
-
- timm
- yacs
- opencv-python==4.1.0.25
https://www.kaggle.com/abhinand05/vit-base-models-pretrained-pytorch
将下面路径修改为你自己的路径:
1)预训练模型路径
2)数据路径
3)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。