windows平台FairMOT的实现

环境:python3.6+pytorch1.1.0+torchvision0.3.0+cuda9.2+vs2015

该项目需要装3个c++库(dcn_v2,apex,cython_bbox)特别坑,各种环境不匹配,各种bug。本人c++小白,但是一路摸索总算成功了。下面总结一下:

一、
1.先把源码下下来,用git(自己手安)克隆或者直接下载zip文件都可:https://github.com/ifzhang/FairMOT

2. 安装anaconda3,并用conda搭建虚拟环境:

conda create -n FairMOT python=3.6 #-n 代表环境名字
activate
conda activate FairMOT #激活这个叫FairMOT的环境

3.安装pytorch1.1.0,在官网上下载https://download.pytorch.org/whl/torch_stable.html,我选择的是:cu100/torch-1.1.0-cp36-cp36m-win_amd64.whl,下载后使用pip安装

pip install torch-1.1.0-cp36-cp36m-win_amd64.whl

4.安装对应的torchvision

pip install torchvision-0.3.0-cp36-cp36m-win_amd64.whl

5.安装cython,ninja

pip install cython
pip install ninja

6.安装opencv-python:

pip install -U opencv-python==4.1.1.26

7.安装依赖项:

pip install -r requirements.txt

8.安装:DCNv2:

下载:GitHub - maohule/DCNv2

进入DCNv2-master文件夹运行:

python setup.py build develop

在Python下import dcn_v2不报错就安装成功了。

9.安装cython_bbox,在这里下载cython-bbox · PyPI,运行下面指令安装:

python setup.py build develop install

可能报错:

cl: 命令行 error D8021 :无效的数值参数“/Wno-cpp”

修改如下一行,然后再执行安装即可。

windows平台FairMOT的实现_第1张图片

在Python下import cython_bbox不报错就安装成功了。

参考:cl: 命令行 error D8021 :无效的数值参数“/Wno-cpp”-CSDN博客

invalid numeric argument '/Wno-cpp' 解决方法-CSDN博客

10.安装apex:

git clone https://github.com/NVIDIA/apex.git这个官方的怎么安都报错,于是乎,找了这个博客(apex安装常见的三个报错并成功解决(亲测有效)_安装apex库错误-CSDN博客)的第四种方法,很有效:

git clone https://github.com/ptrblck/apex.git
cd apex
git checkout apex_no_distributed
pip install -v --no-cache-dir ./

安装后需要将FairMOT代码中的from torch.cuda import amp改为from apex import amp

在Python下import apex不报错就安装成功了。

11.修改FairMOT代码FairMOT-master\src\lib\opts.py

--load_model 设置上一步下载好的模型权重位置

在opts.py的19行

windows平台FairMOT的实现_第2张图片

0代表第一个gpu

windows平台FairMOT的实现_第3张图片

–input-video要测试视频的位置
–output-root测试完视频的输出位置

windows平台FairMOT的实现_第4张图片

12.在src文件夹下运行demo.py文件,即可测试:

python demo.py mot --input-video ../videos/MOT16-03.mp4 --load_model ../models/fairmot_dla34.pth --conf_thres 0.4

运行时出现如下错误:

Traceback (most recent call last):
  File "demo.py", line 14, in 
    from track import eval_seq
  File "C:\Users\93260\FairMOT-master1\src\track.py", line 20, in 
    import datasets.dataset.jde as datasets
ModuleNotFoundError: No module named 'datasets.dataset'

解决:将datasets.dataset.jde改为datasets.dataset.jde

出现错误:找不到models模块

解决:在models块添加__init__.py文件即可

出现错误:

AttributeError: module 'torch' has no attribute 'true_divide'

发生该错误主要原因是我用低版本的torch,torch1.7就没有这样的错误。

解决:将torch.true_divide改为 torch.div即可

二、一些坑的总结:

1.torch1.1.0+torchvision0.3.0+cuda9.2+vs2015可以成功编译dcn_v2,但是编译apex时候报错,参照Pytorch 安装 APEX 疑难杂症解决方案 - 知乎,大概意思是torch的cuda版本

torch.version.cuda

是10.0,但是nvcc --version查到的cuda版本是9.2,所以导致Cuda extensions are being compiled with a version of Cuda that does not错误。因此我按照nvcc查到的版本卸载安装了torch1.2和torchvision0.4即下载安装

利用pip uninstall 卸载torch和torchvision,再安装

pip install torch-1.2.0+cu92-cp36-cp36m-win_amd64.whl
pip install torchvision-0.4.0+cu92-cp36-cp36m-win_amd64.whl

