fast-reid windows系统部署流程
fast-reid为京东提供的一个开源reid工具包。本篇文章将介绍他在windows系统下的部署流程
一、fast-feid工程下载
官方库连接:https://github.com/JDAI-CV/fast-reid 选择Download ZIP,下载源码文件。 解压到任意路径,得到fast-reid-master文件夹
二、环境配置
1.创建conda虚拟环境
打开anaconda命令行,并创建名为fast-reid的虚拟环境
conda create -n fast-reid python=3.7
conda activate fast-rei
2安装pytorch
在pytorch: link官网查询电脑cuda版本对应的pytorch安装命令 注:需要根据自己电脑选择,不要复制,本机cuda为11.1(cuda版本查询命令为nvida-smi)
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
3安装reqiurements.txt中其他包
切换到fast-reid-master文件夹路径下 执行命令
pip install -r docs/requirements.txt
安装fassi-gpu报错: 解决办法手动安装fassi-gpu 输入命令:
conda install -c conda-forge faiss-gpu
又遇到问题: CondaError: Downloaded bytes did not match Content-Length 解决办法: 设置下载超时限制 设置 conda下载的 timeout,然后再进行尝试,如果时长不够可以再设置大一点。
conda config --set remote_read_timeout_secs 600.0
手动安装fassi完毕后,注释掉requirements.txt中的fassi-gpu项 重新执行pip install -r docs/requirements.txt 安装其他包
4编译
切换到fastreid\evaluation\rank_cylib目录下,执行
python setup.py install
安装报错: 解决办法
conda install Cython
三、测试环境
python tools/train_net.py --config-file ./configs/Market1501/bagtricks_R50.yml MODEL.DEVICE "cuda:0"
|