环境:
windows10,CUDA10.0,cudnn7.6.5(使用TensorFlow官方推荐会报错),pycharm。
其余环境配置参考以下博文:windows下MaskRCNN环境配置【保姆级教程--附各个依赖库版本安装】_岳一岁的博客-CSDN博客_maskrcnn环境配置
conda create -n MaskRCNN python=3.6 #创建环境
conda activate MaskCNN #激活环境
python -m ensurepip
python -m pip install --upgrade pip #升级pip
pip install numpy==1.17.0
pip install scipy==1.2.1
pip install Pillow==8.4.0
pip install cython==0.29.28
pip install matplotlib==3.3.4
pip install scikit-image==0.17.2
pip install keras==2.1.6
pip install opencv-python
pip install h5py==2.10.0
pip install imgaug==0.4.0
pip install ipython==7.16.3
pip install tensorflow-gpu==1.15.0
注意安装cocoapi容易安装失败,参考这篇文章:
Win10下安装pycocotools_qq_29750461的博客-CSDN博客
使用运行2.1的demo程序,出现不显示图片问题,解决方法参考Mask R-CNN:UserWarning: Matplotlib is currently using agg, which is a non-GUI backend, so cannot sho_WaitFoF的博客-CSDN博客
#终端中py36下
import matplotlib
matplotlib.matplotlib_fname() #查找设置matplotlib的终端类型的文件位置
根据位置寻找文件,使用记事本打开。
在pycharm中重新载入,在源码的基础上进行修改,如下图:
最后再运行demo.py文件,完结撒花 。
附修改后的代码包:
链接:https://pan.baidu.com/s/1p_aJrBrfkDBsOfXeLyGZqw?pwd=5ojp
提取码:5ojp
conda离线环境包:
https://download.csdn.net/download/Ningmoua/87264424
环境包载入教程:
以上是官网教程,亲测可用。