Mask-RCNN自动获取训练集中图像的长度和宽度，然后用于训练。

一、目前情况

用Mask-RCNN训练自己的数据集时，需要制定图片的长度和宽度，即

IMAGE_MIN_DIM = 448
IMAGE_MAX_DIM = 640

而在Mask_RCNN/mrcnn目录下model.py文件中第1815行到1819行代码

h, w = config.IMAGE_SHAPE[:2]if h / 2**6 != int(h / 2**6) or w / 2**6 != int(w / 2**6):raise Exception("Image size must be dividable by 2 at least 6 times ""to avoid fractions when downscaling and upscaling.""For example, use 256, 320, 384, 448, 512, ... etc. ")

需要将图像处理成指定长宽比例的图像然后才可以用于训练，并且训练集中的图像需要长度和宽度都需一致。

若训练集中的图像有长度和宽度不同时则不能训练，这样极不方便。

二、更改Mask-RCNN代码

在训练数据集的代码train_shapes.ipynb中，在load_shapes()中添加更改代码：

for i in range(count):# 获取图片宽和高filestr = imglist[

用Mask-RCNN训练自定义大小的数据集相关推荐

Mask Rcnn训练自己的航拍数据集
最近想利用Mask Rcnn训练自己的航拍数据(数据来源:自己采集.飞行高度:200m,图片大小WxH=6000x4000),在进行自己的航拍图片训练时也与其他同学进行交流,他们一般进行识别的是一类物 ...
mask rcnn训练自己的数据集
原文首发于微信公众号「3D视觉工坊」--mask rcnn训练自己的数据集前言最近迷上了mask rcnn,也是由于自己工作需要吧,特意研究了其源代码,并基于自己的数据进行训练~ 本博客参考:ht ...
Mask R-CNN训练日记
Mask R-CNN训练集的踩坑日记写在前面 Error1 Error2 Error3 Error4 Error5 Error6 写在结尾写在前面今天代码终于可以跑了,虽然训练的损失率很高,但是 ...
运行mask rcnn训练代码程序在Epoch 1/20时出现程序挂起现象解决办法
问题描述在运行mask rcnn训练代码程序在Epoch 1/20时出现程序挂起现象,即程序运行但是跑不动,也不报错.如下所示: 解决办法将keras版本修正为2.1.6即可,打开cmd,输入下 ...
Mask R-CNN 训练自己的数据集（balloon过程+报错解释）
因项目需要,识别带有多边形标注的图像,舍弃了速度快精度高的yolov3,使用Mask R-CNN网络.作为一名深度学习小白,在摸爬滚打中查找资料修改代码以及不断地调整训练集,途中踩了不少坑,终于达到预 ...
从三个小时到25分钟：我们使用Apache MXNet优化Mask R-CNN训练时间全过程记录
总览在过去的几年中,深度神经网络(DNN)模型的规模和复杂性急剧增加.一方面,这些复杂的模型提高了模型在各种领域(例如计算机视觉,自然语言理解等)的准确性的门槛:另一方面,模型参数的庞大规模及其梯度 ...
在Colab平台上用mmdetection中的Cascade RCNN训练自己的voc数据集
最近在学习目标检测,要用Cascde RCNN训练自己的数据集,因为本地电脑没有GPU,所以就借用Colab云端来进行模型训练,想要记录一下操作过程,以及希望可以给更多电脑没有GPU,但是需要用mmd ...
Mask R-CNN 训练自己的数据集—踩坑与填坑
作者 | Raini 出品 | 北京图特摩斯科技有限公司(thutmose.cn) tensorflow/keras版(项目地址https://github.com/matterport). 服务器c ...
解决Mask RCNN训练时GPU内存溢出问题
首先自己是个小白对于如何使用GPU跑程序这个学习了一下: (1)使用的是putty,安装了Anaconda这个IDE环境,用的conda install tensorflow-gpu安装的GPU版本t ...

用Mask-RCNN训练自定义大小的数据集

一、目前情况

二、更改Mask-RCNN代码

用Mask-RCNN训练自定义大小的数据集相关推荐

最新文章

热门文章