一种基于改进Unet网络模型的辗环分割定位方法技术

本发明专利技术公开了一种基于改进Unet网络模型的辗环分割定位方法，涉及图片处理技术领域，包括以下步骤：制作碾环图片的数据集；将Unet网络的原始编码部分替换为resnet50残差网络模块，在Unet网络的解码网络输出与最终输出卷积之间加入空洞卷积模块，构建改进Unet网络模型；通过数据集对改进Unet网络模型进行训练；使用训练后的改进Unet网络模型对待测碾环图片进行预测分割，对碾环进行定位。本发明专利技术的定位方法通过CCD相机采集视频，接着逐帧分解为图片。对于此图片，放入编码部分由resnet50代替、输出全连接层加入空洞卷积模块的Unet分割网络对其定位，定位实时性好以及定位精度高。定位实时性好以及定位精度高。定位实时性好以及定位精度高。

【技术实现步骤摘要】
一种基于改进Unet网络模型的辗环分割定位方法


[0001]本专利技术涉及辗环定位
，特别是涉及一种基于改进Unet网络模型的辗环分割定位方法。

技术介绍

[0002]大型锻件是国家重型制造装备和重大工程建设所必需的关键和核心基础部件，对于重工业的尺寸检测前应对其定位，目前大多还处于人工定位检测的阶段，如环型锻件的锻造检测过程中，通常需要人工找准目标再执尺子测量，存在人为的判断误差。另一方面，热锻件所处的检测环境温度高且定位速度较慢。
[0003]目前，国内外大型环件的定位方法，依据与被定位对象的接触方式进行分类，可以分为接触式和非接触式。接触式定位法通过点对点方式，能及时对目标物定位，但误差大、精度低，而且对操作人员的伤害较大，不提倡。非接触式定位法则有效地避免了操作人员与热态环件的直接接触，但是现有的非接触式的设备造价昂贵，导致工厂成本增加，且存在定位不及时的问题。

技术实现思路

[0004]本专利技术实施例提供了一种基于改进Unet网络模型的辗环分割定位方法，可以解决现有技术中存在的问题。
[0005本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于改进Unet网络模型的辗环分割定位方法，其特征在于，包括以下步骤：制作碾环图片的数据集；将Unet网络的原始编码部分替换为resnet50残差网络模块，在Unet网络的解码网络输出与最终输出卷积之间加入空洞卷积模块，构建改进Unet网络模型；通过数据集对改进Unet网络模型进行训练；使用训练后的改进Unet网络模型对待测碾环图片进行预测分割，对碾环进行定位。2.如权利要求1所述的一种基于改进Unet网络模型的辗环分割定位方法，其特征在于，所述制作碾环图片的数据集包括以下步骤：通过CCD工业相机采集辗环锻造过程的视频；对所述视频间隔若干帧提取多张碾环图片；将多张所述碾环图片通过缩放尺寸、遮挡、旋转、调色和扩充手段进行处理得到数据集。3.如权利要求2所述的一种基于改进Unet网络模型的辗环分割定位方法，其特征在于，将数据集通过2：8的方式随机分配成验证集和训练集，通过训练集对改进Unet网络模型进行训练，通过验证集对训练后的改进Unet网络模型进行验证。4.如权利要求1所述的一种基于改进Unet网络模型的辗环分割定位方法，其特征在于，所述resnet50残差主网络主要由四个残差结构，每个残差结构按顺序分别有3、4、6、3个残差块且结构不同，每个残差块都有三层卷积；同类型的残差块顺序链接，每个残差结构的残差块的第一个卷积层的核已固定为64、128、256、512。5.如权利要求4所述的一种基于改进Unet网络模型的辗环分割定位方法，其特征在于，在残差块中前两次次卷积后面会直接加上归一化函数和激活函数，最后一次卷积得出结构和输出相加后再加上归一化函数和激活函数。6.如权利要求4所述的一种基于改进Unet网络模型的辗环分...

【专利技术属性】
技术研发人员：黄志开，邹家豪，陈佳磊，王欢，刘宇浩，虞超凡，孙佳思，席贤昌，甯聆怡，王振宁，
申请(专利权)人：南昌工程学院，
类型：发明
国别省市：