这样编译apex就没错了,但是import dcn_v2又报错了。因此安装完apex后又重新卸载安装了torch1.1.0和torchvision0.3.0,重新编译dcn_v2。编译成功后import dcn_v2和import apex均不报错。(遇见flat_hash_map.h错误,没解决,但是通过重安pytorch方法,版本匹配后,就没有这个错误了)

2.'cl.exe' not found

首先确保你安装了C++编译工具,我直接安装Microsoft Visual studio 14.0了。一定要选择上C++编译工具。

配置环境变量:在你的系统添加环境变量,在搜索框输入sysdm.cpl,选择高级,环境变量,系统变量,Path添加:

3.系统可以同时存在多个版本的cuda,使用哪个更改环境变量即可。比如我安装了cuda9.2和cuda11.2,因此如果要用cuda9.2,在系统变量的Path中将9.2版本的cuda放在11.2前面:

windows平台FairMOT的实现_第5张图片

另外在系统变量中将CUDA_PATH的路径指向9.2:

windows平台FairMOT的实现_第6张图片

4. visual studio c++ build tools时安装缺失问题:

解决visual studio C++ build tools时安装包缺失或损坏的问题(亲测绝对有用!!!)_visual studio build tools-CSDN博客win10安装visual studio C++ build tools超简单教程(离线适用)_microsoft c++ build tools 离线-CSDN博客

即百度网盘下载:
链接:https://pan.baidu.com/s/1msbwwqLleQ0GagZ4ymKvhQ
提取码:owk0
下载解压后的文件夹如图所示:

然后双击VisualCppBuildTools_Full.exe离线安装。

windows平台FairMOT的实现_第7张图片

windows平台FairMOT的实现_第8张图片

5.不要卸载setuptools,在安装dcn_v2时,出现错误,根据搜索提示,需要安装低版本的setuptools,我于是卸载了setuptools导致每次运行都会报一个文件出错,于是还重新安装了annaconda。变成低版本的setuptools,只需要直接安装即可,例如:pip install setuptools=59.x.x。如果不小心卸载了setuptools参照以下方法解决:

手贱,pip uninstall setuptools报错!!!-CSDN博客

[python]python2.7卸载setuptools导致No module named pkg_resources_python卸载setuptools_FL1623863129的博客-CSDN博客

6. pip下载安装出现SSLError,解决:关掉软件。

[python]python2.7卸载setuptools导致No module named pkg_resources_python卸载setuptools_FL1623863129的博客-CSDN博客

7.conda激活虚拟环境发生错误:CommandNotFoundError: Your shell has not been properly configured to use ‘conda activate‘。

解决:在shell环境下应该要先启动conda的base环境

activate

再激活虚拟环境:

conda activate FairMOT1

参考:CommandNotFoundError: Your shell has not been properly configured to use ‘conda activate‘.简单解决方案_rothschildlhl的博客-CSDN博客

8.关于apex的一些总结:

apex安装问题解决_nvidia apex安装_一根黄三的博客-CSDN博客

Pytorch 安装 APEX 疑难杂症解决方案 - 知乎

CommandNotFoundError: Your shell has not been properly configured to use ‘conda activate‘.简单解决方案_rothschildlhl的博客-CSDN博客

9.关于dcn_v2的一些总结:

解决在windows上DCNv2编译问题(强推!亲测有效!!)_windows编译dcnv2-CSDN博客

10.flat_hash_map.h总结:

没办法用人工智障跳鸡你太美让我很伤心! - 哔哩哔哩

python - flat_hash_map.h : error C3203: 'templated_iterator': unspecialized class template - Stack Overflow

11.python3.8+pytorch1.7+cuda11.2+vs2019参照:

FairMOT配置(VS2019+Win10+CUDA11.0)_importerror: cannot import name '_nt_quote_args' f-CSDN博客

Tracking(目标跟踪):Mot(多目标跟踪系列)-FairMot,快速入门代码、小白都看懂_he_eeeeeeeeeee的博客-CSDN博客

12.其他的FairMOT的总结:

https://www.cnblogs.com/huiyanliu/p/14080785.html

关于在windows上完成目标检测模型Centernet 以及自己数据的训练 - 灰信网(软件开发博客聚合)FairMOT实时多目标跟踪_fairmot dla-CSDN博客关于在windows上完成目标检测模型Centernet 以及自己数据的训练 - 灰信网(软件开发博客聚合)

FairMOT训练自己的数据集及学习笔记_fairmot修改类别数-CSDN博客

解决在windows上DCNv2编译问题(强推!亲测有效!!)_windows编译dcnv2-CSDN博客

13.conda如何添加,删除镜像channel,管理虚拟环境,以及其他常见命令

conda如何添加,删除镜像channel,管理虚拟环境,以及其他常见命令。_conda config --add channels_音程的博客-CSDN博客

你可能感兴趣的:(windows)