下载文件
下载apex、cocoapi、maskrcnn benchmar文件
链接:https://pan.baidu.com/s/11kxSFkuUjKfnFRx5jRJQ0w
提取码:1n1q
复制这段内容后打开百度网盘手机App,操作更方便哦
创建虚拟环境,并激活
conda create --name maskrcnn_benchmark python=3.7
conda activate maskrcnn_benchmark
这里使用的python3.7版本,因为pytorch-nightly由于之对应的版本。
安装依赖
这按照github的循序依次执行。
conda install ipython
conda install ninja yacs cython matplotlib tqdm opencv
安装pytorch
conda install pytorch=1.2 torchvision=0.2.1 cudatoolkit=10.0
pytorch-nightly使用离线安装。
conda install --offline pytorch-nightly-1.0.0.dev20190328-py3.7_cuda100_cudnn7_0.tar.bz2
替换collect_env.py
将下载好的文件夹中的collect_env.py替换掉Anaconda3\envs\maskrcnn_benchmark\Lib\site-packages\torch\utils路径下的该文件
编译cocoapi
cd cocoapi/PythonAPI
python setup.py build_ext install
编译apex
cd apex
python setup.py install --cuda_ext --cpp_ext
安装maskrcnn-benchmark
cd maskrcnn-benchmark
python setup.py build develop
测试Demo
cd demo
python webcam.py --min-image-size 224