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脚本

 

 

 

posted @ 2019-04-06 01:16  bH1pJ  阅读(87)  评论(0编辑  收藏  举报