本次是以: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
版本是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 <module>
    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
Logo

GitCode 天启AI是一款由 GitCode 团队打造的智能助手,基于先进的LLM(大语言模型)与多智能体 Agent 技术构建,致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话,还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力,真正做到“一句话,让 Al帮你完成复杂任务”。

更多推荐