本次是以: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 configuration
将上面的训练或测试参数粘贴到Parameters
然后点击运行,会报错,这时候要把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
版本是scipy1.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