CN114693908B - 一种基于YOLOv5的工业二维码定位方法 - Google Patents
一种基于YOLOv5的工业二维码定位方法 Download PDFInfo
- Publication number
- CN114693908B CN114693908B CN202210228300.XA CN202210228300A CN114693908B CN 114693908 B CN114693908 B CN 114693908B CN 202210228300 A CN202210228300 A CN 202210228300A CN 114693908 B CN114693908 B CN 114693908B
- Authority
- CN
- China
- Prior art keywords
- dimensional code
- picture
- industrial
- yolov
- qrbox
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Active
Links
- 238000000034 method Methods 0.000 title claims abstract description 27
- 238000012549 training Methods 0.000 claims abstract description 42
- 238000012360 testing method Methods 0.000 claims abstract description 28
- 238000013528 artificial neural network Methods 0.000 claims abstract description 15
- 238000012545 processing Methods 0.000 claims abstract description 12
- 230000000694 effects Effects 0.000 claims abstract description 6
- 238000007781 pre-processing Methods 0.000 claims abstract description 4
- 238000013519 translation Methods 0.000 claims description 7
- 230000004927 fusion Effects 0.000 claims description 6
- 238000013507 mapping Methods 0.000 claims description 6
- 238000007796 conventional method Methods 0.000 claims description 3
- 238000010008 shearing Methods 0.000 claims description 2
- 230000004807 localization Effects 0.000 abstract description 3
- 230000006870 function Effects 0.000 description 9
- 238000010586 diagram Methods 0.000 description 3
- 238000013473 artificial intelligence Methods 0.000 description 2
- 238000011161 development Methods 0.000 description 2
- 238000010606 normalization Methods 0.000 description 2
- 230000009286 beneficial effect Effects 0.000 description 1
- 230000015572 biosynthetic process Effects 0.000 description 1
- 238000006243 chemical reaction Methods 0.000 description 1
- 230000007547 defect Effects 0.000 description 1
- 238000001514 detection method Methods 0.000 description 1
- 238000005516 engineering process Methods 0.000 description 1
- 238000013209 evaluation strategy Methods 0.000 description 1
- 238000010801 machine learning Methods 0.000 description 1
- 238000003672 processing method Methods 0.000 description 1
- 238000003786 synthesis reaction Methods 0.000 description 1
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/25—Fusion techniques
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- Y—GENERAL TAGGING OF NEW TECHNOLOGICAL DEVELOPMENTS; GENERAL TAGGING OF CROSS-SECTIONAL TECHNOLOGIES SPANNING OVER SEVERAL SECTIONS OF THE IPC; TECHNICAL SUBJECTS COVERED BY FORMER USPC CROSS-REFERENCE ART COLLECTIONS [XRACs] AND DIGESTS
- Y02—TECHNOLOGIES OR APPLICATIONS FOR MITIGATION OR ADAPTATION AGAINST CLIMATE CHANGE
- Y02P—CLIMATE CHANGE MITIGATION TECHNOLOGIES IN THE PRODUCTION OR PROCESSING OF GOODS
- Y02P90/00—Enabling technologies with a potential contribution to greenhouse gas [GHG] emissions mitigation
- Y02P90/30—Computing systems specially adapted for manufacturing
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- Life Sciences & Earth Sciences (AREA)
- Artificial Intelligence (AREA)
- General Physics & Mathematics (AREA)
- General Engineering & Computer Science (AREA)
- Evolutionary Computation (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Computational Linguistics (AREA)
- Bioinformatics & Computational Biology (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Health & Medical Sciences (AREA)
- Biomedical Technology (AREA)
- Biophysics (AREA)
- Evolutionary Biology (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Computing Systems (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Image Processing (AREA)
- Image Analysis (AREA)
Abstract
本发明公开了一种基于YOLOv5的工业二维码定位方法,方法包括:对采集的照片进行预处理,得到处理后二维码图片,然后利用标签生成器,生成对应的标签;对标签中的二维码图片进行数据增强处理,在QRBOX_YOLOv5工业二维码定位器中,利用数据增强后的图片对构建的qrbox_yolov5s神经网络进行训练生成权重并测试,将测试效果最好的训练权重作为训练好的QRBOX_YOLOv5工业二维码定位器;将待识别的二维码图片数据输入QRBOX_YOLOv5工业二维码定位器,对待定位的二维码进行实时推理,获得定位结果。本发明构建的QRBOX_YOLOv5工业二维码定位器网络结构可以在工业环境中实现对二维码的实时精确定位。
Description
技术领域
本发明涉及人工智能技术领域,特别涉及一种基于YOLOv5的工业二维码定位方法。
背景技术
人工智能是研究开发能够模拟、延伸和扩展人类智能的理论、方法、技术及应用系统的一门新的技术科学,研究目的是促使智能机器会听(语音识别、机器翻译等)、会看(图像识别、文字识别等)、会说(语音合成、人机对话等)、会思考(人机对弈、定理证明等)、会学习(机器学习、知识表示等)、会行动(机器人、自动驾驶汽车等)。现有的二维码定位是在低资源的芯片上实现的二维码定位,采用的模型简单、训练策略和评价策略单一,使二维码定位的准确度较低;现有的二维码定位方法采用粗定位之后进行精定位的两阶段定位方法,不能达到端到端的定位效果,不能满足工业级的需求。
发明内容
本申请提供了一种基于YOLOv5的工业二维码定位方法,旨在解决上述背景技术中,现有技术的缺陷所带来的技术问题中的至少一种。
为解决上述技术问题,本发明的技术方案如下:
本发明提供一种基于YOLOv5的工业二维码定位方法,包括:
S1:采集包含二维码图片的照片,对照片进行预处理,得到处理后二维码图片;
S2:利用标签生成器,为处理后的二维码图片生成对应的标签;
S3:对标签中的二维码图片进行数据增强处理,获得增强后的图片的数据集;
S4:在QRBOX_YOLOv5工业二维码定位器中,利用数据增强后的图片对构建的qrbox_yolov5s神经网络进行训练生成权重并测试,将测试效果最好的训练权重作为训练好的QRBOX_YOLOv5工业二维码定位器;
S5:将待识别的二维码图片数据输入QRBOX_YOLOv5工业二维码定位器,对待定位的二维码进行实时推理,获得定位结果。
优选地,所述步骤S1中,采集包含二维码图片的照片,具体包括:拍摄二维码图片时工业场景作为图片的背景,模拟实际的工业应用场景进行拍摄,根据实际的工业应用场景读取的二维码图片时的角度和距离拍摄;根据实际应用时摄像头像素,选择拍摄的相机的像素;根据实际应用时二维码种类和工业场景的多样性,在多场景下拍摄多种二维码图片;使用常规的方法对二维码进行识别,清除掉即使定位成功也无法识别的二维码图片;拍摄二维码照片数量在1000到10000之间。
只有前期选取的工业二维码图片是符合工业场景实际情况的,最后对二维码定位时才会更准确。
优选地,所述步骤S2中,给图片生成对应的标签的具体步骤为:
S2.1:编写一个标签生成器,一张图片生成一个标签文件,一个标签文件包含图中所有的二维码信息,可以在图片中画旋转矩形框,并计算矩形框相对图片的四个顶点坐标;
S2.2:在标签生成器中对图片中的二维码画框,并将矩形框的四个顶点坐标转化为矩形框相对于整张图片的宽高w和h,中心点坐标x和y以及两个参数m和n。
S2.3:对宽高w和h,中心点坐标x和y以及与角度相关的两个参数m和n进行归一化处理。
优选地,所述步骤S2.1中,编写一个标签生成器,一张图片生成一个标签文件,一个标签文件包含图中所有的二维码信息,可以在图片中画旋转矩形框,具体包括:标签文件为txt文本格式,一个标签文件包含图中所有的二维码信息,每个二维码信息占一行,每个矩形框代表一个二维码,每个二维码信息包含有7个参数,第一个参数是二维码的种类,不同的类别用不同的数字表示;第二和第三个参数是二维码矩形框的中心点坐标,用x和y表示;第四个和第五个参数是二维码的宽高,用w和h表示;第六个和第七个参数是m和n。
生成的标签关系到整个二维码定位器。标签中一个二维码由7个参数表示,分别是类别,二维码中心点在图中的坐标,二维码的宽高,与二维码旋转角度相关的m和n。精确的标签信息对更加准确地预测二维码的信息很重要。
优选地,所述步骤S2.2中,在标签生成器中将矩形框的四个顶点坐标转化为矩形框相对于整张图片二维码中心点坐标x和y,二维码宽高w和h,两个与角度相关的参数m和n,具体转化方法:以图片正右方为x轴正方向,正下方为y轴正方向。找到矩形最下面的一个点,以此点向x轴正方向衍生出一条虚线,这条虚线与矩形的一条边形成夹角。记录矩形的这条边为w,与之相连的另一条边即为h,记录这条边的两个顶点的坐标,左边的点(x1,y1)和右边的点(x2,y2),m=x2-x1,n=y2-y1。
四个顶点坐标转为6个参数,x和y,w和h,m和n两两之间紧密性很强,对预测的精准性有很大提高,定位会更准确。
优选地,所述步骤S2.3中,对宽高w和h,中心点坐标x和y以及两个参数m和n进行归一化处理,具体包括:w、x、m分别除以图片宽度,h、y、n分别处以图片高度。
归一化处理后的参数不会在训练时因梯度下降过大,跳过优解,提高训练准确度。
优选地,所述步骤S3中,对二维码图片进行处理,获得处理后的数据集的具体步骤为:
S3.1:图中背景部分不动,旋转二维码,保存操作之前和之后的图片;
S3.2:图中背景部分不动,平移二维码,保存操作之前和之后的图片;
S3.3:给新合成的图片生成对应的标签,并作为数据集;
S3.4:把数据集按3:1分成训练集和测试集。
优选地,所述步骤S3.1中,图中背景不动,旋转二维码,具体包括:记录二维码的中心点在图片中的坐标;剪切掉图片中的二维码并保存切后的纯二维码图片和切后背景图,对切后背景图中切掉二维码后空白的部分使用泊松融合使其与背景图片一致,生成一张只有背景的切后背景图片;使切掉的二维码旋转随机角度,通过二维码中心点对应的原图片的坐标,把旋转后的二维码粘贴到背景图中。
对二维码做旋转数据增强,增加若干张增强训练图像,增加训练数据的规模和复杂度,提高角度相关信息预测的准确性。
优选地,所述步骤S3.2中,图中背景不动,平移二维码,具体包括:剪切掉图片中的二维码并保存切后的纯二维码图片和切后背景图,对切后背景图中切掉二维码后空白的部分使用泊松融合使其与背景图片一致,生成一张只有背景的切后背景图片;使切掉的二维码在背景图中上下左右平移一段距离且不移出图片。
对二维码做平移数据增强,增加若干张增强训练图像,增加训练数据的规模和复杂度,提高中心点坐标预测的准确性。
优选地,所述步骤S3.3中,给新合成的图片生成对应的标签。
给数据增强后的图片生成标签,扩充数据集,增加训练数据的规模和复杂度,提高中心点坐标预测的准确性。
优选地,所述步骤S3.4中,把数据集按3:1分成训练集和测试集。
数据集划分必须合理,3:1分成训练集和测试集的合理划分方式使得训练和测试的效果更有说服力。
优选地,所述步骤S5中,将待识别的二维码图片数据输入QRBOX_YOLOv5工业二维码定位器,对待定位的二维码进行实时推理,获得定位结果。
与现有技术相比,本发明技术方案的有益效果是:
本发明首先对工业二维码图片进行处理,将处理后的数据集作为训练集和测试集,增加了数据的规模和复杂度,有助于训练时提高qrbox_yolov5s神经网络的准确度;构建的qrbox_yolov5s神经网络属于轻量级网络,速度快;使用新型损失函数和回归预测方式,使用六个相互关联的参数表示二维码的位置信息,可以在工业环境下实现对物体的实时准确识别。
附图说明
图1为实施例1所述的一种基于YOLOv5的工业二维码定位方法;
图2为实施例1所述的qrbox_yolov5s神经网络的简略结构图;
图3为实施例1所述的QRBOX_YOLOv5工业二维码定位器的结构图。
具体实施方式
附图仅用于示例性说明,不能理解为对本专利的限制;
为了更好说明本实施例,附图某些部件会有省略、放大或缩小,并不代表实际产品的尺寸;
对于本领域技术人员来说,附图中某些公知结构及其说明可能省略是可以理解的。
下面结合附图和实施例对本发明的技术方案做进一步的说明。
实施例1
本发明提供一种基于YOLOv5的工业二维码定位方法,如图1所示,包括:
S1:采集包含二维码图片的照片,对照片进行预处理,得到处理后二维码图片;
本实施例中,根据实际应用时二维码种类和工业场景的多样性,在多个工业场景下拍摄多种二维码图片;使用常规的方法对二维码进行定位和识别,清除掉即使定位成功也无法识别的二维码图片;拍摄二维码照片数量为1000张图片。
S2:利用标签生成器,为处理后的二维码图片生成对应的标签;
包括:
S2.1:编写一个标签生成器,一张图片生成一个标签文件,一个标签文件包含图中所有的二维码信息,可以在图片中画旋转矩形框,并计算矩形框相对图片的四个顶点坐标;
本实施例中,编写了一个标签生成器,可以给每一张输入的图片生成一个标签文件,标签文件格式为txt文本格式,每个标签文件对应一张图片;标签文件中的每一行代表图片中的一个二维码;一行有7个参数,依次分别是:class,x,y,w,h,m,n,7个参数分别表示二维码的类别,二维码相对于整张图片的中心点坐标,二维码相对于整张图片的宽高,与二维码相对于整张图片的旋转角度。
S2.2:在标签生成器中对图片中的二维码画框,并将矩形框的四个顶点坐标转化为矩形框相对于整张图片的宽高w和h,中心点坐标x和y以及与角度相关的两个参数m和n;
本实施例中,在标签生成器中每对二维码画一个包围它的最小框,就生成一行标签信息;以图片正右方为x轴正方向,正下方为y轴正方向;找到矩形最下面的一个点,以此点向x轴正方向衍生出一条虚线,这条虚线与矩形的一条边形成夹角;记录矩形的这条边为w,与之相连的另一条边即为h,记录这条边的两个顶点的坐标,左边的点(x1,y1)和右边的点(x2,y2),m=x2-x1,n=y2-y1。
S2.3:对宽高w和h,中心点坐标x和y以及与角度相关的两个参数m和n进行归一化处理;
本实施例中,归一化处理方式为:w、x、m分别除以图片宽度,h、y、n分别处以图片高度。
S3:对标签中的二维码图片进行数据增强处理,获得增强后的图片的数据集;
包括:
S3.1:图中背景部分不动,旋转二维码,保存操作之前和之后的图片;
本实施例中,记录二维码的中心点在图片中的坐标;剪切掉图片中的二维码并保存切后的纯二维码图片、切后背景图及原图,对切后背景图中切掉二维码后空白的部分使用泊松融合使其与背景图片一致,生成一张只有背景的切后背景图片;使切掉的二维码旋转随机角度,通过二维码中心点对应的原图片的坐标,把旋转后的二维码粘贴到背景图中;计算旋转后的二维码的7个参数并保存在标签里;
操作进行3次,额外生成三张图;此时共计4000张图和4000个标签文件;
S3.2:图中背景部分不动,平移二维码,保存操作之前和之后的图片;
本实施例中,剪切掉图片中的二维码并保存切后的纯二维码图片、切后背景图及原图,对切后背景图中切掉二维码后空白的部分使用泊松融合使其与背景图片一致,生成一张只有背景的切后背景图片;使切掉的二维码在背景图中上下左右平移一段距离且不能移出背景图;计算平移后的二维码的7个参数并保存在标签里;
操作进行2次,额外生成2张图;此时共计12000张图和12000个标签文件;
S3.3:给新合成的图片生成对应的标签,并将其作为数据集。
本实施例中,这一步骤已经融合进了前面两步;
S3.4:把数据集按3:1分成训练集和测试集;
本实施例中,将数据集按照3:1比例随机划分成训练集和测试集。
S4:在QRBOX_YOLOv5工业二维码定位器中,利用数据增强后的图片对构建的qrbox_yolov5s神经网络进行训练生成权重并测试,将测试效果最好的训练权重作为训练好的QRBOX_YOLOv5工业二维码定位器。
本实施例中,qrbox_yolov5s神经网络是基于yolov5s网络构建的轻量级神经网络;
如图2所示,qrbox_yolov5s神经网络大致划分为四层,输入层,Backbone层,Neck层,Prediction层。
S4.1:在数据处理部分已经对数据进行过旋转和平移,在QRBOX_YOLOv5工业二维码定位器中不必再次使用旋转和平移做数据增强,只使用检测器中的剪切和拼接进行数据增强。
本实施例中,旋转和平移已经在步骤3做过,且QRBOX_YOLOv5工业二维码定位器是基于YOLOv5改建的,其中的旋转数据增强不适用于旋转目标,使用后会与平移冲突,不用再做,可使用其他数据增强方式继续扩大数据集规模。
S4.2:在qrbox_yolov5s网络中加入注意力机制。
本实例中,修改yolov5s.yaml文件,将C3模块修改为添加注意力机制后的模块CBAMC3,增强小的二维码目标检测准确度。
S4.3:在预测层使用新型损失函数。
本实例中,修改loss.py文件,使用新型损失函数,使用loss=SmoothL1+(m2+n2-w2)2作为预测参数的损失函数。
S4.4:在回归过程中使用sigmoid函数对m做映射,使用tanh函数对n做映射。
本实例中,修改loss.py和yolo.py文件,在参数回归过程中使用sigmoid函数对m做映射,使用tanh函数对n做映射。
S4.5:训练时选择batch-size设置为4或2,输入图像大小为32的倍数,且不小于真实图像大小的3倍。
本实例中,训练时选择batch-size设置为4,输入图像大小为32的倍数1024。
S4.6:利用处理后的图像数据的训练集对构建的qrbox_yolov5s神经网络进行训练。
本实例中,使用train.py文件对训练集数据进行训练,获得训练后的权重。
S4.7:使用训练出来的权重将测试集输入qrbox_yolov5s神经网络中进行测试。
本实例中,使用test.py文件对测试集数据进行测试,测试其的性能。
S5:将待识别的二维码图片数据输入QRBOX_YOLOv5工业二维码定位器,对待定位的二维码进行实时推理,获得定位结果。
在本实例中,QRBOX_YOLOv5工业二维码定位器是基于YOLOv5改建的,具有极快的定位速度,且通过新型损失函数等方法提高了定位的准确度;
如图3所示,QRBOX_YOLOv5工业二维码定位器主要分为三个模块,数据处理模块,网络训练和测试模块,工业二维码定位模块。在数据处理模块处理好数据,利用处理后的训练集数据在网络训练和测试模块训练,利用训练生成的权重在网络训练和测试模块进行测试,获得测试好的网络权重。
在实际操作过程中,安装python3.7及以上版本,用来搭建开发平台;安装JetBrains PyCharm 2019.1.1x64,更方便开发;得到测试好的权重后,利用测试好的权重,通过工业二维码定位模块读取图片,对其中待定位的二维码进行实时精确定位。
Claims (9)
1.一种基于YOLOv5的工业二维码定位方法,其特征在于,包括:
S1:采集包含二维码图片的照片,对照片进行预处理,得到处理后二维码图片;
S2:利用标签生成器,为处理后的二维码图片生成对应的标签;
所述步骤S2中,为处理后的二维码图片生成对应的标签,包括:
S2.1:编写一个标签生成器,一张图片生成一个标签文件,一个标签文件包含图中所有的二维码信息,在图片中画旋转矩形框,并计算矩形框相对图片的四个顶点坐标;
S2.2:在标签生成器中对图片中的二维码画框,并将矩形框的四个顶点坐标转化为矩形框相对于整张图片的宽高w和h,中心点坐标x和y以及两个参数m和n;
S2.3:对宽高w和h,中心点坐标x和y以及与角度相关的两个参数m和n进行归一化处理;
S3:对标签中的二维码图片进行数据增强处理,获得增强后的图片的数据集;
S4:在QRBOX_YOLOv5工业二维码定位器中,利用数据增强后的图片对构建的qrbox_yolov5s神经网络进行训练生成权重并测试,将测试效果最好的训练权重作为训练好的QRBOX_YOLOv5工业二维码定位器,qrbox_yolov5s神经网络划分为四层,四层分别是:输入层、Backbone层、Neck层和Prediction层;
所述步骤S4,包括:
S4.1:在数据处理部分已经对数据进行过旋转和平移,在QRBOX_YOLOv5工业二维码定位器中不必再次使用旋转和平移做数据增强,只使用检测器中的剪切和拼接进行数据增强;
S4.2:对qrbox_yolov5s网络中加入注意力机制;
S4.3:在预测层使用新型损失函数,公式如下:
loss=SmoothL1+(m2+n2-w2)2;
S4.4:在回归过程中使用sigmoid函数对m做映射,使用tanh函数对n做映射;
S4.5:训练时选择batch-size设置为4或2,输入图像大小为32的倍数,且不小于真实图像大小的3倍;
S4.6:利用处理后的图像数据的训练集对构建的qrbox_yolov5s神经网络进行训练;
S4.7:使用训练出来的权重将测试集输入qrbox_yolov5s神经网络中进行测试;
S5:将待识别的二维码图片数据输入训练好的QRBOX_YOLOv5工业二维码定位器,对待定位的二维码进行实时推理,获得定位结果。
2.根据权利要求1所述的一种基于YOLOv5的工业二维码定位方法,其特征在于,所述步骤S1中,采集包含二维码图片的照片,具体包括:
采集二维码图片时工业场景作为图片的背景,模拟实际的工业应用场景进行拍摄,根据实际的工业应用场景读取的二维码图片时的角度和距离拍摄;根据实际应用时摄像头像素,选择拍摄的相机的像素;根据实际应用时二维码种类和工业场景的多样性,在多场景下拍摄多种二维码图片;使用常规的方法对二维码进行识别,清除掉即使定位成功也无法识别的二维码图片;拍摄二维码照片数量在1000到10000之间。
3.根据权利要求1所述的一种基于YOLOv5的工业二维码定位方法,其特征在于,所述步骤S2.1中,编写一个标签生成器,一张图片生成一个标签文件,一个标签文件包含图中所有的二维码信息,在图片中画旋转矩形框,包括:
标签文件为txt文本格式,一个标签文件包含图中所有的二维码信息,每个二维码信息占一行,每个矩形框代表一个二维码,每个二维码信息包含有7个参数,第一个参数是二维码的种类,不同的类别用不同的数字表示;第二和第三个参数是二维码矩形框的中心点坐标,用x和y表示;第四个和第五个参数是二维码的宽高,用w和h表示;第六个和第七个参数是m和n。
4.根据权利要求1所述的一种基于YOLOv5的工业二维码定位方法,其特征在于,所述步骤S2.2中,将矩形框的四个顶点坐标转化为矩形框相对于整张图片的宽高w和h,两个参数m和n,包括:
以图片正右方为x轴正方向,正下方为y轴正方向,找到矩形最下面的一个点,以此点向x轴正方向衍生出一条虚线,这条虚线与矩形的一条边形成夹角,记录矩形的这条边为w,与之相连的另一条边即为h,记录这条边的两个顶点的坐标,左边的点(x1,y1)和右边的点(x2,y2),m=x2-x1,n=y2-y1。
5.根据权利要求1所述的一种基于YOLOv5的工业二维码定位方法,其特征在于,所述步骤S2.3中,对宽高w和h,中心点坐标x和y以及两个参数m和n进行归一化处理,具体包括:
w、x、m分别除以图片宽度,h、y、n分别除以图片高度。
6.根据权利要求1所述的一种基于YOLOv5的工业二维码定位方法,其特征在于,所述步骤S3中,对标签中的二维码图片进行数据增强处理,获得增强后的图片的数据集的具体步骤为:
S3.1:图中背景部分不动,旋转二维码,保存操作之前和之后的图片;
S3.2:图中背景部分不动,平移二维码,保存操作之前和之后的图片;
S3.3:给新合成的图片生成对应的标签,并作为数据集;
S3.4:把数据集按3:1分成训练集和测试集。
7.根据权利要求6所述的一种基于YOLOv5的工业二维码定位方法,其特征在于,所述步骤S3.1中,图中背景不动,旋转二维码,具体包括:
记录二维码的中心点在图片中的坐标;剪切掉图片中的二维码并保存切后的纯二维码图片和切后背景图,对切后背景图中切掉二维码后空白的部分使用泊松融合使其与背景图片一致,生成一张只有背景的切后背景图片;使切掉的二维码旋转随机角度,通过二维码中心点对应的原图片的坐标,把旋转后的二维码粘贴到背景图中。
8.根据权利要求6所述的一种基于YOLOv5的工业二维码定位方法,其特征在于,所述步骤S3.2中,图中背景不动,平移二维码,具体包括:剪切掉图片中的二维码并保存切后的纯二维码图片和切后背景图,对切后背景图中切掉二维码后空白的部分使用泊松融合使其与背景图片一致,生成一张只有背景的切后背景图片;使切掉的二维码在背景图中上下左右平移一段距离且不移出图片。
9.根据权利要求6所述的一种基于YOLOv5的工业二维码定位方法,其特征在于,所述给新合成的图片生成对应的标签,包括:
给数据增强后的图片生成标签,扩充数据集。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210228300.XA CN114693908B (zh) | 2022-03-08 | 2022-03-08 | 一种基于YOLOv5的工业二维码定位方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210228300.XA CN114693908B (zh) | 2022-03-08 | 2022-03-08 | 一种基于YOLOv5的工业二维码定位方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN114693908A CN114693908A (zh) | 2022-07-01 |
CN114693908B true CN114693908B (zh) | 2024-09-06 |
Family
ID=82137879
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202210228300.XA Active CN114693908B (zh) | 2022-03-08 | 2022-03-08 | 一种基于YOLOv5的工业二维码定位方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN114693908B (zh) |
Families Citing this family (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116993962B (zh) * | 2023-07-20 | 2024-04-26 | 广东南方智媒科技有限公司 | 二维码检测方法、装置、设备及可读存储介质 |
CN116882433B (zh) * | 2023-09-07 | 2023-12-08 | 无锡维凯科技有限公司 | 一种基于机器视觉的扫码识别方法和系统 |
CN117314951B (zh) * | 2023-11-20 | 2024-01-26 | 四川数盾科技有限公司 | 一种二维码识别预处理方法及系统 |
Citations (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN110941970A (zh) * | 2019-12-05 | 2020-03-31 | 深圳牛图科技有限公司 | 基于全卷积神经网络的高速维码定位识别系统 |
CN112580382A (zh) * | 2020-12-28 | 2021-03-30 | 哈尔滨工程大学 | 基于目标检测二维码定位方法 |
Family Cites Families (1)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US10956696B2 (en) * | 2019-05-31 | 2021-03-23 | Advanced New Technologies Co., Ltd. | Two-dimensional code identification and positioning |
-
2022
- 2022-03-08 CN CN202210228300.XA patent/CN114693908B/zh active Active
Patent Citations (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN110941970A (zh) * | 2019-12-05 | 2020-03-31 | 深圳牛图科技有限公司 | 基于全卷积神经网络的高速维码定位识别系统 |
CN112580382A (zh) * | 2020-12-28 | 2021-03-30 | 哈尔滨工程大学 | 基于目标检测二维码定位方法 |
Also Published As
Publication number | Publication date |
---|---|
CN114693908A (zh) | 2022-07-01 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN114693908B (zh) | 一种基于YOLOv5的工业二维码定位方法 | |
Li et al. | Layoutgan: Synthesizing graphic layouts with vector-wireframe adversarial networks | |
CN106845430A (zh) | 基于加速区域卷积神经网络的行人检测与跟踪方法 | |
CN112464912B (zh) | 基于YOLO-RGGNet的机器人端人脸检测方法 | |
CN113378815B (zh) | 一种场景文本定位识别的系统及其训练和识别的方法 | |
CN113762269A (zh) | 基于神经网络的中文字符ocr识别方法、系统、介质及应用 | |
CN105574545B (zh) | 街道环境图像多视角语义切割方法及装置 | |
CN113239753A (zh) | 基于YOLOv4改进的交通标志检测与识别方法 | |
CN114663707A (zh) | 基于Faster RCNN改进的少样本目标检测方法 | |
CN114332473A (zh) | 目标检测方法、装置、计算机设备、存储介质及程序产品 | |
CN114550110A (zh) | 一种基于无监督域适应的车辆重识别方法及系统 | |
CN115186473A (zh) | 一种基于平行智能的场景感知建模与验证方法 | |
CN118351410A (zh) | 一种基于稀疏代理注意力的多模态三维检测方法 | |
CN115937520A (zh) | 基于语义信息引导的点云运动目标分割方法 | |
CN116721139A (zh) | 生成图像数据的深度图像 | |
CN114187506B (zh) | 视点意识的动态路由胶囊网络的遥感图像场景分类方法 | |
CN112329771A (zh) | 一种基于深度学习的建筑材料样本识别方法 | |
CN115953744A (zh) | 一种基于深度学习的车辆识别追踪方法 | |
Wiguna et al. | Evaluation of deep learning models for building damage mapping in emergency response settings | |
Wu et al. | Deep texture exemplar extraction based on trimmed T-CNN | |
Li et al. | Few-shot meta-learning on point cloud for semantic segmentation | |
CN112149528A (zh) | 一种全景图目标检测方法、系统、介质及设备 | |
CN112668662A (zh) | 基于改进YOLOv3网络的野外山林环境目标检测方法 | |
CN118089669A (zh) | 一种基于航空测绘技术的地形测绘系统及测绘方法 | |
Stötzner et al. | CNN based Cuneiform Sign Detection Learned from Annotated 3D Renderings and Mapped Photographs with Illumination Augmentation |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |