CN115546207A - 杂质的快速检测方法、计算设备和存储介质 - Google Patents
杂质的快速检测方法、计算设备和存储介质 Download PDFInfo
- Publication number
- CN115546207A CN115546207A CN202211478924.3A CN202211478924A CN115546207A CN 115546207 A CN115546207 A CN 115546207A CN 202211478924 A CN202211478924 A CN 202211478924A CN 115546207 A CN115546207 A CN 115546207A
- Authority
- CN
- China
- Prior art keywords
- image
- segmentation
- training
- impurity
- neural network
- 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.)
- Withdrawn
Links
- 239000012535 impurity Substances 0.000 title claims abstract description 145
- 238000001514 detection method Methods 0.000 title claims abstract description 54
- 238000003860 storage Methods 0.000 title claims abstract description 17
- 238000003709 image segmentation Methods 0.000 claims abstract description 58
- 238000000034 method Methods 0.000 claims abstract description 52
- 238000013145 classification model Methods 0.000 claims abstract description 12
- 238000012549 training Methods 0.000 claims description 182
- 230000011218 segmentation Effects 0.000 claims description 109
- 238000013527 convolutional neural network Methods 0.000 claims description 60
- 235000020234 walnut Nutrition 0.000 claims description 43
- 238000005070 sampling Methods 0.000 claims description 25
- 238000007781 pre-processing Methods 0.000 claims description 19
- 241000758791 Juglandaceae Species 0.000 claims description 12
- 235000013305 food Nutrition 0.000 claims description 12
- 238000004590 computer program Methods 0.000 claims description 11
- 238000004422 calculation algorithm Methods 0.000 claims description 9
- 238000003062 neural network model Methods 0.000 claims description 9
- 238000011176 pooling Methods 0.000 claims description 9
- 230000008569 process Effects 0.000 claims description 8
- 238000002790 cross-validation Methods 0.000 claims description 4
- 230000001502 supplementing effect Effects 0.000 claims description 3
- 238000012546 transfer Methods 0.000 claims description 3
- 230000008901 benefit Effects 0.000 abstract description 3
- 241000758789 Juglans Species 0.000 description 31
- 235000009496 Juglans regia Nutrition 0.000 description 31
- 238000012360 testing method Methods 0.000 description 12
- 230000006870 function Effects 0.000 description 9
- 239000002184 metal Substances 0.000 description 7
- 238000013528 artificial neural network Methods 0.000 description 6
- 239000004033 plastic Substances 0.000 description 6
- 229920003023 plastic Polymers 0.000 description 6
- 239000012634 fragment Substances 0.000 description 5
- 238000012795 verification Methods 0.000 description 5
- 239000000356 contaminant Substances 0.000 description 3
- 238000012423 maintenance Methods 0.000 description 3
- 235000014571 nuts Nutrition 0.000 description 3
- 230000003287 optical effect Effects 0.000 description 3
- 238000002360 preparation method Methods 0.000 description 3
- 238000012545 processing Methods 0.000 description 3
- 239000003086 colorant Substances 0.000 description 2
- 230000009849 deactivation Effects 0.000 description 2
- 238000010586 diagram Methods 0.000 description 2
- 230000000694 effects Effects 0.000 description 2
- 238000000605 extraction Methods 0.000 description 2
- 238000003384 imaging method Methods 0.000 description 2
- 238000004519 manufacturing process Methods 0.000 description 2
- 239000013307 optical fiber Substances 0.000 description 2
- 230000000644 propagated effect Effects 0.000 description 2
- 238000011160 research Methods 0.000 description 2
- 239000000126 substance Substances 0.000 description 2
- 238000005299 abrasion Methods 0.000 description 1
- 230000004913 activation Effects 0.000 description 1
- 238000003556 assay Methods 0.000 description 1
- 238000004364 calculation method Methods 0.000 description 1
- 238000006243 chemical reaction Methods 0.000 description 1
- 238000007635 classification algorithm Methods 0.000 description 1
- 238000011109 contamination Methods 0.000 description 1
- 238000013135 deep learning Methods 0.000 description 1
- 230000008034 disappearance Effects 0.000 description 1
- 239000000428 dust Substances 0.000 description 1
- 238000005516 engineering process Methods 0.000 description 1
- 238000004880 explosion Methods 0.000 description 1
- 230000004927 fusion Effects 0.000 description 1
- 230000036541 health Effects 0.000 description 1
- 230000002779 inactivation Effects 0.000 description 1
- 239000000463 material Substances 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 238000012544 monitoring process Methods 0.000 description 1
- 238000010606 normalization Methods 0.000 description 1
- 239000005022 packaging material Substances 0.000 description 1
- 239000010893 paper waste Substances 0.000 description 1
- 238000011056 performance test Methods 0.000 description 1
- 239000013502 plastic waste Substances 0.000 description 1
- 238000003908 quality control method Methods 0.000 description 1
- 239000002994 raw material Substances 0.000 description 1
- 239000004065 semiconductor Substances 0.000 description 1
- 241000894007 species Species 0.000 description 1
- 230000017105 transposition Effects 0.000 description 1
- 238000010200 validation analysis Methods 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/0002—Inspection of images, e.g. flaw detection
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/10—Segmentation; Edge detection
- G06T7/11—Region-based segmentation
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/764—Arrangements for image or video recognition or understanding using pattern recognition or machine learning using classification, e.g. of video objects
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/82—Arrangements for image or video recognition or understanding using pattern recognition or machine learning using neural networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/10—Image acquisition modality
- G06T2207/10024—Color image
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20081—Training; Learning
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20084—Artificial neural networks [ANN]
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/30—Subject of image; Context of image processing
- G06T2207/30108—Industrial image inspection
- G06T2207/30128—Food products
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Physics & Mathematics (AREA)
- General Physics & Mathematics (AREA)
- Evolutionary Computation (AREA)
- Computing Systems (AREA)
- Artificial Intelligence (AREA)
- Health & Medical Sciences (AREA)
- Databases & Information Systems (AREA)
- General Health & Medical Sciences (AREA)
- Medical Informatics (AREA)
- Software Systems (AREA)
- Multimedia (AREA)
- Quality & Reliability (AREA)
- Image Analysis (AREA)
Abstract
本发明提供一种杂质的快速检测方法、计算设备和存储介质,该方法包括:获取包含目标物体的待测图像;将所述待测图像输入图像分割模型,以分割所述待测图像中的目标物体区域与杂质区域,并得到分割后的图像;将所述分割后的图像输入图像分类模型,以确定所述待测图像中杂质区域所包含的杂质类别,将所述待测图像中杂质区域所包含的杂质类别作为混入目标物体中的杂质类别。实现了杂质检测的自动化,避免了人工提取特征的麻烦,并且节省了杂质检测的时间,提高了杂质检测结果的准确率。
Description
技术领域
本发明涉及质量控制技术领域,尤其涉及一种杂质的快速检测方法、计算设备和存储介质。
背景技术
在食品安全控制的质量监测中,杂质检测是保证食品生产质量和安全控制的重要手段。杂质检测对于消费者的满意度和健康至关重要,同时也是食品安全法规的重要要求。杂质可能会通过原材料、故障生产线或非法人工污染等途径进入食品中。食品中的外来物质(例如,纸屑、包装材料、塑料废料和金属部件等)是许多食品制造商和执法机构收到客户投诉的最主要原因。在许多企业的坚果处理中,坚果中杂质的检测都是由人来完成的,这种传统的质量检测技术效率低下,且漏检率高,效果不理想。
近年来,越来越多的研究人员利用成像技术检测食品和农产品中的杂质,大多数基于计算机视觉的杂质检测算法都是基于模型的方法。但是,传统的模型方法需要专业设计过的约束条件,往往过于依赖知识专家或专业用户所制定的手工参数,检测性能差,且模型维护成本高。由于不同视角下杂质的形状和颜色各异,利用计算机视觉快速检测出杂质仍然面临着较大的挑战。
因此,需要一种新的杂质的快速检测方法,来提高检测效率。
发明内容
本发明的主要目的是提供一种杂质的快速检测方法、计算设备和存储介质,提高检测速度和检测结果正确率。
第一方面,本发明提供一种杂质的快速检测方法,包括:获取包含目标物体的待测图像;将待测图像输入图像分割模型,以分割待测图像中的目标物体区域与杂质区域,并得到分割后的图像,其中,所述图像分割模型包括:训练后的图像分割卷积神经网络模型,所述图像分割卷积神经网络模型包括:依次连接的图像预处理模块、下采样模块、上采样模块和图像分割模块,其中,所述下采样模块包括若干依次连接的下采样子模块,所述上采样模块包括若干依次连接的上采样子模块,所述下采样子模块由残差单元和最大池化层连接而成,所述上采样子模块由上采样单元和残差单元连接而成;将分割后的图像输入图像分类模型,以确定待测图像中杂质区域所包含的杂质类别,将待测图像中杂质区域所包含的杂质类别作为混入目标物体中的杂质类别。
在一个实施例中,对图像分割卷积神经网络模型进行训练,包括:获取分割训练数据集,其中,分割训练数据集包括若干分割训练样本,每个分割训练样本包括一张包含目标物体的分割训练图像及对该分割训练图像中杂质区域的标记;对每个分割训练样本中的分割训练图像进行预处理,预处理包括:为分割训练图像添加其是否包含杂质区域的标签,和/或调整分割训练图像的尺寸和/或像素;利用预处理之后的分割训练样本组成的分割训练数据集,对图像分割卷积神经网络模型进行训练,以得到训练后的图像分割卷积神经网络模型。
在一个实施例中,当分割训练样本的数量低于预设分割训练样本数量阈值时,在利用预处理之后的分割训练样本组成的分割训练数据集对图像分割卷积神经网络模型进行训练之前,该方法还包括:对若干分割训练样本中的分割训练图像按照预设方式进行调整,基于调整后的分割训练图像生成新的分割训练样本,将新生成的分割训练样本补充到分割训练数据集中,以使分割训练数据集中的分割训练样本的数量不低于预设分割训练样本数量阈值,其中,预设方式包括:旋转、宽度偏移、高度偏移和/或缩放。
在一个实施例中,利用预处理之后的分割训练样本组成的分割训练数据集,对图像分割卷积神经网络模型进行训练,包括:利用预处理之后的分割训练样本组成的分割训练数据集,通过交叉验证的方法,对图像分割卷积神经网络模型进行训练,并利用均方根传递算法对该训练过程进行优化。
在一个实施例中,分类模型包括:训练后的图像分类卷积神经网络模型。
在一个实施例中,对图像分类卷积神经网络模型进行训练,包括:获取分类训练数据集,其中,分类训练数据集包括若干分类训练样本,每个分类训练样本包括一张分割后的训练图像及对该分割后的训练图像中杂质区域所包含的杂质类别的标记;对每个分类训练样本中的分割后的训练图像进行预处理,预处理包括:调整分割后的训练图像的尺寸和/或像素;利用预处理之后的分类训练样本组成的分类训练数据集,对图像分类卷积神经网络模型进行训练,以得到训练后的图像分类卷积神经网络模型。
在一个实施例中,目标物体包括食品,其中,该食品包括核桃。
第二方面,本发明提供一种计算设备,包括处理器和存储器,存储器中存储有计算机程序,当计算机程序被处理器执行时,执行如上文所述的杂质的快速检测方法的步骤。
第三方面,本发明提供一种存储介质,存储有计算机程序,当计算机程序被处理器执行时,执行如上文所述的杂质的快速检测方法的步骤。
通过本发明的方法,首先利用图像分割模型对图像中的目标物体区域和杂质区域进行自动区分,随后利用图像分类模型对分割出的杂质区域的杂质进行分类,实现了杂质检测的自动化,避免了人工提取特征的麻烦,并且节省了杂质检测的时间,提高了杂质检测结果的准确率。在图像分割卷积神经网络模型中,将传统卷积神经网络的卷积层替换为残差单元,可以促进网络结构在不同尺度上协调从图像中学习到的特征。
附图说明
构成本申请的一部分的说明书附图用来提供对本发明的进一步理解,本发明的示意性实施例及其说明用于解释本发明,并不构成对本发明的限定,在附图中:
图1为根据本申请一示例性实施方式的杂质的快速检测方法的流程图;
图2为根据本申请一具体实施例的图像分割卷积神经网络的结构示意图;
图3为根据本申请一具体实施例的图像分类卷积神经网络的结构示意图;
图4为根据本申请一具体实施例的杂质的快速检测方法在对图像分类卷积神经网络模型的训练过程中所得的训练分类准确率和验证分类准确率的曲线图。
具体实施方式
需要说明的是,在不冲突的情况下,本申请中的实施例及实施例中的特征可以相互组合。下面将参考附图并结合实施例来详细说明本发明。
近年来,由于深度学习可以通过学习训练数据的特征来提供准确的结果,深度学习已经成为许多领域的研究热点。
实施例一
本实施例提供一种杂质的快速检测方法,图1为根据本申请一示例性实施方式的杂质的快速检测方法的流程图。如图1所示,本实施例的方法可以包括以下步骤:
S100:获取包含目标物体的待测图像。
S200:将待测图像输入图像分割模型,以分割待测图像中的目标物体区域与杂质区域,并得到分割后的图像。
其中,所述图像分割模型包括:训练后的图像分割卷积神经网络模型,所述图像分割卷积神经网络模型包括:依次连接的图像预处理模块、下采样模块、上采样模块和图像分割模块,其中,所述下采样模块包括若干依次连接的下采样子模块,所述上采样模块包括若干依次连接的上采样子模块,所述下采样子模块由残差单元和最大池化层连接而成,所述上采样子模块由上采样单元和残差单元连接而成。
S300:将分割后的图像输入图像分类模型,以确定待测图像中杂质区域所包含的杂质类别,将待测图像中杂质区域所包含的杂质类别作为混入目标物体中的杂质类别。
通过上述步骤,针对包含目标物体的待测图像,首先对其进行图像分割,以区分目标物体区域与杂质区域,然后对图像分割得到的杂质区域进行杂质分类,最终确定出待测图像中所包含的杂质类别,也即,混入目标物体中的杂质类别。其中,目标物体可以包括各种食品,该食品可以是坚果,例如核桃,当然,目标物体也可以包括其他需要检测其中杂质的物体。在图像分割卷积神经网络模型中,将传统卷积神经网络的卷积层替换为残差单元,可以促进网络结构在不同尺度上协调从图像中学习到的特征。
当然,该图像分割模型也可以是其他类型的模型,例如,深度神经网络模型等,只要能够实现对目标物体区域与杂质区域的有效区分即可。
对图像分割卷积神经网络模型进行训练,可以包括:获取分割训练数据集,其中,分割训练数据集包括若干分割训练样本,每个分割训练样本包括一张包含目标物体的分割训练图像及对该分割训练图像中杂质区域的标记;对每个分割训练样本中的分割训练图像进行预处理,预处理包括:为分割训练图像添加其是否包含杂质区域的标签,和/或调整分割训练图像的尺寸和/或像素;利用预处理之后的分割训练样本组成的分割训练数据集,对图像分割卷积神经网络模型进行训练,以得到训练后的图像分割卷积神经网络模型。
在一个示例中,当分割训练样本的数量低于预设分割训练样本数量阈值时,在利用预处理之后的分割训练样本组成的分割训练数据集对图像分割卷积神经网络模型进行训练之前,该方法还可以包括:对若干分割训练样本中的分割训练图像按照预设方式进行调整,基于调整后的分割训练图像生成新的分割训练样本,将新生成的分割训练样本补充到分割训练数据集中,以使分割训练数据集中的分割训练样本的数量不低于预设分割训练样本数量阈值,其中,预设方式可以包括:旋转、宽度偏移、高度偏移和/或缩放。
利用预处理之后的分割训练样本组成的分割训练数据集,对图像分割卷积神经网络模型进行训练,可以包括:利用预处理之后的分割训练样本组成的分割训练数据集,通过交叉验证的方法,对图像分割卷积神经网络模型进行训练,并利用均方根传递算法对该训练过程进行优化。
在一个具体的例子中,分类模型可以包括:训练后的图像分类卷积神经网络模型。当然,该图像分类模型也可以是其他类型的模型,例如,全连接神经网络模型等,只要能够实现图像分类即可。
对图像分类卷积神经网络模型进行训练,可以包括:获取分类训练数据集,其中,分类训练数据集包括若干分类训练样本,每个分类训练样本包括一张分割后的训练图像及对该分割后的训练图像中杂质区域所包含的杂质类别的标记;对每个分类训练样本中的分割后的训练图像进行预处理,预处理包括:调整分割后的训练图像的尺寸和/或像素;利用预处理之后的分类训练样本组成的分类训练数据集,对图像分类卷积神经网络模型进行训练,以得到训练后的图像分类卷积神经网络模型。
通过本实施例的方法,首先利用图像分割卷积神经网络模型对图像中的目标物体区域和杂质区域进行自动图像分割,随后利用图像分类卷积神经网络模型对分割出的杂质区域的杂质进行分类,实现了杂质检测的自动化,避免了人工提取特征的麻烦,并且节省了杂质检测的时间,提高了杂质检测结果的准确率。
实施例二
在本实施例中,我们利用两阶段卷积神经网络来实时完成对核桃图像的图像分割和杂质分类。
基于卷积神经网络的杂质检测方法可以自动对图像进行杂质区域的分割,以及确定所分割出的杂质区域所包含的杂质类别(例如,树叶碎片、纸屑、塑料屑、金属零件等)。
本实施例提出的杂质检测方法能够实现对核桃杂质的自动检测,更加简单高效,避免了手工提取特征的麻烦,即使针对待测图像中核桃与杂质的聚集部分也能够实现准确检测,还能避免白色传送带表面磨损对检测结果的干扰。
根据实验结果,利用本实施例的方法对被测图像中99.4%的目标区域进行了正确分割,对分割后的图像中96.5%的外来杂质进行了正确分类,并且,对每幅被测图像的分割和分类处理总时间小于60毫秒。
本实施例的两阶段卷积神经网络,在训练时,分别对分割模型和分类模型进行训练。在杂质检测过程中,利用CNN(Convolutional Neural Networks,卷积神经网络)分割模型对待测图像中的核桃区域和杂质区域进行分割,最后利用CNN分类模型对分割后的杂质区域所包含的杂质进行分类,得到杂质检测结果。
一、图像分割
在可见光成像条件下,传统的图像分割方法不能有效地提取核桃图像中的所有感兴趣的区域,这是因为传送带表面被不同的形状、颜色和大小的核桃和杂质进行了磨蚀破坏,该磨蚀破坏的部分给图像分割带来了干扰。
本实施例提出了一种对SEGNET架构的图像分割卷积神经网络模型的改进版本。
如图2中的(a)所示,下采样路径包括5个下采样子模块,每个下采样子模块包括一个残差单元和其后的一个最大池化层。在本实施例中,将传统卷积神经网络的卷积层替换为残差单元,可以促进网络结构在不同尺度上协调从图像中学习到的特征。通过在相邻的残差单元之间插入最大池化层,从(32×32)~(512×512)的空间维度N对核桃图像进行向下采样。
上采样路径为下采样路径的镜像版本,上采样路径采用转置卷积计算。
在残差卷积神经网络中,残差单元通过残差学习,缓解了卷积神经网络中梯度的爆炸和消失。残差单元如图2中的(b)所示,残差单元由三个3×3 CONV(卷积层)、一个1×1CONV(卷积层)、两个批处理归一化(Batch Normalization,简称BN)层和一个RELU(LinearRectification Function,整流线性单元)层组成。将三个3×3卷积层级联在一起,提取多种不同尺度空间的特征。通过1×1卷积层的融合,残差连接可以提高图像分割效果,并获得一些额外的空间信息。
在训练过程中可使用RMSprop(Root Mean Square Prop,均方根传递)算法作为优化器。网络末端的Sigmoid(Sigmoid函数是一种S型函数,在神经网络中一般用作激活函数)层生成概率二通道输出,然后在分割过程中得到二值图像。
在训练过程中,每个卷积结果在下一层形成特征图的一个元素。通过如下的表达式(1)建立了Sigmoid层与其它层之间的连接。
其中,x为输入数据。
因为图像数据的最大池化性能优于其他池化方法,为了在训练过程中减小特征图的空间大小,在网络中集成了降采样函数和最大池化函数。
二、图像分类
杂质自动检测方法的第二个任务是根据分割结果对混入核桃中的杂质进行分类。我们训练了一个图像分类卷积神经网络模型来检测混入核桃中的杂质类别。
在将分割后的图像输入图像分类卷积神经网络模型之前,将待测的RGB图像的分辨率调整为50×50。在对该模型的训练过程中,使用RMSprop作为优化器。批量设置为30,动量设置为0.9,学习率设置为0.0001,在110次迭代后停止学习。一般来说,通过增加深度,神经网络可以很好地模拟训练数据。
如图3所示,图像分类卷积神经网络模型由4个3×3卷积层、4个RELU层、4个最大池化层和4个DROPOUT层(随机失活层)组成。分类器输出单元由DENSE层(全连接层)、FLATTEN层(一维数据转换层)和SOFTMAX层(分类层)组成,SOFTMAX作为分类层对核桃中的杂质进行分类。
图像分类卷积神经网络模型中的DROPOUT策略是通过防止特征检测器的共适应来改进神经网络的一种训练方法,如图3所示,第一、二随机失活率n均为0.15,第三、四随机失活率n均为0.25。
在训练过程中,可使用SOFTMAX分类器对样本进行分类,并选择概率最大的类别作为输出结果。一般情况下,多分类器的SOFTMAX函数输出每个类的概率。而对于核桃图像中存在杂质的情况,我们的主要任务是检测杂质,而不是确定每个类的概率。我们使用了4个类分类器,第一类是核桃,第二类是新鲜树叶碎片和干树叶碎片,第三类是纸屑和塑料屑,第四类是金属零件。利用这些层构建具有该结构的卷积神经网络,并将样本分为4类。分类准确率是用正确分类的数据量除以被测数据总量得到的。
图像分割方面,本实施例所采用的方法,在不进行手工提取分割特征参数的情况下,图像分割的结果是令人满意的,从中心区域和边缘区域均分离出了杂质区域,几乎检测到了所有的区域。
在图像分类方面,基于图像分类卷积神经网络模型的方法可以检测核桃中的杂质类型,检测正确率高。
本实施例的方法在自动完成图像分割的同时,对杂质区域的杂质进行分类。在CNN处理流程中,通过多尺度残差卷积网络中的低层向高层提取核桃图像的详细特征,网络输出部分的单个卷积网络层进行分割,以及由分类网络输出部分的全连接层来完成检测,无需手工进行特征提取。
本实施例的方法使用一般的个人计算机即可,对每张被测图像中所有区域的分割和分类处理总时间在60ms以下,大大节约检测时间,与传统的需要人工重新提取特征和阈值参数的图像检测方法相比较,本实施例的方法的模型维护成本更低,技术适用性更高。
下面以一个具体的例子进行说明。
首先,进行材料准备。
通过彩色图像采集系统采集大量实验样本图像,其中有不包含杂质的纯核桃图像和包含杂质的核桃图像。
彩色图像采集系统由一台1280×960分辨率的工业彩色千兆以太网摄像机(DVP-30GC(M)03 from The ImageSource, Inc, Germany)连接到一台个人计算机(由i7-8700@3.2Hz的CPU,16gb的RAM,和一台包含16gbRAM的Quadro P5000显卡组成)组成。CCD(Charge-coupled Device,电荷耦合器件)相机被放置在场景上方350毫米处。照明系统由四个白色LED灯带组成(每个10盏)。漫反射板被放置在每个LED灯带的后面,以减少场景中不均匀的照明。所有的元件都安装在一个黑色的检测室内。混入异物的核桃在白色传送带上随白色传送带移动到检测室。
样本包括大量的图像,这些样本图像中包括各种形状、颜色和大小的核桃和杂质(例如,树叶碎片、纸屑、塑料碎片和金属部件)。在样本图像中,一些杂质与核桃或重叠或粘连在一起,另一些杂质则是独立的。
图像分割部分
(1)数据集准备
训练数据集的质量将会影响神经网络的性能。在许多研究项目中,很难获得大量高质量的样本训练数据,这可能导致研究人员使用相对轻量级的网络。扩展训练数据集的规模或提高数据质量将有助于优化对神经网络的训练结果。
为训练图像分割卷积神经网络模型,选取100幅纯核桃图像和30幅掺入杂质的核桃图像作为分割训练数据集,另外选取40幅纯核桃图像和30幅掺入杂质的核桃图像作为分割测试数据集。
在人工标注图像中核桃和杂质的区域边界的基础上,可以为训练数据集中的每一幅图像创建一个二值标签,以表明该幅图像是否包含杂质区域。为建立图像分割算法的训练和测试数据集,将这些图像通过填充零值转换为正方形图像,并使用双三次插值将正方形图像调整为512×512像素分辨率。
由于分割训练数据集中的样本数量不足,进行数据增强以增加分割训练数据集中的样本数量。具体而言,对于每个原始分割训练图像,创建四个额外的分割训练图像,以扩充分割训练数据集。
首先,在0度至45度之间选择一个随机的旋转值,相应地旋转分割训练样本图像。
其次,在0-40%之间选择一个随机的宽度偏移范围值,使宽度偏移在图像大小的范围内,使分割训练样本图像的宽度发生相应的偏移。
第三,在0-40%之间选择一个随机的高度偏移范围值,使高度偏移在图像大小的范围内,使分割训练样本图像的高度发生相应的变化。
最后,选择一个随机的缩放比例,对分割训练样本图像进行相应的缩放。
(2)分割网络
选取包含不同杂质的测试图像(例如,叶屑、纸屑、塑料屑和金属零件)对本实施例方法的图像分割算法的性能进行了测试。
测试数据集图像的分割结果如下表1所示。根据实验结果,在不通过手工提取分割特征参数的情况下,图像分割卷积神将网络模型对图像分割的结果是令人满意的,尤其是对图像的中心和边缘均进行了图像分割。
表1 测试数据集图像中各类型杂质的分割结果
对象 | 区域内数量 | 正确分割 | 正确率 | 错误分割 | 错误率 |
核桃 | 2078 | 2068 | 99.52% | 10 | 0.48% |
碎叶片 | 262 | 259 | 98.85% | 3 | 1.15% |
纸屑 | 198 | 188 | 94.95% | 10 | 5.05% |
塑料碎片 | 135 | 132 | 97.78% | 3 | 2.22% |
金属零件 | 175 | 171 | 97.71% | 4 | 2.29% |
总计 | 2848 | 2818 | 98.95% | 30 | 1.05% |
杂质分类部分
杂质自动检测方法的第二个任务是根据分割结果对被测图像中各杂质区域的杂质进行分类。可以训练一个图像分类卷积神经网络模型来检测被测图像中杂质区域的杂质类别。
(1)数据集准备
分类训练数据集包括400张杂质样本图像和300张纯核桃样本图像,分类验证数据集可以包括180张杂质样本图像和100张纯核桃样本图像,分类测试数据集可以包括100张分辨率为1280×960的核桃图像,其中70张图像混合了杂质,该分类测试数据集的100张核桃图像可以包括核桃区域子图像共2427张,杂质区域子图像共398张。
首先,将准备好的分类训练图像的大小调整为50×50,然后输入CNN分类器,利用训练后的CNN分类器确定出验证数据集的被测图像中杂质区域的杂质类别。经过验证的CNN分类器对测试数据集中100张1280×960分辨率的图像中的所有区域子图像进行扫描,生成杂质检测报告。
(2)分类网络
为了测试基于图像分类卷积神经网络模型的分类算法的实现性能,使用不同杂质、不同形态的核桃图像对所提出的算法进行了测试。我们使用DROPOUT策略来训练四类分类器,从而避免过拟合。
每组训练数据集和验证数据集中对杂质的分类准确率如图4所示。在本实施例描述的分类模型中,第一类是核桃,第二类是新鲜叶屑和干叶屑,第三类是纸屑和塑料屑,第四类是金属零件。从图4中可以看出,训练数据集中杂质的训练分类准确率随着迭代次数的增加而增加,最佳分类准确率为97.0%;训练数据集中杂质的验证分类准确率随着迭代次数的增加而增加,分类准确率为96.5%。并进行5次交叉验证,平均准确率为96.1%。验证精度略低于训练精度是正常现象,即检测网络中不存在过拟合。
在不同的采集条件和不同的曝光时间下,基于卷积神经网络模型的方法可以检测核桃中的杂质类别,而传统的图像检测方法则需要人工提取特征和阈值参数,模型维护成本高,技术适用性低。
为了突出杂质的检测性能,将分割后的图像中的杂质区域进行了标记并通过计算机输出了杂质类别。
选择核桃和杂质样品时不存在特殊的人为因素,这对于检测算法的性能测试来说是公平的。
测试数据集中对纯核桃图像和混入杂质的核桃图像的检测结果如下表2所示。测试数据集中100幅图像的中杂质类别的分类结果的正确率为100%。
表2 测试数据集中核桃图像的检测结果
正确组别 | 杂质类别检测正确率 | 杂质类别检测错误率 |
没有杂质的核桃图像(30张) | 30(100.0%) | 0(0.0%) |
混合了杂质的核桃图像(70张) | 70(100.0%) | 0(0.0%) |
总检测(%) | 100% | 0% |
实施例三
本实施例提供一种计算设备,包括处理器和存储器,存储器中存储有计算机程序,当计算机程序被处理器执行时,执行如上文所述的杂质的快速检测方法的步骤。
在一个实施例中,该计算设备可以包括一个或多个处理器(CPU)、输入/输出接口、网络接口和内存。
内存可能包括计算机可读介质中的非永久性存储器,随机存取存储器(RAM)和/或非易失性内存等形式,如只读存储器(ROM)或闪存(flash FLASH RAM)。内存是计算机可读介质的示例。
实施例四
本实施例提供一种存储介质,存储有计算机程序,当计算机程序被处理器执行时,执行如上文所述的杂质的快速检测方法的步骤。
计算机程序可以采用一个或多个存储介质的任意组合。存储介质可以是可读信号介质或可读存储介质。
可读存储介质例如可以包括电、磁、光、电磁、红外线、或半导体的系统、装置或器件,或者任意以上的组合。可读存储介质的更具体的例子(非穷举的列表)可以包括:具有一个或多个导线的电连接、便携式盘、硬盘、随机存取存储器(RAM)、只读存储器(ROM)、可擦式可编程只读存储器(EPROM或闪存)、光纤、便携式紧凑盘只读存储器(CD-ROM)、光存储器件、磁存储器件或者上述的任意合适的组合。
可读信号介质可以包括在基带中或者作为载波一部分传播的数据信号,其中承载了可读计算机程序。这种传播的数据信号可以采用多种形式,例如可以包括电磁信号、光信号或上述的任意合适的组合。可读信号介质还可以是可读存储介质以外的任何存储介质,该存储介质可以发送、传播或者传输用于由指令执行系统、装置或者器件使用或者与其结合使用的程序。
存储介质上包含的计算机程序可以用任何适当的介质传输,例如可以包括无线、有线、光缆、RF等等,或者上述的任意合适的组合。
可以以一种或多种程序设计语言的任意组合来编写用于执行本发明操作的计算机程序。程序设计语言可以包括面向对象的程序设计语言——例如Java、C++等,还可以包括常规的过程式程序设计语言——诸如“C”语言或类似的程序设计语言。计算机程序可以完全地在用户计算设备上执行、部分地在用户设备上执行、或者完全在远程计算设备或服务器上执行。在涉及远程计算设备的情形中,远程计算设备可以通过任意种类的网(例如可以包括局域网或广域网)连接到用户计算设备,或者,可以连接到外部计算设备(例如利用因特网服务提供商来通过因特网连接)。
需要注意的是,这里所使用的术语仅是为了描述具体实施方式,而非意图限制根据本申请的示例性实施方式,当在本说明书中使用术语“包含”和/或“包括”时,其指明存在特征、步骤、操作、器件、组件和/或它们的组合。
需要说明的是,本申请的说明书和权利要求书及附图中的术语“第一”、“第二”等是用于区别类似的对象,而不用于描述特定的顺序或先后次序。应该理解这样使用的术语在适当情况下可以互换。
应当理解的是,本说明书中的示例性实施方式可以由多种不同的形式来实施,并且不应当被解释为只限于这里所阐述的实施方式。附加地或备选地,可以省略某些步骤,将多个步骤合并为一个步骤执行,和/或将一个步骤分解为多个步骤执行。提供这些实施方式是为了使得本申请的公开彻底且完整,并且将这些示例性实施方式的构思充分传达给本领域普通技术人员,而不应当理解为对本发明的限制。
虽然已经参考若干具体实施方式描述了本发明的精神和原理,但是应该理解,本发明并不限于所公开的具体实施方式,对各方面的划分也不意味着这些方面中的特征不能组合以进行受益,这种划分仅是为了表述的方便。本发明旨在涵盖所附权利要求的精神和范围内所包括的各种修改和等同布置。
Claims (9)
1.一种杂质的快速检测方法,其特征在于,包括:
获取包含目标物体的待测图像;
将所述待测图像输入图像分割模型,以分割所述待测图像中的目标物体区域与杂质区域,并得到分割后的图像,其中,所述图像分割模型包括:训练后的图像分割卷积神经网络模型,所述图像分割卷积神经网络模型包括:依次连接的图像预处理模块、下采样模块、上采样模块和图像分割模块,其中,所述下采样模块包括若干依次连接的下采样子模块,所述上采样模块包括若干依次连接的上采样子模块,所述下采样子模块由残差单元和最大池化层连接而成,所述上采样子模块由上采样单元和残差单元连接而成;
将所述分割后的图像输入图像分类模型,以确定所述待测图像中杂质区域所包含的杂质类别,将所述待测图像中杂质区域所包含的杂质类别作为混入目标物体中的杂质类别。
2.根据权利要求1所述的杂质的快速检测方法,其特征在于,对图像分割卷积神经网络模型进行训练,包括:
获取分割训练数据集,其中,所述分割训练数据集包括若干分割训练样本,每个分割训练样本包括一张包含目标物体的分割训练图像及对该分割训练图像中杂质区域的标记;
对每个分割训练样本中的分割训练图像进行预处理,所述预处理包括:为所述分割训练图像添加其是否包含杂质区域的标签,和/或调整所述分割训练图像的尺寸和/或像素;
利用预处理之后的分割训练样本组成的分割训练数据集,对图像分割卷积神经网络模型进行训练,以得到训练后的图像分割卷积神经网络模型。
3.根据权利要求2所述的杂质的快速检测方法,其特征在于,当所述分割训练样本的数量低于预设分割训练样本数量阈值时,在利用预处理之后的分割训练样本组成的分割训练数据集对图像分割卷积神经网络模型进行训练之前,所述方法还包括:
对若干分割训练样本中的分割训练图像按照预设方式进行调整,基于调整后的分割训练图像生成新的分割训练样本,将新生成的分割训练样本补充到所述分割训练数据集中,以使所述分割训练数据集中的分割训练样本的数量不低于预设分割训练样本数量阈值,其中,所述预设方式包括:旋转、宽度偏移、高度偏移和/或缩放。
4.根据权利要求2所述的杂质的快速检测方法,其特征在于,利用预处理之后的分割训练样本组成的分割训练数据集,对图像分割卷积神经网络模型进行训练,包括:
利用预处理之后的分割训练样本组成的分割训练数据集,通过交叉验证的方法,对图像分割卷积神经网络模型进行训练,并利用均方根传递算法对该训练过程进行优化。
5.根据权利要求1所述的杂质的快速检测方法,其特征在于,所述分类模型包括:训练后的图像分类卷积神经网络模型。
6.根据权利要求5所述的杂质的快速检测方法,其特征在于,对图像分类卷积神经网络模型进行训练,包括:
获取分类训练数据集,其中,所述分类训练数据集包括若干分类训练样本,每个分类训练样本包括一张分割后的训练图像及对该分割后的训练图像中杂质区域所包含的杂质类别的标记;
对每个分类训练样本中的分割后的训练图像进行预处理,所述预处理包括:调整所述分割后的训练图像的尺寸和/或像素;
利用预处理之后的分类训练样本组成的分类训练数据集,对图像分类卷积神经网络模型进行训练,以得到训练后的图像分类卷积神经网络模型。
7.根据权利要求1所述的杂质的快速检测方法,其特征在于,所述目标物体包括食品,其中,所述食品包括核桃。
8.一种计算设备,其特征在于,包括处理器和存储器,所述存储器中存储有计算机程序,当所述计算机程序被所述处理器执行时,执行如权利要求1至7中任一项所述的杂质的快速检测方法的步骤。
9.一种存储介质,其特征在于,存储有计算机程序,当所述计算机程序被处理器执行时,执行如权利要求1至7中任一项所述的杂质的快速检测方法的步骤。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202211478924.3A CN115546207A (zh) | 2022-11-24 | 2022-11-24 | 杂质的快速检测方法、计算设备和存储介质 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202211478924.3A CN115546207A (zh) | 2022-11-24 | 2022-11-24 | 杂质的快速检测方法、计算设备和存储介质 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN115546207A true CN115546207A (zh) | 2022-12-30 |
Family
ID=84720551
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202211478924.3A Withdrawn CN115546207A (zh) | 2022-11-24 | 2022-11-24 | 杂质的快速检测方法、计算设备和存储介质 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN115546207A (zh) |
Cited By (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116274170A (zh) * | 2023-03-27 | 2023-06-23 | 中建三局第一建设工程有限责任公司 | 一种激光清洗设备的控制方法、系统及相关装置 |
CN116597364A (zh) * | 2023-03-29 | 2023-08-15 | 阿里巴巴(中国)有限公司 | 图像处理方法及装置 |
CN117011550A (zh) * | 2023-10-08 | 2023-11-07 | 超创数能科技有限公司 | 一种电子显微镜照片中的杂质识别方法及装置 |
Citations (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113019993A (zh) * | 2021-04-19 | 2021-06-25 | 济南大学 | 一种籽棉的杂质分类识别方法及系统 |
CN113837062A (zh) * | 2021-09-22 | 2021-12-24 | 内蒙古工业大学 | 一种分类方法、装置、存储介质及电子设备 |
-
2022
- 2022-11-24 CN CN202211478924.3A patent/CN115546207A/zh not_active Withdrawn
Patent Citations (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113019993A (zh) * | 2021-04-19 | 2021-06-25 | 济南大学 | 一种籽棉的杂质分类识别方法及系统 |
CN113837062A (zh) * | 2021-09-22 | 2021-12-24 | 内蒙古工业大学 | 一种分类方法、装置、存储介质及电子设备 |
Non-Patent Citations (1)
Title |
---|
DIAN RONG ET AL.: "Impurity detection of juglans using deep learning and machine vision", 《COMPUTERS AND ELECTRONICS IN AGRICULTURE》 * |
Cited By (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116274170A (zh) * | 2023-03-27 | 2023-06-23 | 中建三局第一建设工程有限责任公司 | 一种激光清洗设备的控制方法、系统及相关装置 |
CN116274170B (zh) * | 2023-03-27 | 2023-10-13 | 中建三局第一建设工程有限责任公司 | 一种激光清洗设备的控制方法、系统及相关装置 |
CN116597364A (zh) * | 2023-03-29 | 2023-08-15 | 阿里巴巴(中国)有限公司 | 图像处理方法及装置 |
CN116597364B (zh) * | 2023-03-29 | 2024-03-29 | 阿里巴巴(中国)有限公司 | 图像处理方法及装置 |
CN117011550A (zh) * | 2023-10-08 | 2023-11-07 | 超创数能科技有限公司 | 一种电子显微镜照片中的杂质识别方法及装置 |
CN117011550B (zh) * | 2023-10-08 | 2024-01-30 | 超创数能科技有限公司 | 一种电子显微镜照片中的杂质识别方法及装置 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN115546207A (zh) | 杂质的快速检测方法、计算设备和存储介质 | |
CA3017646C (en) | Label and field identification without optical character recognition (ocr) | |
CN109154978B (zh) | 用于检测植物疾病的系统和方法 | |
Dubey et al. | Application of image processing in fruit and vegetable analysis: a review | |
Kumar et al. | Leafsnap: A computer vision system for automatic plant species identification | |
Barnes et al. | Visual detection of blemishes in potatoes using minimalist boosted classifiers | |
Li et al. | Characterness: An indicator of text in the wild | |
CN107067006B (zh) | 一种服务于数据采集的验证码识别方法及系统 | |
Marchant et al. | Automated analysis of foraminifera fossil records by image classification using a convolutional neural network | |
Moses et al. | Deep CNN-based damage classification of milled rice grains using a high-magnification image dataset | |
Dubey et al. | Fruit and vegetable recognition by fusing colour and texture features of the image using machine learning | |
Yang et al. | A framework for improved video text detection and recognition | |
Harraj et al. | OCR accuracy improvement on document images through a novel pre-processing approach | |
CN110503103B (zh) | 一种基于全卷积神经网络的文本行中的字符切分方法 | |
CN109063619A (zh) | 一种基于自适应背景抑制滤波器和组合方向梯度直方图的交通信号灯检测方法和系统 | |
US20210214765A1 (en) | Methods and systems for automated counting and classifying microorganisms | |
Gómez et al. | Cutting Sayre's Knot: reading scene text without segmentation. application to utility meters | |
Ramirez-Paredes et al. | Visual quality assessment of malting barley using color, shape and texture descriptors | |
CN103824090A (zh) | 一种自适应的人脸低层特征选择方法及人脸属性识别方法 | |
US11600088B2 (en) | Utilizing machine learning and image filtering techniques to detect and analyze handwritten text | |
CN114373185A (zh) | 票据图像的分类方法及其装置、电子设备及存储介质 | |
JPH05225378A (ja) | 文書画像の領域分割システム | |
CN110866931B (zh) | 图像分割模型训练方法及基于分类的强化图像分割方法 | |
CN105354405A (zh) | 基于机器学习的免疫组化图像自动化判读系统 | |
CN114581928A (zh) | 一种表格识别方法及系统 |
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 | ||
WW01 | Invention patent application withdrawn after publication |
Application publication date: 20221230 |
|
WW01 | Invention patent application withdrawn after publication |