MASKRCNN 采坑经历

第0步：安装anaconda

Anaconda3-2018.12-Windows-x86_64.exe

第一步

git clone https://github.com/matterport/Mask_RCNN.git

第二步

下载

Shapely-1.6.4.post1-cp37-cp37m-win_amd64.whl

后

python install Shapely-1.6.4.post1-cp37-cp37m-win_amd64.whl

第三步

在工程根目录下：python install -r requirements.txt

第四步

pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI

第五步

在工程根目录下：python setup.py install

 

 

最后一步，应该就能运行demo.py文件了

[

如何运行：使用spyder工具

运行脚本即可，在路径下：D:\KerasProject\MaskRCNN\images\...

让如一张图片即可

]

 

 

 

 

2019年4月11日21:26:27补充

接下来就要用我们训练好的模型了，

首先使用以下别人训练好的模型：

python balloon.py splash --image "D:\KerasProject\MaskRCNN\balloon_dataset\balloon\val\24631331976_defa3bb61f_k.jpg" --weights "D:\KerasProject\MaskRCNN\mask_rcnn_balloon.h5"

 

然后用我们自己训练好的模型：

python CTA.py splash --image "D:\KerasProject\MaskRCNN\cta_dataset\train\XXX.jpg" --weights "D:\KerasProject\MaskRCNN\XXX.h5"

 

spyder 如何执行需要命令行参数的脚本

%run CTA.py splash --image "C:\Users\XXX\Desktop\A3.jpg" --weights "D:\KerasProject\MaskRCNN\mask_rcnn_balloon_0030.h5"

 

pycharm中运行脚本时的参数：

使用训练好的模型：

detect --imageDirPath "D:/KerasProject/MaskRCNN/samples/CTA/10totESt22/" --weights "D:/KerasProject/MaskRCNN/mask_rcnn_balloon_0030.h5" --outputPath "D:/KerasProject/MaskRCNN/samples/CTA/10totESt22Output/"

训练模型：

 

train --dataset "D:\KerasProject\MaskRCNN\cta_dataset" --weights "D:\KerasProject\MaskRCNN\mask_rcnn_balloon.h5"

 

 

 

 

2019年5月3日14:50:26

当时将TensorFlow换成TensorFlow-GPU时出现了第一个错误：

没有：keepdims

解决方案：将keepdims换成keep_dims，问题解决，

参考：https://github.com/openai/pixel-cnn/issues/29

接下来遇到了第二个问题：

raise StopIteration() StopIteration

解决方案：暂时未解决， 很有可能是TensorFlow-GPU环境的问题（我把TensorFlow1.4.0卸载了，又装了一个TensorFlow1.13.0问题仍然没有被解决）

参考：https://github.com/amdegroot/ssd.pytorch/issues/214

最终的解决方案：

创建虚拟环境，安装以下的内容

TensorFlow1.4.0
Cuda8.0
Cudnn6.0
Keras2.1.0

即可成功使用TensorFlow-GPU

 

 

2019年5月24日10:39:48

问题：如何使用MaskRCNN，在pycharm中直接使用已经训练好的模型？

解决：配置好环境，从第一步骤到第五步骤；然后用spyder运行demo脚本