新闻详情

新闻详情

首页 / 资讯中心 / 详情

【亲测通过】MaskRcnn_tf1.x如何升级到MaskRcnn_tf2.x,实现RTX3090环境训练自定义数据集模型。

发布时间:2026/10/1 9:28:49来源:尧图网络
【亲测通过】MaskRcnn_tf1.x如何升级到MaskRcnn_tf2.x,实现RTX3090环境训练自定义数据集模型。
一、背景:之前一篇博文中已经实现了maskrcnn_tf1.15.0环境的win10cpu模型训练但cpu训练实在是非常的耗时据说tf1.x是支持RTX1060的本人未测试但不支持最新的RTX3090查阅了很多资料原因应该是tf1.x与tf2.x的差别比较大必须升级到tf2.x才可以正常使用rtx3090。下面是maskrcnn_tf1.15.0的开发案例本人亲测可用。mask_rcnn 训练自定义数据集本地win10系统cpu已调通采坑无数均已列出解决方法_君临天下tjm的博客-CSDN博客源代码地址https://github.com/junlintianxiatjm/Mask_RCNN-master0071.版本信息python 3.6.9Tensorflow 1.15.0keras 2.2.5Pillow 5.3.0必须否则labelme执行json_to_dataset会出错Cv2必须安装训练模型时用到Wraptopt_einsumGastscikit-imageIPython虚拟环境完整配置库如下..https://blog.csdn.net/shanxiderenheni/article/details/118832615二、maskrcnn_tf1升级到maskrcnn_tf2的解决方法1.安装必要的包用镜像安装速度更快格式如pip install -i https://pypi.douban.com/simple/ tensorflow主要的pip listtensorflow 2.6.0keras 2.6.0matplotlib 3.2.2h5py 3.1.0numpy 1.19.5scikit-image 0.16.2tensorflow-gpu 2.6.2opencv-python 4.5.4.60详细pip list如下(py36_maskrcnn_env_bak) C:\Users\dellpip list Package Version ------------------------ ------------------- absl-py 0.15.0 astor 0.8.1 astunparse 1.6.3 backcall 0.2.0 bleach 1.5.0 cached-property 1.5.2 cachetools 4.2.4 certifi 2020.6.20 charset-normalizer 2.0.9 clang 5.0 colorama 0.4.4 cycler 0.11.0 Cython 0.29.28 dataclasses 0.8 decorator 4.4.2 flatbuffers 1.12 gast 0.4.0 google-auth 1.35.0 google-auth-oauthlib 0.4.6 google-pasta 0.2.0 grpcio 1.44.0 h5py 3.1.0 html5lib 0.9999999 idna 3.3 imageio 2.13.5 imgaug 0.4.0 imgviz 1.4.1 importlib-metadata 4.8.3 ipython 7.16.2 ipython-genutils 0.2.0 jedi 0.17.2 keras 2.6.0 Keras-Applications 1.0.8 Keras-Preprocessing 1.1.2 kiwisolver 1.3.1 labelme 3.16.2 libclang 11.1.0 Markdown 3.3.6 matplotlib 3.2.2 mock 4.0.3 networkx 2.5.1 numpy 1.19.5 nvidia-pyindex 1.0.9 oauthlib 3.1.1 opencv-python 4.5.4.60 opt-einsum 3.3.0 packaging 21.3 parso 0.7.1 pickleshare 0.7.5 Pillow 8.3.2 pip 21.2.2 prompt-toolkit 3.0.24 protobuf 3.17.3 pyasn1 0.4.8 pyasn1-modules 0.2.8 Pygments 2.10.0 pyparsing 3.0.6 PyQt5 5.15.6 PyQt5-Qt5 5.15.2 PyQt5-sip 12.9.0 python-dateutil 2.8.2 PyWavelets 1.1.1 PyYAML 6.0 QtPy 1.9.0 requests 2.26.0 requests-oauthlib 1.3.0 rsa 4.8 scikit-image 0.16.2 scipy 1.4.1 setuptools 58.0.4 Shapely 1.8.1.post1 six 1.15.0 tb-nightly 2.6.0a20210806 tensorboard 2.6.0 tensorboard-data-server 0.6.1 tensorboard-plugin-wit 1.8.1 tensorflow 2.6.0 tensorflow-estimator 2.6.0 tensorflow-gpu 2.6.2 tensorflow-gpu-estimator 2.2.0 termcolor 1.1.0 tf-estimator-nightly 2.7.0.dev2021092408 tifffile 2020.9.3 traitlets 4.3.3 typing-extensions 3.7.4.3 urllib3 1.26.7 wcwidth 0.2.5 Werkzeug 2.0.2 wheel 0.37.0 wincertstore 0.2 wrapt 1.12.1 zipp 3.6.02.用新mrcnn替换原mrcnn路径下的全部代码文件mrcnn 替换掉官方的mrcnn文件夹即可开箱使用新mrcnn代码路径如下https://github.com/junlintianxiatjm/MaskRCNN_TF2https://github.com/junlintianxiatjm/MaskRCNN_TF2注意调小Batch size即调小IMAGES_PER_GPU,本项目测试是8太大会提示内存耗尽。错误信息如下2022-03-11 11:02:03.235518: I tensorflow/stream_executor/cuda/cuda_dnn.cc:369] Loaded cuDNN version 8200 2022-03-11 11:02:09.947906: W tensorflow/core/common_runtime/bfc_allocator.cc:272] Allocator (GPU_0_bfc) ran out of memory trying to allocate 2.40GiB with freed_by_count0. The caller indicates that this is not a failure, but may mean that there could be performance gains if more memory were available.设置 train.py里面的 “IMAGES_PER_GPU 8”或者更新值即可解决。gpu rtx3090训练效果如下
网站建设高端定制企业官网
RELATED

