CN112084950A - 基于稀疏卷积神经网络的目标检测方法及检测装置 - Google Patents
基于稀疏卷积神经网络的目标检测方法及检测装置 Download PDFInfo
- Publication number
- CN112084950A CN112084950A CN202010946308.0A CN202010946308A CN112084950A CN 112084950 A CN112084950 A CN 112084950A CN 202010946308 A CN202010946308 A CN 202010946308A CN 112084950 A CN112084950 A CN 112084950A
- Authority
- CN
- China
- Prior art keywords
- neural network
- convolutional neural
- target
- sparse
- image
- 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.)
- Pending
Links
- 238000013527 convolutional neural network Methods 0.000 title claims abstract description 162
- 238000001514 detection method Methods 0.000 title claims abstract description 122
- 238000013528 artificial neural network Methods 0.000 claims abstract description 23
- 238000000034 method Methods 0.000 claims abstract description 21
- 238000012549 training Methods 0.000 claims abstract description 19
- 238000002372 labelling Methods 0.000 claims abstract description 11
- 230000008569 process Effects 0.000 claims description 15
- 230000006870 function Effects 0.000 claims description 10
- 238000004590 computer program Methods 0.000 claims description 4
- 239000003086 colorant Substances 0.000 claims description 3
- 230000004048 modification Effects 0.000 claims description 3
- 238000012986 modification Methods 0.000 claims description 3
- 238000005457 optimization Methods 0.000 claims description 3
- 238000005070 sampling Methods 0.000 claims description 3
- 239000004576 sand Substances 0.000 claims description 2
- 230000000007 visual effect Effects 0.000 claims 2
- 230000008859 change Effects 0.000 description 2
- 238000006243 chemical reaction Methods 0.000 description 2
- 238000013461 design Methods 0.000 description 2
- 238000001914 filtration Methods 0.000 description 1
- 230000005012 migration Effects 0.000 description 1
- 238000013508 migration Methods 0.000 description 1
- 230000001737 promoting effect Effects 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V20/00—Scenes; Scene-specific elements
- G06V20/20—Scenes; Scene-specific elements in augmented reality scenes
-
- 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
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V2201/00—Indexing scheme relating to image or video recognition or understanding
- G06V2201/07—Target detection
Abstract
本申请提供了一种基于稀疏卷积神经网络的目标检测方法及检测装置,目标检测方法包括:利用获取的目标场景图像构建数据集,并对数据集中每幅图像中包含的所有检测目标进行标注;根据检测目标在图像中的比例范围,优化基准卷积神经网络的结构,得到优化后的卷积神经网络;在构建的数据集上对优化后的卷积神经网络进行损失函数添加稀疏正则化项的预训练,得到卷积神经网络;对卷积神经网络进行卷积神经网络稀疏化,得到稀疏化的卷积神经网络;在数据集上对稀疏化的卷积神经网络进行微调训练,得到最终的稀疏卷积神经网络;利用最终的稀疏卷积神经网络对目标进行检测。本申请能够在廉价的硬件上针对大场景中的小物体进行快速、精准的目标检测。
Description
技术领域
本申请属于目标检测技术领域,特别适用于对大场景中的小物体进行检测,具体涉及一种基于稀疏卷积神经网络的目标检测方法及检测装置。
背景技术
现有的基于卷积神经网络的快速目标检测算法,通常通过基准模型(例如Yolo模型)在自定义数据集上的迁移学习训练来实现。其中,卷积神经网络基准模型有复杂的深层结构,从检测目标来说,检测的大多为尺度不一的目标,且目标在图像中不能太小;从检测速度来说,如果要达到小于5ms的超高速检测,则大多对硬件要求非常高。现有的基于卷积神经网络的目标检测方法对大场景中小物体的检测精度不高,且其神经网络结构对具体任务有较多冗余,严重影响检测速度。
发明内容
为至少在一定程度上克服相关技术中存在的问题,本申请提供了一种基于稀疏卷积神经网络的目标检测方法及检测装置。
根据本申请实施例的第一方面,本申请提供了一种基于稀疏卷积神经网络的目标检测方法,其包括以下步骤:
利用获取的目标场景图像构建数据集,并对数据集中每幅图像中包含的所有检测目标进行标注;
根据检测目标在图像中的比例范围,优化基准卷积神经网络的结构,得到优化后的卷积神经网络;
在构建的数据集上对优化后的卷积神经网络进行损失函数添加稀疏正则化项的预训练,得到卷积神经网络;
对卷积神经网络进行卷积神经网络稀疏化,得到稀疏化的卷积神经网络;
在数据集上对稀疏化的卷积神经网络进行微调训练,得到最终的稀疏卷积神经网络;
利用最终的稀疏卷积神经网络对目标进行检测。
上述基于稀疏卷积神经网络的目标检测方法中,所述目标场景图像包括多幅检测目标在场景中随机分布的图像和多幅无检测目标的场景图像。
进一步地,所述检测目标为不同颜色的乒乓球时,所述目标场景图像包括多幅乒乓球在全乒乓球桌视野范围中随机分布的图像和多幅全乒乓球桌视野范围中无乒乓球的图像,通过架设在乒乓球桌附近的相机获取目标场景图像。
上述基于稀疏卷积神经网络的目标检测方法中,所述对数据集中每幅图像中包含的所有检测目标进行标注包括标注检测目标的中心像素坐标以及检测目标所占的像素宽度和像素高度。
上述基于稀疏卷积神经网络的目标检测方法中,所述步骤根据检测目标在图像中的比例范围,优化基准卷积神经网络N的结构,得到优化后的卷积神经网络的具体过程为:
计算数据集的每幅图像中检测目标所占的比例;
其中,检测目标在目标场景图像中的宽度比例rw和高度比例rh为:
其中,wo表示目标场景图像中检测目标的宽度,wi表示目标场景图像的宽度,ho表示目标场景图像中检测目标的高度,hi表示目标场景图像的高度;
将满足基准卷积神经网络要求的最小输入大小的图像作为基准卷积神经网络的输入图像;
其中,基准卷积神经网络的输入图像的宽度nw和高度nh满足以下约束:
基准卷积神经网络N的输入图像的宽度nw和高度nh在符合基准卷积神经网络N要求的基础上,选取满足上述约束的最小值;
其中,wmin表示对检测目标进行检测所需的最小像素宽度,hmin表示对检测目标进行检测所需的最小像素高度;[rw1,rw2]表示所有检测目标在目标场景图像中的宽度比例所构成的宽度比例范围,[rh1,rh2]表示所有目标在目标场景图像中的高度比例所构成的高度比例范围;
优化基准卷积神经网络N的结构,得到优化后的卷积神经网络Ni,其具体过程为:
计算输入卷积神经网络的图像宽度缩放比例rwn和图像高度缩放比例rhn,
计算基准缩放比例rm,rm为满足以下约束的最大值:
将基准卷积神经网络N的维度最大的输出层的链接层更改链接到维度为(nw/rm)*(nh/rm)的网络层,修改相应的上采样倍率;将其他卷积神经网络输出层的链接层修改到同比例的链接层,得到优化后的卷积神经网络Ni。
上述基于稀疏卷积神经网络的目标检测方法中,所述步骤对卷积神经网络进行卷积神经网络稀疏化,得到稀疏化的卷积神经网络的具体过程为:
遍历搜索卷积神经网络Nt中的比例权重ai,将小于阈值ts的比例权重ai对应的网络结构i删除;其中,阈值ts的取值范围为(0,1);
对卷积神经网络Nt进行卷积神经网络稀疏化后得到稀疏化的卷积神经网络Ns_。
上述基于稀疏卷积神经网络的目标检测方法中,所述步骤在数据集上对稀疏化的卷积神经网络进行微调训练,得到最终的稀疏卷积神经网络的具体过程为:
在对优化后的卷积神经网络Ni进行预训练得到卷积神经网络Nt和稀疏化的卷积神经网络Ns_的基础上,在数据集D上通过损失函数L0进行卷积神经网络训练,更新参数数值,得到最终的稀疏卷积神经网络Ns。
上述基于稀疏卷积神经网络的目标检测方法中,所述步骤利用最终的稀疏卷积神经网络对目标进行检测的具体过程为:
将待检测图像缩放成大小为nw*nh的图像Iresized;
将缩放后的图像Iresized输入最终的稀疏卷积神经网络Ns中进行运算,得到最终的稀疏卷积神经网络Ns的输出结果Output;
将最终的稀疏卷积神经网络Ns的输出结果Output转换为目标检测结果,其中,目标检测结果为所有检测目标在图像中的中心像素坐标及检测目标所占的像素宽度和像素高度。
根据本申请实施例的第二方面,本申请还提供了一种基于稀疏卷积神经网络的目标检测装置,其包括:
存储器和处理器,
所述处理器被配置为基于存储在存储器中的指令,执行上述任一项所述的基于稀疏卷积神经网络的目标检测方法。
根据本申请实施例的第三方面,本申请还提供了一种计算机存储介质,其上存储有计算机程序,所述计算机程序被处理器执行时实现上述任一项所述的基于稀疏卷积神经网络的目标检测方法。
根据本申请的上述具体实施方式可知,至少具有以下有益效果:本申请基于稀疏卷积神经网络的目标检测方法通过改进现有卷积神经网络的结构,能够在廉价的硬件上针对大场景中的小物体进行快速、精准的目标检测。
应了解的是,上述一般描述及以下具体实施方式仅为示例性及阐释性的,其并不能限制本申请所欲主张的范围。
附图说明
下面的所附附图是本申请的说明书的一部分,其示出了本申请的实施例,所附附图与说明书的描述一起用来说明本申请的原理。
图1为本申请实施例提供的一种基于稀疏卷积神经网络的目标检测方法的流程图。
具体实施方式
为使本申请实施例的目的、技术方案和优点更加清楚明白,下面将以附图及详细叙述清楚说明本申请所揭示内容的精神,任何所属技术领域技术人员在了解本申请内容的实施例后,当可由本申请内容所教示的技术,加以改变及修饰,其并不脱离本申请内容的精神与范围。
本申请的示意性实施例及其说明用于解释本申请,但并不作为对本申请的限定。另外,在附图及实施方式中所使用相同或类似标号的元件/构件是用来代表相同或类似部分。
关于本文中所使用的“第一”、“第二”、…等,并非特别指称次序或顺位的意思,也非用以限定本申请,其仅为了区别以相同技术用语描述的元件或操作。
关于本文中所使用的方向用语,例如:上、下、左、右、前或后等,仅是参考附图的方向。因此,使用的方向用语是用来说明并非用来限制本创作。
关于本文中所使用的“包含”、“包括”、“具有”、“含有”等等,均为开放性的用语,即意指包含但不限于。
关于本文中所使用的“及/或”,包括所述事物的任一或全部组合。
关于本文中的“多个”包括“两个”及“两个以上”;关于本文中的“多组”包括“两组”及“两组以上”。
关于本文中所使用的用语“大致”、“约”等,用以修饰任何可以细微变化的数量或误差,但这些微变化或误差并不会改变其本质。一般而言,此类用语所修饰的细微变化或误差的范围在部分实施例中可为20%,在部分实施例中可为10%,在部分实施例中可为5%或是其他数值。本领域技术人员应当了解,前述提及的数值可依实际需求而调整,并不以此为限。
某些用以描述本申请的用词将于下或在此说明书的别处讨论,以提供本领域技术人员在有关本申请的描述上额外的引导。
本申请提供的基于稀疏卷积神经网络的目标检测方法,改进了现有卷积神经网络的结构,能够在廉价的硬件上,对大场景中的小物体进行快速、精准的目标检测。例如,本申请提供的基于稀疏卷积神经网络的目标检测方法能够在NVIDIA1050ti显卡上对全乒乓球桌视野范围内的乒乓球进行快速、精准的目标检测,其中,检测时间为2ms。
图1为本申请实施例提供的一种基于稀疏卷积神经网络的目标检测方法的流程图。
如图1所示,本申请提供的基于稀疏卷积神经网络的目标检测方法包括以下步骤:
S1、利用获取的目标场景图像构建数据集,并对数据集中每幅图像中包含的所有检测目标进行标注;
其中,目标场景图像包括多幅检测目标在场景中随机分布的图像和多幅无检测目标的场景图像。
数据集即多幅检测目标在场景中随机分布的图像和多幅无检测目标的场景图像的集合。
例如,检测目标为不同颜色的乒乓球,则目标场景图像包括多幅乒乓球在全乒乓球桌视野范围中随机分布的图像和多幅全乒乓球桌视野范围中无乒乓球的图像。
在发球机随机发球和运动员回球的场景下,可以通过架设在乒乓球桌附近的相机获取目标场景图像。
对数据集中每幅图像中包含的所有检测目标进行标注时,具体地,包括标注检测目标的中心像素坐标以及检测目标所占的像素宽度和像素高度。通常,标注时通过在图像上绘制包围检测目标的矩形来实现。
S2、根据检测目标在图像中的比例范围,优化基准卷积神经网络N的结构,得到优化后的卷积神经网络Ni,从而提升大场景下小目标的检测精度,其具体过程为:
S21、计算数据集的每幅图像中检测目标所占的比例;
假设目标场景图像中检测目标的宽度为wo,高度为ho;目标场景图像的宽度为wi,高度为hi,则检测目标在目标场景图像中的宽度比例rw和高度比例rh为:
所有检测目标在目标场景图像中的宽度比例构成宽度比例范围[rw1,rw2],所有目标在目标场景图像中的高度比例构成高度比例范围[rh1,rh2]。
S22、将满足基准卷积神经网络N要求的最小输入大小的图像作为基准卷积神经网络N的输入图像。
其中,基准卷积神经网络N的输入图像的宽度为nw,高度为nh,宽度nw和高度nh满足如下约束:
其中,wmin表示对检测目标进行检测所需的最小像素宽度,hmin表示对检测目标进行检测所需的最小像素高度。最小像素宽度wmin和最小像素高度hmin均为正整数,其具体取值根据待检测目标的特征复杂程度确定。
输入图像的宽度nw和高度nh在符合基准卷积神经网络N要求的基础上,选取满足上述约束的最小值,从而从网络输入的层面最大程度的减小计算量,提高运算速度。
以Yolov3-tiny模型作为基准卷积神经网络N为例,其输入图像的宽度nw和高度nh均需是32的整数倍。例如,对于乒乓球进行目标检测来说,可以取wmin=hmin=4。
S23、优化基准卷积神经网络N的结构,得到优化后的卷积神经网络Ni,其具体过程为:
计算输入卷积神经网络的图像宽度缩放比例rwn和图像高度缩放比例rhn,
计算基准缩放比例rm,其中,基准缩放比例rm为满足以下约束的最大值:
也就是说,基准缩放比例为:
将基准卷积神经网络N的维度最大的输出层的链接层更改链接到维度为(nw/rm)*(nh/rm)的网络层,修改相应的上采样倍率;将其他卷积神经网络输出层的链接层修改到同比例的链接层,得到优化后的卷积神经网络Ni。
以Yolov3-tiny模型为例,如果rm=4,则将基准卷积神经网络的输出层(即第23层)的链接层(即第20层)更改链接到维度为(nw/4)*(nh/4)的网络层(即第4层),即更改第20层的链接[-1,8]为[-1,4],同时修改第19层的上采样倍率(从2修改为8);Yolov3-tiny模型的其他卷积神经网络的输出层没有链接层,因此无需做其他改动。
S3、在构建的数据集上对优化后的卷积神经网络Ni进行损失函数添加稀疏正则化项的预训练,得到卷积神经网络Nt。
卷积神经网络Nt与优化后的卷积神经网络Ni具有相同的神经网络结构,只是其中的具体参数数值不同。损失函数L用于指导卷积神经网络训练过程中具体参数数值的更新。
假设原损失函数为L0,则新的损失函数L为:
L=L0+Lsr,
其中,Lsr为稀疏正则化项。例如,稀疏正则化项Lsr的表达式可以为:
其中,λ表示比例因子,其取值范围为:(0,1];γ表示卷积神经网络中的各项权重,Γ表示卷积神经网络中的所有权重构成的集合,g(γ)表示稀疏惩罚项。例如,稀疏惩罚项g(γ)可以为g(γ)=|γ|。
S4、对卷积神经网络Nt进行卷积神经网络稀疏化,去除卷积神经网络Nt中的冗余设计,得到稀疏化的卷积神经网络Ns_,从而在相同硬件上达到更快的检测速度,其具体过程为:
遍历搜索卷积神经网络Nt中的比例权重ai,将小于阈值ts的比例权重ai对应的网络结构i删除。其中,阈值ts的取值范围为(0,1),具体地,阈值ts的取值可以为0.02。需要说明的是,此处的比例权重ai与稀疏正则化项Lsr的表达式中的权重γ对应。
对卷积神经网络Nt进行卷积神经网络稀疏化后得到稀疏化的卷积神经网络Ns_。
S5、在数据集D上对稀疏化的卷积神经网络Ns_进行微调训练,得到最终的稀疏卷积神经网络Ns。
需要说明的是,微调训练指的是,从步骤S3对优化后的卷积神经网络Ni进行预训练得到卷积神经网络Nt和步骤S4得到稀疏化的卷积神经网络Ns_的基础上,在数据集D上通过损失函数L0进行卷积神经网络训练,从而更新参数数值得到最终的稀疏卷积神经网络Ns。其中,最终的稀疏卷积神经网络Ns与稀疏化的卷积神经网络Ns_具有相同的结构,不同的参数数值。
S6、利用最终的稀疏卷积神经网络Ns对目标进行检测,其具体过程为:
将待检测图像缩放成大小为nw*nh的图像Iresized。
将缩放后的图像Iresized输入最终的稀疏卷积神经网络Ns中进行运算,得到最终的稀疏卷积神经网络Ns的输出结果Output。
将最终的稀疏卷积神经网络Ns的输出结果Output转换为目标检测结果,其中,目标检测结果为所有检测目标在图像中的中心像素坐标及检测目标所占的像素宽度和像素高度。
需要说明的是,目标检测结果的具体转换方法与所选的基准卷积神经网络N的输出定义相关。每个基准卷积神经网络N均有其标准的转换方法,该方法不属于本申请的保护范围。
以Yolov3-tiny模型为例,从最终的稀疏卷积神经网络Ns的输出结果Output中可以直接得到多个目标检测结果框(包括中心像素坐标以及宽度和高度)以及每个目标检测结果框的概率;然后利用NMS(non-maximum-suppression,非极大值抑制)算法过滤其中的重复无效检测结果框,剩下的检测结果框即为最终的目标检测结果。
本申请基于稀疏卷积神经网络的目标检测方法通过对基准卷积神经网络N的结构进行优化,能够大大提升大场景下小目标物体的检测精度;通过对卷积神经网络进行稀疏化,去除卷积神经网络中的冗余设计,能够在相同的硬件上达到更快的检测速度;通过改进现有卷积神经网络的结构,能够在廉价的硬件上针对大场景中的小物体进行快速、精准的目标检测。
在示例性实施例中,本申请实施例还提供了一种基于稀疏卷积神经网络的目标检测装置,其包括存储器和处理器,处理器被配置为基于存储在存储器中的指令,执行本申请中任一个实施例中的基于稀疏卷积神经网络的目标检测方法。
其中,存储器可以为系统存储器或固定非易失性存储介质等,系统存储器可以存储有操作系统、应用程序、引导装载程序、数据库以及其他程序等。
在示例性实施例中,本申请实施例还提供了一种计算机存储介质,是计算机可读存储介质,例如,包括计算机程序的存储器,上述计算机程序可由处理器执行,以完成本申请中任一个实施例中的基于稀疏卷积神经网络的目标检测方法。
上述的本申请实施例可在各种硬件、软件编码或两者组合中进行实施。例如,本申请的实施例也可表示在数据信号处理器(Digital Signal Processor,DSP)中执行上述方法的程序代码。本申请也可涉及计算机处理器、数字信号处理器、微处理器或现场可编程门阵列(Field Programmable Gate Array,FPGA)执行的多种功能。可根据本申请配置上述处理器执行特定任务,其通过执行定义了本申请揭示的特定方法的机器可读软件代码或固件代码来完成。可将软件代码或固件代码发展表示不同的程序语言与不同的格式或形式。也可表示不同的目标平台编译软件代码。然而,根据本申请执行任务的软件代码与其他类型配置代码的不同代码样式、类型与语言不脱离本申请的精神与范围。
以上所述仅表示本申请示意性的具体实施方式,在不脱离本申请的构思和原则的前提下,任何本领域的技术人员所做出的等同变化与修改,均应属于本申请保护的范围。
Claims (10)
1.一种基于稀疏卷积神经网络的目标检测方法,其特征在于,包括以下步骤:
利用获取的目标场景图像构建数据集,并对数据集中每幅图像中包含的所有检测目标进行标注;
根据检测目标在图像中的比例范围,优化基准卷积神经网络的结构,得到优化后的卷积神经网络;
在构建的数据集上对优化后的卷积神经网络进行损失函数添加稀疏正则化项的预训练,得到卷积神经网络;
对卷积神经网络进行卷积神经网络稀疏化,得到稀疏化的卷积神经网络;
在数据集上对稀疏化的卷积神经网络进行微调训练,得到最终的稀疏卷积神经网络;
利用最终的稀疏卷积神经网络对目标进行检测。
2.根据权利要求1所述的基于稀疏卷积神经网络的目标检测方法,其特征在于,所述目标场景图像包括多幅检测目标在场景中随机分布的图像和多幅无检测目标的场景图像。
3.根据权利要求2所述的基于稀疏卷积神经网络的目标检测方法,其特征在于,所述检测目标为不同颜色的乒乓球时,所述目标场景图像包括多幅乒乓球在全乒乓球桌视野范围中随机分布的图像和多幅全乒乓球桌视野范围中无乒乓球的图像,通过架设在乒乓球桌附近的相机获取目标场景图像。
4.根据权利要求1所述的基于稀疏卷积神经网络的目标检测方法,其特征在于,所述对数据集中每幅图像中包含的所有检测目标进行标注包括标注检测目标的中心像素坐标以及检测目标所占的像素宽度和像素高度。
5.根据权利要求1所述的基于稀疏卷积神经网络的目标检测方法,其特征在于,所述步骤根据检测目标在图像中的比例范围,优化基准卷积神经网络的结构,得到优化后的卷积神经网络的具体过程为:
计算数据集的每幅图像中检测目标所占的比例;
其中,检测目标在目标场景图像中的宽度比例rw和高度比例rh为:
其中,wo表示目标场景图像中检测目标的宽度,wi表示目标场景图像的宽度,ho表示目标场景图像中检测目标的高度,hi表示目标场景图像的高度;
将满足基准卷积神经网络要求的最小输入大小的图像作为基准卷积神经网络的输入图像;
其中,基准卷积神经网络的输入图像的宽度nw和高度nh满足以下约束:
基准卷积神经网络N的输入图像的宽度nw和高度nh在符合基准卷积神经网络N要求的基础上,选取满足上述约束的最小值;
其中,wmin表示对检测目标进行检测所需的最小像素宽度,hmin表示对检测目标进行检测所需的最小像素高度;[rw1,rw2]表示所有检测目标在目标场景图像中的宽度比例所构成的宽度比例范围,[rh1,rh2]表示所有目标在目标场景图像中的高度比例所构成的高度比例范围;
优化基准卷积神经网络N的结构,得到优化后的卷积神经网络Ni,其具体过程为:
计算输入卷积神经网络的图像宽度缩放比例rwn和图像高度缩放比例rhn,
计算基准缩放比例rm,rm为满足以下约束的最大值:
将基准卷积神经网络N的维度最大的输出层的链接层更改链接到维度为(nw/rm)*(nh/rm)的网络层,修改相应的上采样倍率;将其他卷积神经网络输出层的链接层修改到同比例的链接层,得到优化后的卷积神经网络Ni。
6.根据权利要求1所述的基于稀疏卷积神经网络的目标检测方法,其特征在于,所述步骤对卷积神经网络进行卷积神经网络稀疏化,得到稀疏化的卷积神经网络的具体过程为:
遍历搜索卷积神经网络Nt中的比例权重ai,将小于阈值ts的比例权重ai对应的网络结构i删除;其中,阈值ts的取值范围为(0,1);
对卷积神经网络Nt进行卷积神经网络稀疏化后得到稀疏化的卷积神经网络Ns_。
7.根据权利要求1所述的基于稀疏卷积神经网络的目标检测方法,其特征在于,所述步骤在数据集上对稀疏化的卷积神经网络进行微调训练,得到最终的稀疏卷积神经网络的具体过程为:
在对优化后的卷积神经网络Ni进行预训练得到卷积神经网络Nt和稀疏化的卷积神经网络Ns_的基础上,在数据集D上通过损失函数L0进行卷积神经网络训练,更新参数数值,得到最终的稀疏卷积神经网络Ns。
8.根据权利要求1所述的基于稀疏卷积神经网络的目标检测方法,其特征在于,所述步骤利用最终的稀疏卷积神经网络对目标进行检测的具体过程为:
将待检测图像缩放成大小为nw*nh的图像Iresized;
将缩放后的图像Iresized输入最终的稀疏卷积神经网络Ns中进行运算,得到最终的稀疏卷积神经网络Ns的输出结果Output;
将最终的稀疏卷积神经网络Ns的输出结果Output转换为目标检测结果,其中,目标检测结果为所有检测目标在图像中的中心像素坐标及检测目标所占的像素宽度和像素高度。
9.一种基于稀疏卷积神经网络的目标检测装置,其特征在于,包括:
存储器和处理器,
所述处理器被配置为基于存储在存储器中的指令,执行权利要求1~8任一项所述的基于稀疏卷积神经网络的目标检测方法。
10.一种计算机存储介质,其特征在于,其上存储有计算机程序,所述计算机程序被处理器执行时实现权利要求1~8任一项所述的基于稀疏卷积神经网络的目标检测方法。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202010946308.0A CN112084950A (zh) | 2020-09-10 | 2020-09-10 | 基于稀疏卷积神经网络的目标检测方法及检测装置 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202010946308.0A CN112084950A (zh) | 2020-09-10 | 2020-09-10 | 基于稀疏卷积神经网络的目标检测方法及检测装置 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN112084950A true CN112084950A (zh) | 2020-12-15 |
Family
ID=73732893
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202010946308.0A Pending CN112084950A (zh) | 2020-09-10 | 2020-09-10 | 基于稀疏卷积神经网络的目标检测方法及检测装置 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN112084950A (zh) |
Citations (7)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN107609525A (zh) * | 2017-09-19 | 2018-01-19 | 吉林大学 | 基于剪枝策略构建卷积神经网络的遥感图像目标检测方法 |
CN108629295A (zh) * | 2018-04-17 | 2018-10-09 | 华中科技大学 | 转角地标识别模型训练方法、转角地标识别方法及装置 |
CN110147834A (zh) * | 2019-05-10 | 2019-08-20 | 上海理工大学 | 基于稀疏化双线性卷积神经网络的细粒度图像分类方法 |
CN110188720A (zh) * | 2019-06-05 | 2019-08-30 | 上海云绅智能科技有限公司 | 一种基于卷积神经网络的目标检测方法及系统 |
US20200151088A1 (en) * | 2018-11-14 | 2020-05-14 | The Mathworks, Inc. | Systems and methods for configuring programmable logic devices for deep learning networks |
CN111259923A (zh) * | 2020-01-06 | 2020-06-09 | 燕山大学 | 一种基于改进三维r-cnn算法的多目标检测方法 |
CN111494910A (zh) * | 2020-05-11 | 2020-08-07 | 上海庞勃特科技有限公司 | 步进电机循环间歇运动控制系统及控制方法 |
-
2020
- 2020-09-10 CN CN202010946308.0A patent/CN112084950A/zh active Pending
Patent Citations (7)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN107609525A (zh) * | 2017-09-19 | 2018-01-19 | 吉林大学 | 基于剪枝策略构建卷积神经网络的遥感图像目标检测方法 |
CN108629295A (zh) * | 2018-04-17 | 2018-10-09 | 华中科技大学 | 转角地标识别模型训练方法、转角地标识别方法及装置 |
US20200151088A1 (en) * | 2018-11-14 | 2020-05-14 | The Mathworks, Inc. | Systems and methods for configuring programmable logic devices for deep learning networks |
CN110147834A (zh) * | 2019-05-10 | 2019-08-20 | 上海理工大学 | 基于稀疏化双线性卷积神经网络的细粒度图像分类方法 |
CN110188720A (zh) * | 2019-06-05 | 2019-08-30 | 上海云绅智能科技有限公司 | 一种基于卷积神经网络的目标检测方法及系统 |
CN111259923A (zh) * | 2020-01-06 | 2020-06-09 | 燕山大学 | 一种基于改进三维r-cnn算法的多目标检测方法 |
CN111494910A (zh) * | 2020-05-11 | 2020-08-07 | 上海庞勃特科技有限公司 | 步进电机循环间歇运动控制系统及控制方法 |
Non-Patent Citations (1)
Title |
---|
王玺坤等: "基于改进型YOLO算法的遥感图像舰船检测", 《北京航空航天大学学报》, vol. 46, no. 06, pages 1184 - 1191 * |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN109934121B (zh) | 一种基于YOLOv3算法的果园行人检测方法 | |
CN110930454B (zh) | 一种基于边界框外关键点定位的六自由度位姿估计算法 | |
CN111160085A (zh) | 一种人体图像关键点姿态估计方法 | |
CN110659664B (zh) | 一种基于ssd的高精度识别小物体的方法 | |
CN112800964B (zh) | 基于多模块融合的遥感影像目标检测方法及系统 | |
CN111291637A (zh) | 一种基于卷积神经网络的人脸检测方法、装置及设备 | |
CN110443258B (zh) | 文字检测方法、装置、电子设备及存储介质 | |
CN110991560B (zh) | 一种结合上下文信息的目标检测方法及系统 | |
CN113095418B (zh) | 一种目标检测方法及系统 | |
CN111783772A (zh) | 一种基于RP-ResNet网络的抓取检测方法 | |
CN113362329A (zh) | 病灶检测模型的训练方法及识别图像中的病灶的方法 | |
CN113139543B (zh) | 目标对象检测模型的训练方法、目标对象检测方法和设备 | |
CN111274981B (zh) | 目标检测网络构建方法及装置、目标检测方法 | |
CN116645592B (zh) | 一种基于图像处理的裂缝检测方法和存储介质 | |
CN111598087B (zh) | 不规则文字的识别方法、装置、计算机设备及存储介质 | |
CN112800955A (zh) | 基于加权双向特征金字塔的遥感影像旋转目标检测方法及系统 | |
CN112418212A (zh) | 一种基于EIoU改进的YOLOv3算法 | |
CN110310305B (zh) | 一种基于bssd检测与卡尔曼滤波的目标跟踪方法与装置 | |
CN113850129A (zh) | 一种旋转等变的空间局部注意力遥感图像目标检测方法 | |
CN112149563A (zh) | 一种注意力机制人体图像关键点姿态估计方法及系统 | |
CN116994140A (zh) | 基于遥感影像的耕地提取方法、装置、设备和介质 | |
CN115810133B (zh) | 基于图像处理和点云处理的焊接控制方法及相关设备 | |
CN112734837A (zh) | 图像匹配的方法及装置、电子设备及车辆 | |
CN115995042A (zh) | 一种视频sar运动目标检测方法及装置 | |
CN114694005A (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 | ||
CB02 | Change of applicant information | ||
CB02 | Change of applicant information |
Address after: 257 jinzang Road, Pudong New Area, Shanghai, 201206 Applicant after: Shanghai Chuangyi Technology Co.,Ltd. Address before: 257 jinzang Road, Pudong New Area, Shanghai, 201206 Applicant before: Shanghai pont Technology Co., Ltd |