参考代码:
https://github.com/JDAI-CV/fast-reid
0.环境
ubuntu16.04
cuda9.0
python3.6
torch==1.1.0
torchvision==0.3.0
Cython
yacs
tensorboard
future
termcolor
sklearn
tqdm
opencv-python==4.1.0.25
matplotlib
scikit-image
numpy==1.16.4
安装apex(不要直接通过pip安装):
git clone https://www.github.com/nvidia/apex
cd apex
# python setup.py install --cuda_ext --cpp_ext
pip install -v --no-cache-dir ./
1.准备数据
参考https://blog.csdn.net/qq_35975447/article/details/106664593
数据目录结构如下:
fas