相关资讯

更多精彩内容,欢迎继续阅读

较早相关资讯

最新相关资讯

第一次给自己定目标试试。 2026/10/1 10:17:17

第一次给自己定目标试试。

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

阅读更多 →
8年Java后端忠告:CRUD已卷死,大模型才是后端工程师的第二条黄金曲线 2026/10/1 10:17:11

8年Java后端忠告:CRUD已卷死,大模型才是后端工程师的第二条黄金曲线

先把核心结论说透: 不是 Java 后端不行了,是纯 CRUD 后端的路,已经越走越窄、越走越卷、越走越贬值; 而大模型应用开发,是未来 5–10 年,所有后端工程师最稳、薪资最高、缺口最大的职业第二条增长曲线。 深…

阅读更多 →
流式输出(Streaming)是OpenAI API的核心能力之一,其本质是基于SSE(服务器推送事件)协议的分块传输机制 2026/10/1 10:17:10

流式输出(Streaming)是OpenAI API的核心能力之一,其本质是基于SSE(服务器推送事件)协议的分块传输机制

流式输出(Streaming)是OpenAI API的核心能力之一,其本质是基于SSE(服务器推送事件)协议的分块传输机制。大模型生成文本遵循自回归逻辑:每次仅预测下一个token(词元),而非…

阅读更多 →
2026 年版|别再混淆 AI Agent 与 Agentic AI!概念、架构、落地场景全解析 2026/10/1 10:17:04

2026 年版|别再混淆 AI Agent 与 Agentic AI!概念、架构、落地场景全解析

本文详细解析了AI Agent和Agentic AI的概念、联系与区别。AI Agent是指能够感知环境、自主决策并采取行动的智能系统;而Agentic AI则更强调系统整体围绕复杂目标自主行动的能力,通常涉及多智能体协作、跨系统编排等。文章还讨论了两者在现实中的应用场景…

阅读更多 →
字节跳动职位分析:AI数据运营(PGC方向)-TikTok 2026/10/1 10:16:51

字节跳动职位分析:AI数据运营(PGC方向)-TikTok

一、职位概述该职位隶属于字节跳动 TikTok 业务线,聚焦 PGC(专业生产内容)方向的数据运营工作,核心职责是主导数据标注项目的端到端管理,确保交付质量与效率,并推动流程优化与自动化建设。职位兼具项目管理…

阅读更多 →
物联网卡丢包导致传感器数据缺失?从链路排查到本地缓存重传的完整实战 2026/10/1 10:16:51

物联网卡丢包导致传感器数据缺失?从链路排查到本地缓存重传的完整实战

1. 问题现场:传感器数据为什么总在"最后一公里"消失做过工业物联网项目的人,大概率都遇到过这种让人抓狂的场景:现场传感器明明在正常采集,PLC 上的数值也在跳动,但传到云端或者本地服务器之后,数…

阅读更多 →

今日资讯

本周资讯

本月资讯

看完文章仍有疑问?

联系尧图顾问,获取一对一建站咨询

立即免费咨询 📞 400-888-8888
📞 ✉