MASK-RCNN tensorflow环境搭建

此教程默认你已经安装了Anaconda,且tensorflow 为cpu版本。为什么不用gpu版本,原因下面解释。

此教程默认你已经安装了Anaconda。

因为tensorflow2.1后的gpu版·,不支持windows。并且·只有高版本的tensorflow才对应我的CUDA12.2;

而且,我之前安装了pytorch跑tolov8,cuda都很高。安装tensorflow-gpu的话,需要重新安装我的cuda,感觉太麻烦了。过程中一直遇到各种错误。索性直接放弃gpu版本了。

第一步,安装CUDA和CUDnn。

window+r,进入命令行,输入nvidia-smi.显示信息如下:

所以我安装的CUDA是12.2版本的。

进入CUDA安装路径,查看CUDnn 版本:

我电脑为win10.显卡为3060,CUDA、CUDnn版本如上图所示。安装自己电脑对应的CUDA和CUDnn。pycharm使用的版本是:PyCharm 2023.2.3。CSDN上也有对应CUDA下载教程。

在Anaconda中创建虚拟环境

conda create -n maskrcnn python=3.6

激活并进入环境

conda activate maskrcnn

安装tensorflow,使用清华镜像安装:

pip install tensorflow==1.5.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

安装之后,进入python,看一下是否安装成功;

没有报错,至于出现下面那一推东西,不用管他。能用就行。

安装其他依赖包,一个一个复制安装:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple numpy1.17.0
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple scipy1.2.1

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pillow8.4.0
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple cython0.29.28

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple matplotlib3.3.4
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple scikit-image0.17.2

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-python4.3.0.38
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple h5py2.10.0

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple imgaug0.4.0
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple ipython7.16.3

到这里tensorflow就算安装完成了。下一步需要下载官方的MASKRCNN代码包。

地址:https://gitcode.com/mirrors/matterport/mask_rcnn/tree/master

除了代码包以外,还需要下载他的预训练权重和数据集。我数据集下载的是气球数据集。

找不到权重文件,数据集,源码的小伙伴也可以在我分享的网盘里下载。

百度网盘链接:

链接:https://pan.baidu.com/s/1N5VRfzuvdNGjvJthbq9CJA

提取码:2222

下载完后,把xmaskrcnn放到自己电脑盘中,最好不要有中文。

把数据集,权重文件解压到目录下:

我把整个文件都打包上传,读者就不需要解压,移动了。直接下一步

pycharm打开源码文件

设置python解释器,就是刚刚创建的那个Anaconda环境下的maskrcnn,里面有个python.exe文件:

设置->项目:MaskRCNN->python解释器:添加解释器

选择完后,自己可以新建一个test.py,看看tensorflow能不能用。

找到下面这个ballon.py文件,并且在ballon文件夹下面新建一个logs文件夹。

里面代码修改如下(只修改这部分,里面的路径根据自己实际情况改):

编辑配置


点开之后,输入:

点击应用,确定。

就可以运行ballon.py这个文件了。结果如下图:

相关推荐
兵慌码乱10 小时前
面向桌面端的资产管理系统分层架构设计与核心模块实现
python·系统架构·sqlite·pyqt5·数据库设计·桌面应用开发·mvc架构
hboot12 小时前
AI工程师第三课 - 机器学习基础
python·scikit-learn·kaggle
顾林海16 小时前
Agent入门阶段-编程基础-Python:流程控制
python·agent·ai编程
呱呱复呱呱19 小时前
Django CBV 源码解读:一个请求是怎么找到你的 get() 方法的
python·django
曲幽1 天前
刚部署的 LibreTranslate 频频翻车?我掏出了 20 年前的 StarDict 词典,用 FastAPI 搭了个本地词典翻译 API
python·fastapi·web·translate·goldendict·libretranslate·stardict·pystardict
荣码1 天前
用Streamlit给AI应用套个界面,10行代码出Web页面
java·python
兵慌码乱1 天前
基于Python+PyQt5+SQLite的药房管理系统实现:事务一致性与界面解耦全流程解析
python·sqlite·信号与槽·pyqt5·数据库设计·桌面应用开发·事务处理
金銀銅鐵1 天前
[Python] 体验用欧几里得算法计算最大公约数的过程
python·数学
FreakStudio2 天前
W55MH32L-EVB 上手测评:硬件 TCP/IP 加持的以太网单片机,MicroPython 零门槛开发
python·单片机·嵌入式·大学生·面向对象·并行计算·电子diy·电子计算机
用户0332126663672 天前
使用 Python 从零创建 Word 文档
python