当前位置: 首页 > news >正文

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,看一下是否安装成功;

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

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

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple numpy1.17.0
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple scipy
1.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 cython
0.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-image
0.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 h5py
2.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 ipython
7.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这个文件了。结果如下图:
在这里插入图片描述
在这里插入图片描述

http://www.lryc.cn/news/214164.html

相关文章:

  • 企业级开发命名规范有哪些?
  • sitespeedio.io 前端页面监控安装部署接入influxdb 到grafana
  • ModStartCMS v7.5.0 内外网映射节流,安全使用增强
  • 【LVS实战】02 搭建一个LVS-NAT模式实验
  • Word 将文档中的【第几条】批量加粗
  • 苹果AirTag固件更新
  • 04.Oracle的体系架构
  • 01【保姆级】-GO语言特点和安装使用和hello
  • EVM6678L 开发教程: IBL-TFTP 引导 elf 文件
  • 接口测试 —— Jmeter 之测试片段的应用
  • 接入文心一言实战(一):API申请与测试
  • 518抽奖软件,安全稳定,不怕手抖误按键
  • 【STM32G431RBTx】备战蓝桥杯嵌入式汇总
  • 【抽奖系统】
  • 如何在外SSH远程连接Ubuntu系统【无公网IP】
  • 在防御CC攻击上,高防CDN起着什么作用?
  • 被这7款在线涂鸦画板惊艳到,手残也能画出涂鸦大片!
  • pt权重转onnx记录
  • 瑞数专题五
  • 【Androidstudio学习笔记】设计引导页功能
  • 【python】中位数(暴力+最大最小堆)
  • Avro 如何生成java Bean
  • EG4003-一颗为微波、红外信号放大及处理输出的数模混合芯片
  • kafka生产者源码精华总结
  • 边界缩小维护最值——倒序枚举/中部切开:1101T2
  • vue实现购物车案例
  • 工业4G路由器桥接多网络,提升工业环境网络覆盖
  • docker 存储目录迁移
  • Yolo-Z:改进的YOLOv5用于小目标检测
  • 系列八、Spring IOC有哪些扩展点,在什么时候调用