Mask RCNN Demo之遇到的问题及解决方案

互相交流的QQ:3239516597

Mask RCNN的代码已经有大神在gitHub上公布,而且facebook官方也已经公布了源码。我这里用的是github上的基于tensorflow和keras的Mask RCNN。网址:点这里
下面就说一说在跑第一个简单的Demo时遇到的问题,供大家参考。首先是tensorflow+keras的环境配置,这里就不多说了,具体步骤的我会在后面的博客贴出来。下面就来说说具体遇到的问题:
(1)提示找不到imgaug模块:no module named ‘imgaug’
Mask RCNN Demo之遇到的问题及解决方案_第1张图片
找不到那么就安装呗,pip install imgaug,结果安装失败,提示缺少shapely模块,于是到网站下载此模块,网址:https://www.lfd.uci.edu/~gohlke/pythonlibs/#shapely,这是一个非常好的扩展包非正式网址,建议大家收藏起来。找到相应的模块,下载下来,下载完成后,直接把他放到相应的文件夹下,cd进到里面,敲如图所示的代码安装。
Mask RCNN Demo之遇到的问题及解决方案_第2张图片
Mask RCNN Demo之遇到的问题及解决方案_第3张图片
这个安装完成后,直接pip install imgaug即可。
(2)出现ImportError: No module named cv2
你安装pip install cv2会提示你找不到相应的版本。应该改成pip install opencv-python即可。
(3)no module named pycocotools报错
Mask RCNN Demo之遇到的问题及解决方案_第4张图片
这个是因为要用到coco数据集,是一个第三方python库GitHUb(https://github.com/philferriere/cocoapi)上的解决方案是直接pip安装,代码是:pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
但是我试了一下,对我并不管用,报错cannot find command git,我没有找到解决方法。于是把github上的库文件下载下来,进入pythonAPI文件夹中,敲入代码:
python setup.py build_ext --inplace
python setup.py build_ext install
完美安装。。。。。。
(4)出现错误softmax() got an unexpected keyword argument ‘axis’
Mask RCNN Demo之遇到的问题及解决方案_第5张图片
这个错误一般是tensorflow版本和keras版本不匹配,查看tensorflow和keras版本,找到对应的版本即可解决问题,我的tensorflow是1.4,之前的keras是2.2.4.版本,可能版本太高可,我换成了2.1.3.版本没有问题。
最终所有问题都解决到了,欣赏一下结果。
Mask RCNN Demo之遇到的问题及解决方案_第6张图片

你可能感兴趣的:(人工智能)