从零开始的maskrcnn

采坑开始

重要提示：本次实验尝试了教程中的cuda9.0版本 坑太多
因此最终还是改为cuda10.1进行试验

安装环境

conda create --name maskrcnn
conda activate maskrcnn

conda install ipython
pip install ninja yacs cython matplotlib tqdm opencv-python

conda install -c pytorch torchvision cudatoolkit=10.1

安装COCO

git clone https://github.com/cocodataset/cocoapi.git
cd cocoapi/PythonAPI
python setup.py build_ext install

安装APEX

git clone https://github.com/NVIDIA/apex.git
cd apex
python setup.py install --cuda_ext --cpp_ext

安装主体

git clone https://github.com/facebookresearch/maskrcnn-benchmark.git
cd maskrcnn-benchmark
python setup.py build develop

CV2错误

Traceback (most recent call last):
  File "F:/DL/maskrcnn-benchmark-master/demo/predictor.py", line 2, in <module>
    import cv2
  File "D:\ProgramData\Anaconda3\envs\maskrcnn\lib\site-packages\cv2\__init__.py", line 5, in <module>
    from .cv2 import *
ModuleNotFoundError: No module named 'cv2.cv2'

解决方案：降低opencv-python版本 4.4.0.42→4.1.0.25

Matplotlib错误

Traceback (most recent call last):
  File "F:/DL/maskrcnn-benchmark-master/demo/predictor.py", line 413, in <module>
    import matplotlib.pyplot as plt
  File "D:\ProgramData\Anaconda3\envs\maskrcnn\lib\site-packages\matplotlib\__init__.py", line 174, in <module>
    _check_versions()
  File "D:\ProgramData\Anaconda3\envs\maskrcnn\lib\site-packages\matplotlib\__init__.py", line 159, in _check_versions
    from . import ft2font
ImportError: cannot import name 'ft2font' from 'matplotlib' (D:\ProgramData\Anaconda3\envs\maskrcnn\lib\site-packages\matplotlib\__init__.py)

解决方案：降低matplotlib版本 3.3.1→3.0.3

付conda环境

后续步骤 参考