MaskRCNN(matterport)模型搭建与实验

本次是以:https://github.com/matterport/Mask_RCNN为基础进行的,这是一个公开的MaskRCNN代码,基于tensorflow1.3.0,由于比较老,所以在环境搭建是容易出现很多问题。
1.使用conda创建一个python3.6的环境

conda create --name tensorflow_gpu1.3.0 python=3.6

2.进入项目根目录,安装相关依赖,这里直接使用:

pip3 install -r requirements.txt

3.运行安装程序:

python3 setup.py install

这里会产生一些报错,但是可以不用管
4.然后可以尝试运行下samples文件下的ballon.py,接下来是一大堆报错。首先设置运行参数:

训练配置参数
train
--dataset=../../balloon
--weights=coco

测试配置参数
splash 
--weights=./logs/balloon20191023T1451/mask_rcnn_balloon_0030.h5
--image=../../balloon/val/14898532020_ba6199dd22_k.jpg

5.右击选择modify Run configurationMaskRCNN(matterport)模型搭建与实验_第1张图片
将上面的训练或测试参数粘贴到Parameters
MaskRCNN(matterport)模型搭建与实验_第2张图片
然后点击运行,会报错,这时候要把tensorflow和gpu改成requirements.txt里要求的最低版本,即tensorflow1.3.0,keras2.0.8

然后再次运行还会报错,需要找到topology.py,将decode删除

下一步应该是utils.py文件报错:module ‘scipy.misc’ has no attribute ‘imresize’,降低scipy版本: pip install scipy1.2.1
版本是scipy
1.2.1时,如果还报错,那就安装Pillow即可: pip install pillow

python import cv2报错
(1) 错误

 import cv2
  File "/usr/local/lib/python3.8/dist-packages/cv2/__init__.py", line 181, in 
    bootstrap()
  File "/usr/local/lib/python3.8/dist-packages/cv2/__init__.py", line 153, in bootstrap
    native_module = importlib.import_module("cv2")
  File "/usr/lib/python3.8/importlib/__init__.py", line 127, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
ImportError: libGL.so.1: cannot open shared object file: No such file or directory

解决方法:

sudo apt update
sudo apt install libgl1-mesa-glx

你可能感兴趣的:(人工智能,python,tensorflow,开发语言)