当前位置: 首页 > 工具软件 > Mask_RCNN > 使用案例 >

mask rcnn算法原理图_Mask_RCNN官方示例算法测试

徐嘉谊
2023-12-01

Mask_RCNN官方示例算法链接

https://github.com/matterport/Mask_RCNN#getting-started

环境:tf_py365 (python3.6.5)

jupyter notebook中运行demo.ipynb

1 安装skimage.io

(安装前运行会报错:ModuleNotFoundError: No module named ‘skimage’)

pip install scikit-image

2 安装imageaug

(安装前运行会报错:ModuleNotFoundError: No module named ‘imgaug’)

pip install imageaug

3 安装cython(python加速工具)

(安装前运行会报错error: Unable to find vcvarsall.bat等)

conda install cython

4 安装VS编译好的pycocotools

(安装前运行报错:ModuleNotFoundError: No module named ‘pycocotools’)

网上搜索VS编译好的pycocotools,下载,放入指定目录下:

(1)python工程目录

(2)anaconda/虚拟环境/lib/site-packages

两种方法都可以from pycocotools.coco import COCO。路径(1)因为是放到工程里,所以这个工程在所有的python3.6、3.5环境下都可以使用pycocotools;路径(2)是放到虚拟环境里,其他工程加载环境就可以直接用。

(python 3.7环境或无法使用)

若不安装编译好的pycocotools,可官网下载源码,在VS2015中编译后使用。(注意:pycocotools编译对VS版本有要求,VS2013以下或不成功)

5 匹配keras版本

匹配前运行会报错:

`AttributeError: module 'keras.engine.topology' has no attribute 'load_weights_from_hdf5_group_by_name'`

原因是版本不匹配。在安装mask_rcnn时,requirements.txt要求的是keras>=2.0.8,默认安装的是keras2.2.2,但是load_weights_from_hdf5_group_by_name只存在于keras2.0.8。

解决方法:

pip uninstall keras

pip install keras==2.0.8

注意:

模型mask_rcnn_coco.h5可能下载不成功,在第一个模块执行时报错:

OSError: Unable to open file (truncated file: eof = 102137856, sblock->base_addr = 0, stored_eof = 257557808)

删除模型文件mask_rcnn_coco.h5,重新下载即可。

参考及链接:

pycocotools包网盘链接:

https://pan.baidu.com/s/1AVyld-kCdujuLwBilhbY8w

提取码:w4qi

 类似资料: