配置信息:
ubuntu16.04
python3.6.5
tensorflow1.4.1
keras2.1.5
cuda8
cudnn6
利用jupyter
打开Terminal,输入如下命令来启动Anaconda-Navigator图形化界面:
anaconda-navigator
然后Launch“jupyter notebook”,然后打开samples文件夹下的demo.ipynb文件。点击Run,就可以逐块(cell)运行,每个cell上面都有文字说明
错误1
运行第一块,出现如下错误:提示没有“pycocotools”
解决方法:
在这里cocoapi下载cocoapi,解压,进入到cocoapi-master/PythonAPI目录下,打开Terminal,输入:
make
然后把cocoapi-master/PythonAPI目录下的pycocotools拷贝到maskrcnn根目录下。
然后会提示成功下载了权重文件mask_rcnn_coco.h5:
Downloading pretrained model to /home/xxxxx/python/Mask_RCNN-master/mask_rcnn_coco.h5 ...
... done downloading pretrained model!
然后就没有别的错误了,能输出随机一张图片的检测结果:
参考资料:
利用pycharm
在pycharm中,打开maskrcnn文件夹,然后在samples文件夹下新建一个demo.py的文件。
然后把demo.ipynb的代码拷贝过去,然后运行。
错误1
解决方法%matplotlib inline是内置在jupyter中的函数,在pycharm中不支持该语句。
解决办法是注释掉%matplotlib inline,然后在代码的最后加上:plt.show()。
参考资料: