CN113159215A - 一种基于Faster Rcnn的小目标检测识别方法 - Google Patents
一种基于Faster Rcnn的小目标检测识别方法 Download PDFInfo
- Publication number
- CN113159215A CN113159215A CN202110504503.2A CN202110504503A CN113159215A CN 113159215 A CN113159215 A CN 113159215A CN 202110504503 A CN202110504503 A CN 202110504503A CN 113159215 A CN113159215 A CN 113159215A
- Authority
- CN
- China
- Prior art keywords
- frame
- target
- network
- frames
- anchor
- 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
- 238000000034 method Methods 0.000 title claims abstract description 55
- 238000001514 detection method Methods 0.000 title claims abstract description 54
- 238000012549 training Methods 0.000 claims abstract description 15
- 238000007781 pre-processing Methods 0.000 claims abstract description 8
- 230000006870 function Effects 0.000 claims description 23
- 230000001629 suppression Effects 0.000 claims description 19
- 238000013527 convolutional neural network Methods 0.000 claims description 15
- 239000013598 vector Substances 0.000 claims description 15
- 238000013507 mapping Methods 0.000 claims description 7
- 238000012545 processing Methods 0.000 claims description 7
- 238000010586 diagram Methods 0.000 claims description 5
- GIYXAJPCNFJEHY-UHFFFAOYSA-N N-methyl-3-phenyl-3-[4-(trifluoromethyl)phenoxy]-1-propanamine hydrochloride (1:1) Chemical compound Cl.C=1C=CC=CC=1C(CCNC)OC1=CC=C(C(F)(F)F)C=C1 GIYXAJPCNFJEHY-UHFFFAOYSA-N 0.000 claims description 3
- 238000002372 labelling Methods 0.000 claims description 3
- 239000011159 matrix material Substances 0.000 claims description 3
- 230000007246 mechanism Effects 0.000 claims description 3
- 238000011176 pooling Methods 0.000 claims description 3
- 230000008569 process Effects 0.000 abstract description 7
- 238000004364 calculation method Methods 0.000 abstract description 3
- 230000009286 beneficial effect Effects 0.000 abstract description 2
- 238000013139 quantization Methods 0.000 abstract description 2
- 230000000052 comparative effect Effects 0.000 description 26
- 238000012360 testing method Methods 0.000 description 8
- 238000011156 evaluation Methods 0.000 description 7
- 210000000601 blood cell Anatomy 0.000 description 4
- 230000000694 effects Effects 0.000 description 4
- 238000002474 experimental method Methods 0.000 description 4
- 238000013135 deep learning Methods 0.000 description 3
- 238000005516 engineering process Methods 0.000 description 3
- 210000003743 erythrocyte Anatomy 0.000 description 3
- 210000000265 leukocyte Anatomy 0.000 description 3
- 238000013528 artificial neural network Methods 0.000 description 2
- 210000004027 cell Anatomy 0.000 description 2
- 230000008859 change Effects 0.000 description 2
- 230000002708 enhancing effect Effects 0.000 description 2
- 239000000284 extract Substances 0.000 description 2
- 238000000605 extraction Methods 0.000 description 2
- 238000000926 separation method Methods 0.000 description 2
- 238000012546 transfer Methods 0.000 description 2
- HPTJABJPZMULFH-UHFFFAOYSA-N 12-[(Cyclohexylcarbamoyl)amino]dodecanoic acid Chemical compound OC(=O)CCCCCCCCCCCNC(=O)NC1CCCCC1 HPTJABJPZMULFH-UHFFFAOYSA-N 0.000 description 1
- 238000004458 analytical method Methods 0.000 description 1
- 230000002238 attenuated effect Effects 0.000 description 1
- 239000003086 colorant Substances 0.000 description 1
- 238000012937 correction Methods 0.000 description 1
- 238000011161 development Methods 0.000 description 1
- 230000007613 environmental effect Effects 0.000 description 1
- 238000001914 filtration Methods 0.000 description 1
- 238000011478 gradient descent method Methods 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
- 238000005457 optimization Methods 0.000 description 1
- OLBCVFGFOZPWHH-UHFFFAOYSA-N propofol Chemical compound CC(C)C1=CC=CC(C(C)C)=C1O OLBCVFGFOZPWHH-UHFFFAOYSA-N 0.000 description 1
- 229960004134 propofol Drugs 0.000 description 1
- 238000011160 research Methods 0.000 description 1
- 238000005070 sampling Methods 0.000 description 1
- 238000012795 verification Methods 0.000 description 1
Images
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/24—Classification 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
- G06N3/084—Backpropagation, e.g. using gradient descent
-
- 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/40—Extraction of image or video features
- G06V10/46—Descriptors for shape, contour or point-related descriptors, e.g. scale invariant feature transform [SIFT] or bags of words [BoW]; Salient regional features
- G06V10/462—Salient features, e.g. scale invariant feature transforms [SIFT]
-
- 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
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- General Physics & Mathematics (AREA)
- Life Sciences & Earth Sciences (AREA)
- Artificial Intelligence (AREA)
- General Engineering & Computer Science (AREA)
- Evolutionary Computation (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Computational Linguistics (AREA)
- Evolutionary Biology (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Health & Medical Sciences (AREA)
- Biomedical Technology (AREA)
- Biophysics (AREA)
- Bioinformatics & Computational Biology (AREA)
- General Health & Medical Sciences (AREA)
- Mathematical Physics (AREA)
- Computing Systems (AREA)
- Molecular Biology (AREA)
- Software Systems (AREA)
- Multimedia (AREA)
- Image Analysis (AREA)
Abstract
本发明涉及一种基于Faster Rcnn的小目标检测识别方法,属于计算机视觉识别***技术领域,具体包括配置环境,搭建基于Faster Rcnn的小目标检测识别模型以及模型训练平台,小目标检测识别模型包括数据预处理;骨干网络、区域建议网络和头部网络。本申请的有益效果为:用resnet50作为主干网络提取图像特征,以及配合使用Roi Align方法,减小了网络的量化误差,还可以精确获得了原图中物体的位置坐标信息,采用DIOU算法,解决了在计算传统Iou时两框包含与分离时的值不变的问题。
Description
技术领域
本发明涉及计算机视觉识别***技术领域,特别涉及一种基于Faster Rcnn的小目标检测识别方法。
背景技术
目标检测是计算机视觉领域的一大基本任务,被广泛应用于人脸识别、自动驾驶、视频监控、军事侦察等领域。近年来,随着计算机信息技术的发展,深度学习被大量应用于目标检测任务中,使得目标检测完成了从手动提取人工设计的特征到应用卷积神经网络让计算机自动提取特征的转变,极大的提高了速度和准确度,使得基于深度学习的目标检测算法成为图像处理领域研究的主流。
基于深度学习的目标检测算法主要分为目标定位和目标识别这两个基础任务,目的就是对图像中的每一个目标进行精确的分类并用包围盒框出每一个目标具体的位置。目标比较流行的目标检测方法主要分为两类,其中一类是YOLO系列算法(YOLOv3、YOLOv4、YOLOv4_tiny),该系列算法是one-stage的,主要步骤是直接对输入图像进行卷积操作,提取图像特征并在特征图上进行边框回归;另一类是R-CNN系列算法(R-CNN,Fast R-CNN,Faster R-CNN),该系列算法是two-stage的,主要步骤是先通过各种图像处理技术生成可能包含目标的候选区域集合,再利用卷积神经网络对候选区域提取特征,最后进行精确的目标分类和边框回归。YOLO系列算法速度快,准确度要低一点,R-CNN系列算法速度慢,准确度要高一点。
尽管上述两类算法经过不断改进,在速度和精度上逐渐提升,但是对自然场景中的小型目标的检测依然存在精度不高或漏检等问题。小目标一般是指尺寸小于32*32像素的目标,由于小目标的像素数在原图中占比小,携带的特征少,纹理、形状、颜色等外观信息缺乏,再经过卷积神经网络中的下采样操作,特征信息逐步丢失,使得小目标的检测难度增大,检测精度普遍偏低。
综上所述,现有技术中针对小目标检测识别方法还多有不足,还需要进行改进。
发明内容
本发明的目的在于提供一种基于Faster Rcnn的小目标检测识别方法,以解决上述背景技术中提出的问题。
为实现上述目的,本发明采用如下技术方案:一种基于Faster Rcnn(FasterRegion-based ConvolutionalNetwork method,基于区域的更快卷积神经网络)的小目标检测识别方法,包括配置环境,搭建基于Faster Rcnn的小目标检测识别模型以及模型训练平台,所述小目标检测识别模型包括四个部分:
1)、数据预处理:对训练数据集中的原始图片进行数据增强,并对数据增强后的训练数据集中图像小目标进行标注,形成小目标数据集;数据预处理对网络模型是十分重要的,它直接影响到网络训练的结果,因为一般成功的神经网络都需要大量的参数,而这些参数都需要从数据只提取出来,但是有时训练数据集可能没有这么多数据,这就需要数据预处理来提高图片的数量。由于小目标在图中所占比例小,所以这里使用随机的翻转、错切、缩放、旋转、对比度拉伸、裁剪的一种或多种组合对原始图片进行数据增强的方法来提高小目标在图像中的多样性,给网络提供更多的特征数量。
2)、骨干网络:采用带有残差边的ResNet网络提取小目标数据集中图像特征;ResNet网络优选为ResNet50,其是由Conv Block(卷积块)和Identity Block(标识块)两个基本块组成,其中Conv Block(卷积块)中运用了直连通道传递低层特征,保证小目标特征不会严重丢失,改变了网络的维度。Identity Block(标识块)没有直连通道,只是加深网络的深度,精细提取物体特征,提高了分类精度。
3)、区域建议网络:在原始图像上提取出目标可能存在的候选区域,并在候选区域生成多个锚框,采用DIOU处理锚框并同时对锚框进行软非极大值抑制操作;
区域建议网络是根据卷积神经网络生成特征图的,先在原始图片上生成多个锚框;然后对锚框分别进行分类和回归,得到锚框中是否有物体和锚框的具***置信息。区域建议网络利用了滑动窗口机制,窗口每滑动一次可以产生k个锚框,然后进行有无物体的分类和位置信息的回归。分类分为正类和负类两个类,可以得到2k个分类输出。回归可以得到锚框的左上角和右下角的位置坐标信息,有4k个输出。
在区域建议网络中,我们需要对大量的锚框进行处理,比如对锚框正负样本的定义和对多余锚框的抑制(NMS,non maximum suppression,非极大值抑制)。
在现有的Faster Rcnn中,当预测框和真实框处于包含关系和分离关系时,现有算法并不能明确表示出两框的具***置关系,为了解决这个问题,本申请利用DIOU处理锚框,DIOU尤其对像素小的目标,提高了定位精度,使得对微小的偏差都有很强的鲁棒性。
具体的,DIOU公式如下:
IOU公式如下:
式(1)和式(2)中,A代表预测框(锚框),B代表真实框(标签值),∩代表取交集,∪代表取并集,d表示预测框和真实框中心点的距离,c表示外包框的对角线距离,IOU表示交并比,DIOU表示距离交并比;
在用DIOU处理锚框的基础上,还需要对得出的锚框进行非极大值抑制操作。由于Faster Rcnn是两阶段的检测算法,会得到大量的目标框,这些目标框都可以检测到目标,所以要从这些目标框中找出最符合实际的目标框,传统的方式是用非极大值抑制方法。
非极大值抑制的原理是利用目标框和真实框的重叠率排序,然后选出其中重叠率最大的目标框并把其余的目标框删除,选出的目标框就是最终的检测框。但是这种方法可能会导致误删目标框,因此,本申请对其进行了改进,采用soft_nms(软非极大值抑制),软非极大值抑制操作利用目标框和真实框的重叠率排序,然后选出其中重叠率最大的目标框并把其余的目标框删除,选出的目标框就是最终的检测框,公式为:
式(3)中:M为当前得分最高的框,bi为待处理的框,F为最终的得分,Nt为事先设定的阈值,当重叠率越大,F的值越高,重叠率越小,F的值越低,但是并不会为0,不会导致目标框误删除的结果;
在选出目标框后还需要对目标框进行微调操作,看出目标框和真实框是有一定差距的,为了使得最终的精度更高,就需要微调目标框,得到最终框。微调公式的过程如下:f(Px,Py,Pw,P)=(Zx,Zy,Zw,Zh) (4)
Zx=Px+Pwdx(P),Zy=Py+Phdy(P) (5)
式(4)(5)(6)中,(Px,Py,Pw,Ph)代表目标框的左上角坐标(Px,Py)和框的长(Pw)和高(Ph),(Zx,Zy,Zw,Zh)代表真实框的左上角坐标(Zx,Zy)和框的长(Zw)和框(Zh),dx、dy、dw、dh是四个映射函数,由以下公式得到:
式(7)-(10)中,t*代表预测值和真实值的差距,d*(P)是得到的预测值,对应dx或dy或dw或dh四个映射函数,是输入的特征向量,W*是要学习的参数,tx是x轴方向上的预测框与真实框的差距,ty是y轴方向上的预测框与真实框的差距,Gx是真实框左上角x轴坐标,Gy是真实框左上角y轴坐标,Px是目标框左上角x轴坐标,Py是目标框左上角y轴坐标,Pw是目标框的长,Ph是目标框的高,Gw是真实框的长,Gh是真实框的高,tw是目标框与真实框的长的差距,th是目标框与真实框的高的差距,Pi是输入的特征向量,γ是随机参数,W* T是W*的转置矩阵。
4)、头部网络:根据区域建议网络的结果,在特征图中将相应的区域转化为固定尺度的特征图,然后再进行精确的目标分类和目标定位;头部网络包含Roi Align和两个分支网络,在Roi Align中使用双三次插值来精确的寻找每个锚框块对应的特征,并从特征图中提取出固定长度的特征向量。然后把特征向量送入FC全连接层,最后再送入两个分支网络中,其中一个分支网络为分类网络,进行分类预测,输出具体的类别;另一个分支网络为边界回归网络,进行回归预测,输出目标精确的位置;每一个锚框块区域都定义了多任务的损失函数:L=Lcof+Llof,其中Lcof为分类损失函数,对应分类网络,Llof为定位损失函,对应边界回归网络,损失函数为现有技术,本申请不进行详细描述,其可以采用现有技术中。
头部网络中Roi Align先遍历每一个候选区域,并且保持浮点数;然后将候选区域分割成k*k个单元,最后利用双三次插值计算出每个单元中固定的四个坐标位置,并进行最大池化操作;双三次插值是一种复杂的插值方法,该方法利用待采样点周围确定的16个点的值进行三次插值,不仅考虑到了相邻点的灰度值的影响还考虑到了相邻点的灰度值变化率的影响。虽然加大了运算量,但是可以得到更接近高分辨率图像的效果,使得小目标的信息丢失较少。计算双三次插值的方法是先构造BiCubic(双三次)函数,从函数中获得周围16个像数点的权重值,然后再根据这到目标点的距离和权重值计算出最终的插值结果。
双三次插值公式为:
式(11)和(12)中,a取值为-0.5,W(i)表示横坐标权重,W(j)表示纵坐标权重,x代表某个像素点的像素值,aij代表第i行j列的像素点。
本申请的有益效果为:用resnet50作为主干网络提取图像特征,以及配合使用RoiAlign方法,减小了网络的量化误差,还可以精确获得了原图中物体的位置坐标信息,采用DIOU算法,解决了在计算传统IOU时两框包含与分离时的值不变的问题。实验结果表明,本申请的小目标检测识别方法平均识别精度高,消除了漏检问题,对很模糊的小目标也能有很好的识别精度。
附图说明
图1为本发明实施例1的模型结构。
图2为本发明对比例2的原理图。
图3为实施例1、对比例1、对比例2分别对BCCD血细胞数据集进行目标检测识别的输出结果图。
具体实施方式
下面通过对实施例的描述,对本发明作进一步详细说明,目的是帮助本领域的技术人员对本发明的构思、技术方案有更完整、准确和深入的理解,并有助于其实施。
一、测试平台和环境
测试平台配置如下:CoreTM i9-9900KF [email protected],32G内存Windows10操作***,显卡型号为RTX2080Ti.程序运行环境如下:Python3.6,keras版本为2.1.5,CUDA版本为10.0;选用的数据集为BCCD血细胞数据集。BCCD血细胞数据集是公知的数据集,该数据集共有三类364张图像:WBC(白细胞),RBC(红细胞)和Platelets。3个类别中有4888个标签(有0个空示例)。
为了提高对目标的检测精度和模型的泛化能力,在数据量有限的情况下,对原始的细胞数据集依次采取如下的几种数据增强方法,得到不同情况下的不同数据,增加数据的多样性。1)缩放:将图像随机缩小或放大;2)旋转:把图像按随机的角度进行旋转;3)随机裁剪:从图像中随机裁剪出一块区域。最终得到的数据集有2000张有效图片,12944个物体标注实例,并严格按照Pascal Voc2007的格式进行模型的训练。
二、实验细节及评价标准
本文网络优化采用SGD随机梯度下降法,对网络进行40000次迭代,其计算式如式(13)。其中每一批的训练数据包含128张图片,初始化学习率为0.0001,每4000次迭代后学习率衰减为原始学习率的一半,最后设置误差反向传播的动量因子为0.9.
式13中:θj为算法中的第j个参数,α为学习率,m为训练的样本数量,x(i)为第i组的x,y(i)为第j组的y。
实验采用AP平均精度对实验结构进行衡量:
式14-16中:A表示某一类的平均精度,M为全部样本中的正例数量,P为不同Recall下的最大Precision,Recall是召回率,代表分类器认为是正类并且确实是正类的部分占所有确实是正类的比例,Precision是精确度,代表分类器认为是正类并且确实是正类的部分占所有分类器认为是正类的比例,TP是分类器认为是正样本而且确实是正样本的例子,FP是分类器认为是正样本但实际上不是正样本的例子,FN是分类器认为是负样本但实际上不是负样本的例子。
三、实施例1
一种基于Faster Rcnn(Faster Region-based ConvolutionalNetwork method,基于区域的更快卷积神经网络)的小目标检测识别方法,包括配置环境,搭建如图1所示的基于Faster Rcnn的小目标检测识别模型以及搭建模型训练平台,所述小目标检测识别模型包括四个部分:
1)、数据预处理:对训练数据集中的原始图片进行数据增强,并对数据增强后的训练数据集中图像小目标进行标注,形成小目标数据集;数据预处理对网络模型是十分重要的,它直接影响到网络训练的结果,因为一般成功的神经网络都需要大量的参数,而这些参数都需要从数据只提取出来,但是有时训练数据集可能没有这么多数据,这就需要数据预处理来提高图片的数量。由于小目标在图中所占比例小,所以这里使用随机的翻转、错切、缩放、旋转、对比度拉伸、裁剪的一种或多种组合对原始图片进行数据增强的方法来提高小目标在图像中的多样性,给网络提供更多的特征数量。在本实施例中,数据集采用(一)测试平台和环境中所提供的增强数据集;
2)、骨干网络:采用带有残差边的ResNet网络提取小目标数据集中图像特征;ResNet网络优选为ResNet50,其是由Conv Block(卷积块)和Identity Block(标识块)两个基本块组成,其中Conv Block((卷积块))中运用了直连通道传递低层特征,保证小目标特征不会严重丢失,改变了网络的维度。Identity Block((标识块))没有直连通道,只是加深网络的深度,精细提取物体特征,提高了分类精度。
3)、区域建议网络:在原始图像上提取出目标可能存在的候选区域,并在候选区域生成多个锚框,采用DIOU处理锚框并同时对锚框进行软非极大值抑制操作;
区域建议网络是根据卷积神经网络生成特征图的,先在原始图片上生成多个锚框;然后对锚框分别进行分类和回归,得到锚框中是否有物体和锚框的具***置信息。区域建议网络利用了滑动窗口机制,窗口每滑动一次可以产生k个锚框,然后进行有无物体的分类和位置信息的回归。分类分为正类和负类两个类,可以得到2k个分类输出。回归可以得到锚框的左上角和右下角的位置坐标信息,有4k个输出。
在特征提取网络提取出来的特征图上,进行候选区域生成,提取出候选框。首先在特征图上经过滑窗操作,得到256维的特征,然后在这256维特征上经过2个FC全连接层,会在256维特征的每个点上画出预先设定的9个不同大小的框,并输出每个框的中心点坐标,框的长高和每个框中是否有物体的得分,如果有物体,则为1分,没物体则为0。
在区域建议网络中,我们需要对大量的锚框进行处理,比如对锚框正负样本的定义和对多余锚框的抑制(NMS,non maximum suppression,非极大值抑制)。
在现有的Faster Rcnn中,当预测框和真实框处于包含关系和分离关系时,现有算法并不能明确表示出两框的具***置关系,为了解决这个问题,本申请利用DIOU处理锚框,DIOU尤其对像素小的目标,提高了定位精度,使得对微小的偏差都有很强的鲁棒性。
具体的,DIOU公式如下:
IOU公式如下:
式(1)和式(2)中,A代表预测框(锚框),B代表真实框(标签值),∩代表取交集,∪代表取并集,d表示预测框和真实框中心点的距离,c表示外包框的对角线距离,IOU表示交并比,DIOU表示距离交并比;
在用DIOU处理锚框的基础上,还需要对得出的锚框进行非极大值抑制操作。由于Faster Rcnn是两阶段的检测算法,会得到大量的目标框,这些目标框都可以检测到目标,所以要从这些目标框中找出最符合实际的目标框,传统的方式是用非极大值抑制方法。
非极大值抑制的原理是利用目标框和真实框的重叠率排序,然后选出其中重叠率最大的目标框并把其余的目标框删除,选出的目标框就是最终的检测框。但是这种方法可能会导致误删目标框,因此,本申请对其进行了改进,采用soft_nms(软非极大值抑制),软非极大值抑制操作利用目标框和真实框的重叠率排序,然后选出其中重叠率最大的目标框并把其余的目标框删除,选出的目标框就是最终的检测框,公式为:
式(3)中:M为当前得分最高的框,bi为待处理的框,F为最终的得分,Nt为事先设定的阈值,当重叠率越大,F的值越高,重叠率越小,F的值越低,但是并不会为0,不会导致目标框误删除的结果;
在选出目标框后还需要对目标框进行微调操作,看出目标框和真实框是有一定差距的,为了使得最终的精度更高,就需要微调目标框,得到最终框。微调公式的过程如下:f(Px,Py,Pw,)=(Zx,Zy,Zw,Zh) (4)
Zx=Px+Pwdx(P),Zy=Py+Phdy(P) (5)
式(4)(5)(6)中,(Px,Py,Pw,Ph)代表目标框的左上角坐标(Px,Py)和框的长(Pw)和高(Ph),(Zx,Zy,Zw,Zh)代表真实框的左上角坐标(Zx,Zy)和框的长(Zw)和框(Zh),dx、dy、dw、dh是四个映射函数,由以下公式得到:
式(7)-(10)中,t*代表预测值和真实值的差距,d*(P)是得到的预测值,对应dx或dy或dw或dh四个映射函数,是输入的特征向量,W*是要学习的参数,tx是x轴方向上的预测框与真实框的差距,ty是y轴方向上的预测框与真实框的差距,Gx是真实框左上角x轴坐标,Gy是真实框左上角y轴坐标,Px是目标框左上角x轴坐标,Py是目标框左上角y轴坐标,Pw是目标框的长,Ph是目标框的高,Gw是真实框的长,Gh是真实框的高,tw是目标框与真实框的长的差距,th是目标框与真实框的高的差距,Pi是输入的特征向量,γ是随机参数,是W*的转置矩阵。
4)、头部网络:根据区域建议网络的结果,在特征图中将相应的区域转化为固定尺度的特征图,然后再进行精确的目标分类和目标定位;头部网络包含Roi Align和两个分支网络,针对区域建议网络RPN生成的大量的候选框,Roi Align结合实际的预选框的位置坐标,把候选框中得分为1的区域使用双三次插值来精确的寻找每个锚框块对应的特征,得到尺寸固定,且更贴和物体实际大小的特征区域,并从特征图中提取出固定长度的特征向量。然后把特征向量送入FC全连接层,最后再送入两个分支网络中,其中一个分支网络为分类网络,进行分类预测,输出具体的类别;另一个分支网络为边界回归网络,进行回归预测,输出目标精确的位置;每一个锚框块区域都定义了多任务的损失函数:L=Lcof+Llof,其中Lcof为分类损失函数,对应分类网络,Llof为定位损失函,对应边界回归网络,损失函数为现有技术,本申请不进行详细描述,其可以采用现有技术中。
头部网络中Roi Align先遍历每一个候选区域,并且保持浮点数;然后将候选区域分割成k*k个单元,最后利用双三次插值计算出每个单元中固定的四个坐标位置,并进行最大池化操作;双三次插值是一种复杂的插值方法,该方法利用待采样点周围确定的16个点的值进行三次插值,不仅考虑到了相邻点的灰度值的影响还考虑到了相邻点的灰度值变化率的影响。虽然加大了运算量,但是可以得到更接近高分辨率图像的效果,使得小目标的信息丢失较少。计算双三次插值的方法是先构造BiCubic(双三次)函数,从函数中获得周围16个像数点的权重值,然后再根据这到目标点的距离和权重值计算出最终的插值结果。
双三次插值公式为:
式(11)和(12)中,a取值为-0.5,W(i)表示横坐标权重,W(j)表示纵坐标权重,x代表某个像素点的像素值,aij代表第i行j列的像素点。
四、对比例1
本对比例采用现有技术的Faster Rcnn在(一)测试平台和环境基础上进行小目标检测识别。
Faster Rcnn的步骤为:
(1)输入测试图像;
(2)将整张图片输入CNN,进行特征提取;
(3)用RPN先生成一堆Anchor box,对其进行裁剪过滤后通过softmax判断anchors属于前景(foreground)或者后景(background),即是物体or不是物体,所以这是一个二分类;同时,另一分支bounding box regression修正anchor box,形成较精确的proposal(注:这里的较精确是相对于后面全连接层的再一次box regression而言)
(4)把建议窗口映射到CNN的最后一层卷积feature map上;
(5)通过RoI pooling层使每个RoI生成固定尺寸的feature map;
(6)利用Softmax Loss(探测分类概率)和Smooth L1 Loss(探测边框回归)对分类概率和边框回归(Bounding box regression)联合训练。
五、对比例2
本对比例采用现有技术中的YOLOv3方法在(一)测试平台和环境基础上进行小目标检测识别。
YOLOv3的原理图如图2所示,其属于现有技术,本对比例不再进行详细描述。
六、实验对比结果分析
实施例1、对比例1、对比例2在(一)测试平台和环境、(二)实验细节及评价标准的规则下,对BCCD血细胞数据集进行训练、目标检测识别和结果评价,目标检测识别输出结果如图3所示,评价结果如下表1所示:
表1:实施例1、对比例1、对比例2的评价结果
白细胞(%) | 红细胞(%) | 血小板(%) | mAP(%) | |
实施例1 | 99 | 80 | 96 | 91.86 |
对比例1 | 99 | 75 | 81 | 85.23 |
对比例2 | 95 | 79 | 69 | 80.91 |
由表1可见,本申请方法对属于小目标的血小板的检测精度最高,并且申请方法的总体精度达到了91.86%,远远高于现有的YOLOv3方法和Faster Rcnn方法;本申请的方法比现有的Faster Rcnn方法平均精度提升了7%,小目标的检测提升了15%,并且速度还没有降低,说明本申请的方法在小目标的检查上有很好的鲁棒性。
此外,在复杂密集的背景下检测目标时,不管是对大目标还是小目标,都会出现大量的漏检现象,效果很不好。图3中显示了实施例1、对比例1和2在细胞数据集中的测试输出结果,图3(a)为实施例1的小目标检测识别结果,可以看出该方法明显消除了漏检问题,对很模糊的小目标也能有很好的识别精度。图3(b)为对比例1的小目标检测识别结果,从中可以看出有不少漏检现象。图3(c)为对比例2的小目标检测识别结果,从中可以看出该方法存在大量的漏检现象,在小目标的检测上效果稍微好一点,但是着也是牺牲了大目标的检测精度换来的,只能检测出很小的一部分,而且检测的置信度也不高。
七、普适性验证
为了验证本申请的技术手段在其他数据集识别中的普适性,实施例1、对比例1、对比例2分别在一个新的遥感数据集上进行了对比实验。新的数据集是NWPU VHR-10,是由西北工业大学标注的航天遥感小目标检测数据集,该数据集有飞机、舰船、油罐、棒球场、网球场、篮球场、田径场、港口、桥梁、车辆10个类别,大部分图片都是小目标。实施例1、对比例1、对比例2的评价结果如表2所示。
表2:实施例1、对比例1、对比例2的评价结果
F1(%) | Recall(%) | mAP(%) | |
实施例1 | 79.11 | 91.2 | 89.21 |
对比例1 | 69.6 | 82.6 | 81.62 |
对比例2 | 74.8 | 71.04 | 79.56 |
从表2来看,本申请的精确明显高于对比例1和对比例2的方法,该对比实验验证了本申请的方法并不局限于专门的数据集,对其他数据集具有很好的普适性,可以用于大多数的小目标检测算法中。
以上对本发明进行了示例性描述,只要是采用了本发明的方法构思和技术方案进行的各种非实质性的改进;或未经改进,将本发明的上述构思和技术方案直接应用于其它场合的,均在本发明的保护范围之内。
Claims (5)
1.一种基于Faster Rcnn的小目标检测识别方法,其特征在于:包括配置环境,搭建基于Faster Rcnn的小目标检测识别模型以及模型训练平台,所述小目标检测识别模型包括四个部分:
1)、数据预处理:对训练数据集中的原始图片进行数据增强,并对数据增强后的训练数据集中图像小目标进行标注,形成小目标数据集;
2)、骨干网络:采用带有残差边的ResNet网络提取小目标数据集中图像特征;
3)、区域建议网络:在原始图像上提取出目标可能存在的候选区域,并在候选区域生成多个锚框,采用DIOU处理锚框并同时对锚框进行软非极大值抑制操作;
4)、头部网络:根据区域建议网络的结果,在特征图中将相应的区域转化为固定尺度的特征图,然后再进行精确的目标分类和目标定位;头部网络包含Roi Align和两个分支网络,在Roi Align中使用双三次插值来精确的寻找每个锚框块对应的特征,并从特征图中提取出固定长度的特征向量;然后把特征向量送入FC全连接层,最后再送入两个分支网络中,其中一个分支网络为分类网络,进行分类预测,输出具体的类别;另一个分支网络为边界回归网络,进行回归预测,输出目标精确的位置;每一个锚框块区域都定义了多任务的损失函数:L=Lcof+Llof,其中Lcof为分类损失函数,对应分类网络,Llof为定位损失函数,对应边界回归网络。
2.根据权利要求1所述的基于Faster Rcnn的小目标检测识别方法,其特征在于:原始图片进行数据增强的方式包括随机的翻转、错切、缩放、旋转、对比度拉伸、裁剪的一种或多种组合。
3.根据权利要求1所述的基于Faster Rcnn的小目标检测识别方法,其特征在于:所述ResNet网络为ResNet50。
4.根据权利要求1所述的基于Faster Rcnn的小目标检测识别方法,其特征在于:区域建议网络根据卷积神经网络生成特征图,先在原始图片上生成多个锚框;区域建议网络利用了滑动窗口机制,窗口每滑动一次可以产生k个锚框;利用DIOU定位锚框位置,利用软非极大值抑制操作确定最终的检测框;
DIOU公式如下:
IOU公式如下:
式(1)和式(2)中,A代表预测框(锚框),B代表真实框(标签值),∩代表取交集,∪代表取并集,d表示预测框和真实框中心点的距离,c表示外包框的对角线距离,IOU表示交并比,DIOU表示距离交并比;
所述软非极大值抑制操作利用目标框和真实框的重叠率排序,然后选出其中重叠率最大的目标框并把其余的目标框删除,选出的目标框就是最终的检测框,公式为:
式(3)中:M为当前得分最高的框,bi为待处理的框,F为最终的得分,Nt为事先设定的阈值,当重叠率越大,F的值越高,重叠率越小,F的值越低,但是并不会为0;
在选出目标框后还需要对目标框进行微调操作,微调公式如下:
f(Px,Py,Pw,)=(Zx,Zy,Zw,Zh) (4)
Zx=Px+Pwdx(P),Zy=Py+Phdy(P) (5)
式(4)(5)(6)中,(Px,Py,Pw,Ph)代表目标框的左上角坐标(Px,Py)和框的长(Pw)和高(Ph),(Zx,Zy,Zw,Zh)代表真实框的左上角坐标(Zx,Zy)和框的长(Zw)和框(Zh),dx、dy、dw、dh是四个映射函数,由以下公式得到:
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110504503.2A CN113159215A (zh) | 2021-05-10 | 2021-05-10 | 一种基于Faster Rcnn的小目标检测识别方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110504503.2A CN113159215A (zh) | 2021-05-10 | 2021-05-10 | 一种基于Faster Rcnn的小目标检测识别方法 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN113159215A true CN113159215A (zh) | 2021-07-23 |
Family
ID=76874095
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202110504503.2A Pending CN113159215A (zh) | 2021-05-10 | 2021-05-10 | 一种基于Faster Rcnn的小目标检测识别方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN113159215A (zh) |
Cited By (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113537119A (zh) * | 2021-07-28 | 2021-10-22 | 国网河南省电力公司电力科学研究院 | 基于改进Yolov4-tiny的输电线路连接部件检测方法 |
CN113673498A (zh) * | 2021-07-28 | 2021-11-19 | 浙江大华技术股份有限公司 | 目标检测方法、装置、设备和计算机可读存储介质 |
CN114005268A (zh) * | 2021-10-21 | 2022-02-01 | 广州通达汽车电气股份有限公司 | 一种公交车区间调度方法、装置、设备及存储介质 |
CN116665015A (zh) * | 2023-06-26 | 2023-08-29 | 中国科学院长春光学精密机械与物理研究所 | 一种基于YOLOv5的红外序列图像弱小目标检测方法 |
Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN110046572A (zh) * | 2019-04-15 | 2019-07-23 | 重庆邮电大学 | 一种基于深度学习的地标建筑物识别与检测方法 |
CN110503112A (zh) * | 2019-08-27 | 2019-11-26 | 电子科技大学 | 一种增强特征学习的小目标检测及识别方法 |
WO2020173036A1 (zh) * | 2019-02-26 | 2020-09-03 | 博众精工科技股份有限公司 | 基于深度学习的定位方法和*** |
-
2021
- 2021-05-10 CN CN202110504503.2A patent/CN113159215A/zh active Pending
Patent Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
WO2020173036A1 (zh) * | 2019-02-26 | 2020-09-03 | 博众精工科技股份有限公司 | 基于深度学习的定位方法和*** |
CN110046572A (zh) * | 2019-04-15 | 2019-07-23 | 重庆邮电大学 | 一种基于深度学习的地标建筑物识别与检测方法 |
CN110503112A (zh) * | 2019-08-27 | 2019-11-26 | 电子科技大学 | 一种增强特征学习的小目标检测及识别方法 |
Non-Patent Citations (4)
Title |
---|
冯冬青: "基于深度学习的船只光学遥感图像检测和分割", 《中国优秀博硕士学位论文全文数据库(硕士)工程科技Ⅱ辑》 * |
明悦著: "《多源视觉信息感知与识别》", 31 August 2020 * |
朱秀红: "基于深度卷积网络的输电线路变尺度多目标检测算法研究", 《中国优秀博硕士学位论文全文数据库(硕士) 工程科技II辑》 * |
黄宁霞等: "改进深度学习框架Faster RCNN的人行道障碍物目标检测", 《机械设计与研究》 * |
Cited By (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113537119A (zh) * | 2021-07-28 | 2021-10-22 | 国网河南省电力公司电力科学研究院 | 基于改进Yolov4-tiny的输电线路连接部件检测方法 |
CN113673498A (zh) * | 2021-07-28 | 2021-11-19 | 浙江大华技术股份有限公司 | 目标检测方法、装置、设备和计算机可读存储介质 |
CN114005268A (zh) * | 2021-10-21 | 2022-02-01 | 广州通达汽车电气股份有限公司 | 一种公交车区间调度方法、装置、设备及存储介质 |
CN116665015A (zh) * | 2023-06-26 | 2023-08-29 | 中国科学院长春光学精密机械与物理研究所 | 一种基于YOLOv5的红外序列图像弱小目标检测方法 |
CN116665015B (zh) * | 2023-06-26 | 2024-04-02 | 中国科学院长春光学精密机械与物理研究所 | 一种基于YOLOv5的红外序列图像弱小目标检测方法 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN109977918B (zh) | 一种基于无监督域适应的目标检测定位优化方法 | |
CN107609525B (zh) | 基于剪枝策略构建卷积神经网络的遥感图像目标检测方法 | |
CN110580699A (zh) | 基于改进Faster RCNN算法的病理图像细胞核检测方法 | |
CN113159215A (zh) | 一种基于Faster Rcnn的小目标检测识别方法 | |
CN111091105A (zh) | 基于新的边框回归损失函数的遥感图像目标检测方法 | |
CN110321967B (zh) | 基于卷积神经网络的图像分类改进方法 | |
CN110309842B (zh) | 基于卷积神经网络的物体检测方法及装置 | |
CN111738055B (zh) | 多类别文本检测***和基于该***的票据表单检测方法 | |
CN109446922B (zh) | 一种实时鲁棒的人脸检测方法 | |
CN113361432B (zh) | 一种基于深度学习的视频文字端到端检测与识别的方法 | |
CN110008899B (zh) | 一种可见光遥感图像候选目标提取与分类方法 | |
CN110569738A (zh) | 基于密集连接网络的自然场景文本检测方法、设备和介质 | |
CN111753787A (zh) | 一种分离式的交通标志检测与识别方法 | |
CN111753682A (zh) | 一种基于目标检测算法的吊装区域动态监控方法 | |
CN114332921A (zh) | 基于改进聚类算法的Faster R-CNN网络的行人检测方法 | |
CN112766136A (zh) | 一种基于深度学习的空间车位检测方法 | |
CN112560852A (zh) | 基于YOLOv3网络具有旋转适应能力的单阶段目标检测方法 | |
CN113177503A (zh) | 基于yolov5的任意朝向目标十二参数检测方法 | |
CN112733942A (zh) | 一种基于多级特征自适应融合的变尺度目标检测方法 | |
CN110852327A (zh) | 图像处理方法、装置、电子设备及存储介质 | |
CN105046278A (zh) | 基于Haar特征的Adaboost检测算法的优化方法 | |
CN115620180A (zh) | 一种基于改进YOLOv5的航拍图像目标检测方法 | |
CN113269049A (zh) | 一种用于检测手写汉字区域的方法 | |
CN110991374B (zh) | 一种基于rcnn的指纹奇异点检测方法 | |
Xiang et al. | A real-time vehicle traffic light detection algorithm based on modified YOLOv3 |
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 | ||
RJ01 | Rejection of invention patent application after publication | ||
RJ01 | Rejection of invention patent application after publication |
Application publication date: 20210723 |