CN109766856B - 一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法 - Google Patents

一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法 Download PDF

Info

Publication number
CN109766856B
CN109766856B CN201910040870.4A CN201910040870A CN109766856B CN 109766856 B CN109766856 B CN 109766856B CN 201910040870 A CN201910040870 A CN 201910040870A CN 109766856 B CN109766856 B CN 109766856B
Authority
CN
China
Prior art keywords
rgb
image
cnn
roi
depth 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.)
Active
Application number
CN201910040870.4A
Other languages
English (en)
Other versions
CN109766856A (zh
Inventor
薛月菊
朱勋沐
郑婵
杨晓帆
陈畅新
王卫星
甘海明
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
South China Agricultural University
Original Assignee
South China Agricultural University
Priority date (The priority date 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 date listed.)
Filing date
Publication date
Application filed by South China Agricultural University filed Critical South China Agricultural University
Priority to CN201910040870.4A priority Critical patent/CN109766856B/zh
Publication of CN109766856A publication Critical patent/CN109766856A/zh
Application granted granted Critical
Publication of CN109766856B publication Critical patent/CN109766856B/zh
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Images

Abstract

本发明公开了一种双流RGB‑D FasterR‑CNN的识别哺乳母猪姿态方法,即提出一种在特征提取阶段融合RGB‑D图像特征的端对端的双流RGB‑D Faster R‑CNN算法,用于识别自由栏母猪场景下的哺乳母猪站立、坐立、俯卧、腹卧和侧卧5类姿态。基于Faster R‑CNN,首先使用两个CNN网络分别提取RGB图像特征和深度图像特征;然后利用RGB‑D图像的映射关系,仅采用一个RPN网络生成RGB图像特征图和深度图像特征图的感兴趣区域;对感兴趣区域特征池化后,使用一个独立的网络层实现RGB‑D特征的拼接融合;最后在Fast R‑CNN阶段,引入NOC结构继续卷积提取融合后的特征,再送入分类器和回归器。本发明提供了一种融合了RGB‑D数据信息端对端的高精度、小模型和实时的母猪姿态识别方法,为进一步分析母猪行为奠定了基础。

Description

一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法
技术领域
本发明涉及计算机视觉中多模态的目标检测和识别技术领域,特别涉及一种基于Faster R-CNN目标检测算法,利用RGB-D数据,采用双流CNN提取 RGB-D特征后在特征提取阶段融合的端对端的哺乳母猪姿态识别方法。
背景技术
养猪场中猪的行为是其福利和健康状态的重要表现形式,直接影响到养猪场的经济效益。在动物行为的监测技术中,与传统的人工监测和传感器技术相比,使用计算机视觉进行自动识别是一种低成本、高效的、非接触式的方式,能持续提供有价值的行为信息。
近年来,使用计算机视觉对猪进行行为识别受到了广泛的研究。例如: 2018年华南农业大学薛月菊等人公开号为CN108830144A的专利利用深度图像数据,引入残差结构和CenterLoss改进Faster R-CNN算法来自动识别自由栏中哺乳母猪的五种姿态。该团队2017年公开号为CN201710890676的专利利用深度图像,首先采用DPM算法检测母猪,在对检测框使用CNN网络识别母猪姿态,公开号为CN107527351A的专利则利用RGB图像和FCN算法自动分割场景中的母猪。2018年华南农业大学肖德琴等人公开号为 CN108717523A的专利公开了一种基于机器视觉的母猪发情行为检测方法,该方法首先进行爬跨行为判断,然后根据发情母猪的位移运动信息,公猪的接近程度和爬跨频次等信息训练神经网络模型实现行为检测。2016年中国农业大学劳凤丹等人公开号为CN104881636A的专利公开了一种识别猪躺卧行为的方法及装置,该方法获取猪的深度图像和二值图像,对二值图像进行椭圆拟合,并计算深度图像特定点到地面的高度,最后通过阈值比较识别猪的躺卧行为。2017年江苏大学朱伟兴等人公开号为CN108090426A的专利公开了一种基于机器视觉的群养猪身份识别方法,通过采集群养猪的俯视视频,频域中提取猪个体的多尺度和多方向的特征,空间域提取猪个体微小细节特征,对特征降维后训练分类器,实现识别判段,此外该团队公开号为CN107679463A的专利公开了一种采用机器视觉技术识别群养猪攻击行为的分析方法,公开号为CN107437069A的专利公开了基于轮廓的猪只饮水识别方法等等。
目前基于计算机视觉的猪的行为识别研究中,大多研究仅使用RGB图像或深度图像数据进行研究,这导致在真实场景中难以获得鲁棒的特征表示,容易达到识别精度的瓶颈。摄像机将3维世界映射为2维空间的RGB图像会不可避免地导致信息丢失,使用深度图像来作为补偿这部分信息是可行的。此外,深度图像因缺少RGB图像的纹理、色彩等信息而缺失目标物体的细节信息特征,当目标物体形状高度相似时难以做精准的识别。特别地,在俯视拍摄的母猪姿态识别任务中,一方面,目标的高度是判断不同姿态的重要信息,这在RGB图像中是无法体现的,另一方面,母猪的部分姿态的高度和形状又是相近的(例如俯卧与腹卧),仅利用深度信息难以精准区分。
因此,提供一种高精度的双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法是本领域技术人员亟待解决的技术问题。
发明内容
有鉴于此,本发明的目的在于提供一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法,对自由栏中的哺乳母猪实现自动、更高精度和实时的姿态识别。为实现上述目的其具体方案如下:
本发明公开了一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法,包括如下步骤:
S1、采集哺乳母猪的RGB-D视频图像,包括RGB图像和深度图像,并建立母猪姿态识别RGB-D视频图像库;
S2、通过相机标定方法计算获得RGB图像与深度图像之间的映射关系;
S3、基于Faster R-CNN算法,使用两个CNN网络分别卷积RGB图像和深度图像,得到RGB图像特征图和深度图像特征图;
S4、仅使用一个RPN网络,以深度图像特征图为基础生成感兴趣区域 D-ROI,通过RGB图像与深度图像之间的映射关系,对每一个D-ROI一一对应生成RGB图像特征图的感兴趣区域RGB-ROI;
S5、对每一个D-ROI和RGB-ROI分别使用ROI Pooling层池化为固定大小,对池化后的每一组D-ROI和RGB-ROI特征图使用拼接融合方法,融合两者特征;
S6、对融合后的特征图使用NOC结构的Fast R-CNN进一步提取融合特征,在经过全局平均池化层之后交由分类器和回归器处理,得到双流RGB-D Faster R-CNN母猪姿态识别模型,并输出识别结果;
S7、从母猪姿态识别RGB-D视频图像库获取训练集和测试集,使用训练集训练双流RGB-D Faster R-CNN母猪姿态识别模型,使用测试集测试模型性能,最终筛选最佳性能模型。
优选的,所述步骤S1的具体过程如下:
S11、将RGB-D传感器固定俯视拍摄采集猪栏RGB-D视频图像;RGB-D 传感器不仅能捕获客观世界的色彩信息还能捕获目标的深度信息,其拍摄的 RGB图像包含了色彩、形状和纹理等信息,而深度图像包含了清晰的边缘信息和对光线鲁棒的深度信息;
S12、从获取的RGB-D视频图像数据中采样获取训练集和测试集,其中训练数据集占70%,测试集占30%以测试模型性能;
S13、在训练集和测试集中,对深度图像进行预处理,包括滤波去噪和图像增强,然后对预处理后的深度图像进行目标标注,即标注目标外包围框和姿态类别,RGB图像则无需处理;再对处理后的训练集数据进行旋转和镜像扩增用于训练模型。
优选的,所述步骤S2的具体过程如下:
S21、使用相机标定方法获得RGB图像的内参矩阵为Krgb,深度图像的内参矩阵Kd,针对相机标定使用的同一幅棋盘格图像获得RGB图像外参矩阵 Rrgb和Trgb,以及深度图像的外参矩阵Rd和Td;令RGB图像的非齐次像素坐标为Prgb=[Urgb,Vrgb,1]T,Urgb表示横坐标,Vrgb表示纵坐标,深度图像的非齐次像素坐标为Pd=[Ud,Vd,1]T,Ud表示横坐标,Vd表示纵坐标;则深度图像坐标映射到RGB图像坐标的旋转矩阵R,平移矩阵T分别为:
Figure GDA0003829443480000041
Figure GDA0003829443480000042
S22、深度图像的像素坐标与RGB图像的像素坐标的映射关系为:
Prgb=(R*Zd*Pd+T)/Zrgb
由上式,可以通过获取深度图像的坐标值Pd和其像素值Zd,以及拍摄距离Zrgb得到相应像素点映射对应的RGB图像的坐标值Prgb
优选的,所述步骤S3的具体过程如下:
在Faster R-CNN的共享卷积层部分,使用两个相同的CNN网络分别以深度图像和RGB图像作为输入,以深度图像作为输入的CNN网络为Conv-D,以RGB图像作为输入的CNN网络为Conv-RGB。
优选的,所述步骤S4的具体过程如下:
S41、在Faster R-CNN算法的RPN阶段,仅用一个RPN网络以Conv-D 输出的深度图像特征图作为输入生成深度图像的感兴趣区域D-ROI;
S42、利用RGB图像与深度图像之间的映射关系,对每一个D-ROI一一对应生成Conv-RGB输出的RGB图像特征图的RGB图像感兴趣区域 RGB-ROI。
优选的,所述步骤S5的具体过程如下:
S51、对每一组D-ROI与RGB-ROI使用ROI pooling层池化为固定大小;
S52、对池化后的D-ROI与RGB-ROI的特征图进行串联堆叠融合,即对相同的空间位置i,j(1≤i≤H,1≤j≤W)在通道上d(1≤d≤D)堆叠,对于两个通道数量为D的特征图,堆叠后输出特征图通道数量为2D:
Figure GDA0003829443480000051
其中,
Figure GDA0003829443480000052
分别表示融合前的RGB-ROI, D-ROI与融合后的特征图。
优选的,所述步骤S6的具体过程如下:
S61、对融合后的特征图继续使用由若干层卷积层组合而成的NOC结构卷积进一步提取融合特征;
S62、再使用全局平均池化层池化后输入给Fast R-CNN的分类器与回归器。
优选的,所述步骤S11的RGB-D传感器为Kinect2.0传感器。
优选的,所述步骤S3的Conv-D与Conv-RGB为相同的卷积结构且均为 ZF结构。
优选的,所述步骤S51的采用空间金字塔池化层方式,将特征图分为6*6 的网格,并对每一个网格采用最大值池化的方式生成固定大小6*6、通道数为 256的特征图。
与现有技术相比,本发明技术方案的有益效果是:
(1)本发明提出双流RGB-D Faster R-CNN算法,融合了RGB图像与深度图像各自的优越性,大大提高了识别精度,而不增加太多时间成本。
(2)本发明通过全卷积的结构设计大大压缩了模型大小同时保证了实时性。
(3)本发明建立了哺乳母猪RGB-D视频图像数据库,为后续基于RGB-D 视频图像的算法设计和模型训练提供数据源。
附图说明
为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本发明的实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据提供的附图获得其他的附图。
图1为本发明双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法的流程图;
图2附图为本发明双流RGB-D Faster R-CNN的母猪姿态识别模型结构图;
图3附图为本发明识别结果示意图。
具体实施方式
下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
深度学习融合不同的图像特征能获得更好的图像特征表示,在目标识别任务中,融合RGB-D特征可以提取RGB图像和深度图像特征之间的互补性,将有助于提高特征学习的鲁棒性,有助于获得具有目标辨识力的特征。本发明提出一种端对端的RGB-D特征提取阶段的融合策略,即首先分别使用两个 CNN网络提取两种数据的差异性特征,再对两者特征融合后继续使用CNN 网络提取两种数据特征之间的内在的互补性特征信息。最终基于FasterR-CNN算法,提出一种能充分利用RGB-D数据信息的双流RGB-D Faster R-CNN算法用于高精度的识别哺乳母猪姿态。
图1附图所示的阶段一为生成RGB-D视频图像的感兴趣区域ROI,首先分别使用深度图像流的CNN网络Conv-D以深度图像做输入提取深度图像特征,RGB图像流的CNN网络Conv-RGB以RGB图像做输入提取RGB图像特征。然后使用RPN网络以深度图像特征图作为输入生成感兴趣区域D-ROI,再利用RGB图像与深度图像之间的映射关系,即RGB-D映射关系,一一对应生成RGB-ROI,D-ROI与RGB-ROI的特征图分别使用ROI-Pooling层池化为固定大小尺寸,然后使用拼接融合两者的特征图。阶段二为Fast R-CNN对 ROI进行分类识别的阶段,融合后的特征继续使用NOC结构进行卷积,以进一步融合RGB图像与深度图像特征,提取鲁棒的RGB-D特征,最后交由分类器和回归器处理,输出识别结果。
深度学习融合不同的图像特征能获得更好的图像特征表示,在目标识别任务中,融合RGB-D特征可以提取RGB图像和深度图像特征之间的互补性,将有助于提高特征学习的鲁棒性,有助于获得具有目标辨识力的特征。本发明提出一种端对端的RGB-D特征提取阶段的融合策略,即首先分别使用两个 CNN网络提取两种数据的差异性特征,再对两者特征融合后继续使用CNN 网络提取两种数据特征之间的内在的互补性特征信息。最终基于FasterR-CNN算法,提出一种能充分利用RGB-D数据信息的双流RGB-D Faster R-CNN算法用于高精度的识别哺乳母猪姿态。该方法基于Nvidia GTX 980Ti 型号的GPU硬件平台,在Ubuntu14.04操作系统搭建Caffe深度学习框架,使用python作为编程语言,进行母猪姿态识别模型的训练和测试。
具体实现为:
步骤一、采集哺乳母猪的RGB-D视频图像,包括RGB图像和深度图像,并建立母猪姿态识别RGB-D视频图像库;
步骤二、通过相机标定计算获得RGB图像与深度图像之间的映射关系;
步骤三、基于Faster R-CNN算法,使用两个CNN网络分别卷积RGB图像和深度图像;
步骤四、仅使用一个RPN网络,以深度图像特征图为基础生成感兴趣区域D-ROI,通过RGB图像与深度图像之间的映射关系,对每一个D-ROI一一对应生成RGB图像特征图的感兴趣区域RGB-ROI;
步骤五、对每一个D-ROI和RGB-ROI使用ROI Pooling层池化为固定大小,对池化后的每一组D-ROI和RGB-ROI特征使用拼接融合方法,融合两者特征;
步骤六、对融合后的特征图继续使用由若干层卷积层组合而成的NOC结构卷积进一步提取融合特征,即RGB-D特征,在经过全局平均池化层之后交由分类器和回归器处理,得到双流RGB-D Faster R-CNN母猪姿态识别模型,并输出识别结果;
步骤七、使用母猪姿态识别RGB-D视频图像库中的训练集训练双流 RGB-D FasterR-CNN母猪姿态识别模型,使用测试集测试模型性能,最终筛选最佳性能模型。
步骤一的数据库建立方法具体包括:
1)对28栏猪进行数据采集,采集猪舍为自由栏,大小约为3.8m×2.0m,每栏内包含一只哺乳母猪和8-10头仔猪。使用微软的Kinect v2.0传感器,距离猪舍地面190cm~270cm的高度,自上而下俯视拍摄,以每秒5帧的速度获取RGB-D数据。其中获取的RGB图像像素为1080*1920,为了在后续的算法使用过程中节省显存(GPU memory)和加快对RGB图像处理的计算速度,将RGB图像缩放到540*960的分辨率大小。获取的深度图像像素为424*512 的分辨率,深度图像的像素值反映了物体到传感器之间的距离。
2)从拍摄的前三次共计21栏数据中,对连续的视频图像随机隔10~40 帧选取一组,共随机采样5类姿态的RGB-D图像组,站立2522组,坐立2568 组,俯卧2505组,腹卧2497组,侧卧2508组图片,总计12600组RGB-D 图像作为原始训练集。从第四次拍摄的7栏数据中随机采样站立1127组,坐立1033组,俯卧1151组,腹卧1076组,侧卧1146组图片,总计5533组RGB-D图像作为测试集,用于测试模型性能。其中每一组RGB-D图像包括一张RGB图像和其对应的深度图像。在总的数据集中,训练集约占70%,测试集约占30%。
3)首先对采样后获取的深度图像进行中值滤波和自适应直方图均衡化处理,以提高对比度,RGB图像则不做预处理。在人工标注阶段,对每一组 RGB-D视频图像的数据,我们对深度图像进行人工标注,即对数据集中的每一张深度图像标注母猪的外边界框,获取母猪在图中的坐标位置信息。为了增强后续模型训练的泛化能力和鲁棒性,实验中对原始训练集数据进行数据扩增处理,即进行90度,180度,270度的顺时针旋转以及水平镜像和垂直镜像处理。处理后的RGB-D数据达到75600组,作为训练数据集,用于训练模型。
表1哺乳母猪5类姿态介绍
Figure GDA0003829443480000101
步骤二的使用相机标定方法获得RGB-D映射关系的方法具体包括:
使用相机标定方法获得RGB图像的内参矩阵为Krgb,深度图像的内参矩阵Kd,针对同一幅棋盘格图像获得RGB外参矩阵Rrgb和Trgb,以及深度图像的外参矩阵Rd和Td,这里棋盘格图像为相机标定所使用的在实验中打印出来的棋盘格图像。令RGB图像的非齐次像素坐标为Prgb=[Urgb,Vrgb,1]T,Urgb表示横坐标,Vrgb表示纵坐标,深度图像的非齐次像素坐标为Pd=[Ud,Vd,1]T,Ud表示横坐标,Vd表示纵坐标。则深度图像坐标映射到RGB图像坐标的旋转矩阵R,平移矩阵T分别为:
Figure GDA0003829443480000111
Figure GDA0003829443480000112
故深度图像的像素坐标与RGB图像的像素坐标的映射关系为:
Prgb=(R*Zd*Pd+T)/Zrgb
由上式,我们可以通过获取深度图像的坐标值Pd和其像素值Zd,以及拍摄距离Zrgb得到相应像素点映射对应的RGB图像的坐标值Prgb
步骤三的基于Faster R-CNN算法,使用两个CNN网络分别卷积RGB图像和深度图像的方法具体包括:
1)基于Faster R-CNN算法以ZF网络为例,网络结构首先使用ZF网络结构的一系列的卷积层和maxpooling层来独立地处理这两种数据,提取两种图像数据的特征。其中Conv1~Conv5以及Pool1和Pool2组成Conv-D,用于提取深度图像特征,Conv1_1~Conv5_1以及Pool1_1和Pool2_1组成Conv-RGB,用于提取RGB图像特征。Conv-D输入为512*424*1的深度图像,输出为33*28 大小,通道为256的特征图,Conv-RGB输入为960*540*3的RGB图像,输出为61*35大小,通道为256的特征图,输出的特征图如图2所示。
步骤四的仅使用一个RPN网络RGB-D数据特征的ROI的方法具体包括:
1)在RPN阶段,双流输出的特征图共享一个RPN网络,即以深度图特征图为基础生成D-ROI,并通过RGB-D映射关系,对RGB图像特征图生成 RGB-ROI。其中对于RPN网络,在每个滑动窗口位置,分别取3种面积尺度 {96,192,384}及3种长宽比{1:1,1:3,3:1}的9个锚点。
步骤五的对每一组D-ROI和RGB-ROI池化为固定大小后进行特征融合的方法具体包括:
1)分别使用两个ROI-Pooling层(空间金字塔池化层)对生成的不同尺寸的D-ROI和RGB-ROI以H*W(H,W设置为6)的网格,并采用最大值池化方式,生成固定大小的特征图,即每一个ROI特征图均池化为大小为6*6,通道数为256的特征图。
2)在特征融合阶段。定义ROI融合层的融合函数
Figure GDA0003829443480000121
Figure GDA0003829443480000122
为RGB-ROI的特征图,
Figure GDA0003829443480000123
则为D-ROI的特征图,t为第t 组ROI特征图(本文实验中t为128),H和W分别为特征图尺寸的高度和宽度,D为通道数量,在本文的方法中RGB-ROI和D-ROI的尺寸在 ROI-Pooling之后是完全相同的(实验中设置为6*6)。融合后输出特征图
Figure GDA0003829443480000124
Figure GDA0003829443480000125
与输入特征图大小相同,亦为H,W,通道数则为D′。为方便讨论,不考虑是第几组特征(每一组都采用相同的特征融合方式),省略下标t用以分析。
拼接融合公式为Ycat=Fcat(Xrgb,Xd)。即串联堆叠两个特征图。对相同的空间位置i,j(1≤i≤H,1≤j≤W)在通道上d(1≤d≤D)堆叠,对于两个通道数量为D的特征图,堆叠后输出特征图通道数量为2D:
Figure GDA0003829443480000131
其中,
Figure GDA0003829443480000132
串联堆叠并不会直接使两特征图之间的信息流通,但通过后续的卷积层的卷积后,便可实现两者数据的信息流通和信息融合。如图2所示。
步骤六的对融合后的特征使用NOC结构的Fast R-CNN的方法具体包括:
对融合后的特征在Fast R-CNN阶段使用四个卷积层Conv6、Conv7、 Conv8与Conv9组成的NOC结构继续卷积融合后的特征图,以促进融合特征通道间信息的流通,从而进一步抽象出RGB-D数据的特征,再经过全局平均池化层后,最后连接Fast R-CNN的分类器和回归器。如图2所示。
步骤七的使用训练集训练双流RGB-D Faster R-CNN模型,测试集测试模型性能,最终筛选最佳性能模型方法具体包括:
使用准备的RGB-D数据库中的训练集进行模型训练,设置图像输入的 mini batch为1,冲量为0.9,权值衰减系数为5-4,最大迭代次数为14*105,基础学习率为10-4,衰减步长step为6*105,衰减系数gamma为0.1。在8*105次迭代后,每隔1*105次迭代保存一个模型,选取测试集精度最高的模型作为比较。以最佳的模型作为最终模型。
下面详细说明本发明的实验结果:
本发明采用业界公认的3个评价指标对测试集的母猪姿态识别结果进行统计,并将本文提出的方法与仅使用深度图像的方法(Depth image)、仅使用RGB图像的方法(RGBimage)、将RGB-D简单拼接作为四通道图像输入的前融合方法(RGB-D early fusion)以及采用了两个CNN、两个RPN和两个FastR-CNN在对输出结果融合的后融合的方法(RGB-Dlater fusion)比较,其中前融合与后融合中将深度图像分辨率缩放到540*960后与RGB图像配准作为输入,结果如下:
本发明采用AP(Average Precision,平均准确率)、MAP(Mean AveragePrecision,平均准确率均值)、识别速度和模型大小进行评价。如下表2所示:
表2模型的识别性能对比
Figure GDA0003829443480000141
Figure GDA0003829443480000151
本发明提出的方法在融合了RGB数据和深度图像数据后,对站立、坐立、俯卧、腹卧和侧卧五类姿态的AP(Average Precision,平均准确率)分别达到 99.74%、96.49%、90.77%、90.91%和99.45%,五类姿态的MAP(Mean Average Precision,平均准确率均值)达到95.47%,超过仅使用RGB图像的方法7.11%,超过仅使用深度图像的方法5.36%,超过前融合方法1.55%,超过后融合的方法0.15%。在识别速度达则到12.3FPS,能够满足实时性的识别需求。模型大小仅为70.1MB远远小于其他方法,展现出极大的优越性。综上所述,本发明的方法在识别精度和模型大小的性能优越,同时还保持了实时的识别性能。
以上对本发明所提供的一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法进行了详细介绍,本文中应用了具体个例对本发明的原理及实施方式进行了阐述,以上实施例的说明只是用于帮助理解本发明的方法及其核心思想;同时,对于本领域的一般技术人员,依据本发明的思想,在具体实施方式及应用范围上均会有改变之处,综上所述,本说明书内容不应理解为对本发明的限制。
在本文中,诸如第一和第二等之类的关系术语仅仅用来将一个实体或者操作与另一个实体或操作区分开来,而不一定要求或者暗示这些实体或操作之间存在任何这种实际的关系或者顺序。而且,术语“包括”、“包含”或者其任何其他变体意在涵盖非排他性的包含,从而使得包括一系列要素的过程、方法、物品或者设备不仅包括那些要素,而且还包括没有明确列出的其他要素,或者是还包括为这种过程、方法、物品或者设备所固有的要素。在没有更多限制的情况下,由语句“包括一个……”限定的要素,并不排除在包括所述要素的过程、方法、物品或者设备中还存在另外的相同要素。

Claims (10)

1.一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法,其特征在于,包括如下步骤:
S1、采集哺乳母猪的RGB-D视频图像,包括RGB图像和深度图像,并建立母猪姿态识别RGB-D视频图像库;
S2、通过相机标定方法计算获得RGB图像与深度图像之间的映射关系;
S3、基于Faster R-CNN算法,使用两个CNN网络分别卷积RGB图像和深度图像,得到RGB图像特征图和深度图像特征图;
S4、仅使用一个RPN网络,以深度图像特征图为基础生成感兴趣区域D-ROI,通过RGB图像与深度图像之间的映射关系,对每一个D-ROI一一对应生成RGB图像特征图的感兴趣区域RGB-ROI;
S5、对每一个D-ROI和RGB-ROI分别使用ROI Pooling层池化为固定大小,对池化后的每一组D-ROI和RGB-ROI特征图使用拼接融合方法,融合两者特征;
S6、对融合后的特征图使用NOC结构的Fast R-CNN进一步提取融合特征,在经过全局平均池化层之后交由分类器和回归器处理,得到双流RGB-D Faster R-CNN母猪姿态识别模型,并输出识别结果;
S7、从母猪姿态识别RGB-D视频图像库获取训练集和测试集,使用训练集训练双流RGB-D Faster R-CNN母猪姿态识别模型,使用测试集测试模型性能,最终筛选最佳性能模型。
2.根据权利要求1所述的一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法,其特征在于,所述步骤S1的具体过程如下:
S11、将RGB-D传感器固定俯视拍摄采集猪栏RGB-D视频图像;
S12、从获取的RGB-D视频图像数据中采样获取训练集和测试集,其中训练数据集占70%,测试集占30%以测试模型性能;
S13、在训练集和测试集中,对深度图像进行预处理,包括滤波去噪和图像增强,然后对预处理后的深度图像进行目标标注,即标注目标外包围框和姿态类别,RGB图像则无需处理;再对处理后的训练集数据进行旋转和镜像扩增用于训练模型。
3.根据权利要求1所述的一种双流RGB-D FasterR-CNN识别哺乳母猪姿态的方法,其特征在于,所述步骤S2的具体过程如下:
S21、使用相机标定方法获得RGB图像的内参矩阵为Krgb,深度图像的内参矩阵Kd,针对相机标定使用的同一幅棋盘格图像获得RGB图像外参矩阵Rrgb和Trgb,以及深度图像的外参矩阵Rd和Td;令RGB图像的非齐次像素坐标为Prgb=[Urgb,Vrgb,1]T,Urgb表示横坐标,Vrgb表示纵坐标,深度图像的非齐次像素坐标为Pd=[Ud,Vd,1]T,Ud表示横坐标,Vd表示纵坐标;则深度图像坐标映射到RGB图像坐标的旋转矩阵R,平移矩阵T分别为:
Figure FDA0003829443470000021
Figure FDA0003829443470000022
S22、深度图像的像素坐标与RGB图像的像素坐标的映射关系为:
Prgb=(R*Zd*Pd+T)/Zrgb
由上式,可以通过获取深度图像的坐标值Pd和其像素值Zd,以及拍摄距离Zrgb得到相应像素点映射对应的RGB图像的坐标值Prgb
4.根据权利要求1所述的一种双流RGB-D FasterR-CNN识别哺乳母猪姿态的方法,其特征在于,所述步骤S3的具体过程如下:
在Faster R-CNN的共享卷积层部分,使用两个相同的CNN网络分别以深度图像和RGB图像作为输入,以深度图像作为输入的CNN网络为Conv-D,以RGB图像作为输入的CNN网络为Conv-RGB。
5.根据权利要求4所述的一种双流RGB-D FasterR-CNN识别哺乳母猪姿态的方法,其特征在于,所述步骤S4的具体过程如下:
S41、在Faster R-CNN算法的RPN阶段,仅用一个RPN网络以Conv-D输出的深度图像特征图作为输入生成深度图像的感兴趣区域D-ROI;
S42、利用RGB图像与深度图像之间的映射关系,对每一个D-ROI一一对应生成Conv-RGB输出的RGB图像特征图的RGB图像感兴趣区域RGB-ROI。
6.根据权利要求1所述的一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法,其特征在于,所述步骤S5的具体过程如下:
S51、对每一组D-ROI与RGB-ROI使用ROI pooling层池化为固定大小;
S52、对池化后的D-ROI与RGB-ROI的特征图进行串联堆叠融合,即对相同的空间位置i,j,在通道上d堆叠,其中1≤i≤H,1≤j≤W,1≤d≤D,对于两个通道数量为D的特征图,堆叠后输出特征图通道数量为2D:
Figure FDA0003829443470000031
其中,
Figure FDA0003829443470000032
分别表示融合前的RGB-ROI,D-ROI与融合后的特征图。
7.根据权利要求1所述的一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法,其特征在于,所述步骤S6的具体过程如下:
S61、对融合后的特征图继续使用由若干层卷积层组合而成的NOC结构卷积进一步提取融合特征;
S62、再使用全局平均池化层池化后输入给Fast R-CNN的分类器与回归器。
8.根据权利要求2所述的一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法,其特征在于,所述步骤S11的RGB-D传感器为Kinect2.0传感器。
9.根据权利要求4所述的一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法,其特征在于,所述步骤S3的Conv-D与Conv-RGB为相同的卷积结构且均为ZF结构。
10.根据权利要求6所述的一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法,其特征在于,所述步骤S51的采用空间金字塔池化层方式,将特征图分为6*6的网格,并对每一个网格采用最大值池化的方式生成固定大小6*6、通道数为256的特征图。
CN201910040870.4A 2019-01-16 2019-01-16 一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法 Active CN109766856B (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN201910040870.4A CN109766856B (zh) 2019-01-16 2019-01-16 一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN201910040870.4A CN109766856B (zh) 2019-01-16 2019-01-16 一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法

Publications (2)

Publication Number Publication Date
CN109766856A CN109766856A (zh) 2019-05-17
CN109766856B true CN109766856B (zh) 2022-11-15

Family

ID=66452306

Family Applications (1)

Application Number Title Priority Date Filing Date
CN201910040870.4A Active CN109766856B (zh) 2019-01-16 2019-01-16 一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法

Country Status (1)

Country Link
CN (1) CN109766856B (zh)

Families Citing this family (11)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN110309786B (zh) * 2019-07-03 2023-04-07 华南农业大学 一种基于深度视频的哺乳母猪姿态转换识别方法
CN110532854B (zh) * 2019-07-11 2021-11-26 中国农业大学 一种生猪爬跨行为检测方法及系统
CN110378953B (zh) * 2019-07-17 2023-05-02 重庆市畜牧科学院 一种智能识别猪群圈内空间分布行为的方法
CN110598658B (zh) * 2019-09-18 2022-03-01 华南农业大学 一种母猪哺乳行为的卷积网络识别方法
CN111104921A (zh) * 2019-12-30 2020-05-05 西安交通大学 一种基于Faster rcnn的多模态行人检测模型及方法
CN111368666B (zh) * 2020-02-25 2023-08-18 上海蠡图信息科技有限公司 一种基于新型池化及注意力机制双流网络的活体检测方法
CN111753658A (zh) * 2020-05-20 2020-10-09 高新兴科技集团股份有限公司 一种睡岗告警方法、装置和计算机设备
CN112088795B (zh) * 2020-07-07 2022-04-29 南京农业大学 基于激光定位的限位栏猪只姿态识别方法和系统
CN112101259A (zh) * 2020-09-21 2020-12-18 中国农业大学 一种基于堆叠沙漏网络的单只猪体姿态识别系统及方法
CN113313688B (zh) * 2021-05-28 2022-08-05 武汉乾峯智能科技有限公司 一种含能材料药桶识别方法、系统、电子设备及存储介质
CN116519106B (zh) * 2023-06-30 2023-09-15 中国农业大学 一种用于测定生猪体重的方法、装置、存储介质和设备

Citations (13)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN102282570A (zh) * 2008-10-30 2011-12-14 聪慧系统公司 用于立体视场多动物行为特征的系统和方法
CN102521563A (zh) * 2011-11-19 2012-06-27 江苏大学 基于椭圆拟合的猪行走姿态识别方法
CN104881636A (zh) * 2015-05-08 2015-09-02 中国农业大学 识别猪躺卧行为的方法及装置
CN106295558A (zh) * 2016-08-08 2017-01-04 华南农业大学 一种猪只行为节律分析方法
CN106456057A (zh) * 2014-03-21 2017-02-22 凯耐特赛斯公司 用于评判哺乳动物动力学的动作捕获和分析系统
CN106778784A (zh) * 2016-12-20 2017-05-31 江苏大学 基于机器视觉的猪个体识别与饮水行为分析方法
CN106815579A (zh) * 2017-01-22 2017-06-09 深圳市唯特视科技有限公司 一种基于多区域双流卷积神经网络模型的动作检测方法
CN107527351A (zh) * 2017-08-31 2017-12-29 华南农业大学 一种融合fcn和阈值分割的哺乳母猪图像分割方法
CN107844797A (zh) * 2017-09-27 2018-03-27 华南农业大学 一种基于深度图像的哺乳母猪姿态自动识别的方法
CN108074224A (zh) * 2016-11-09 2018-05-25 环境保护部环境规划院 一种陆生哺乳动物与鸟类的监测方法及其监测装置
CN108830144A (zh) * 2018-05-03 2018-11-16 华南农业大学 一种基于改进Faster-R-CNN的哺乳母猪姿态识别方法
CN108846326A (zh) * 2018-05-23 2018-11-20 盐城工学院 猪只体态识别方法、装置及电子设备
CN108921037A (zh) * 2018-06-07 2018-11-30 四川大学 一种基于BN-inception双流网络的情绪识别方法

Patent Citations (13)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN102282570A (zh) * 2008-10-30 2011-12-14 聪慧系统公司 用于立体视场多动物行为特征的系统和方法
CN102521563A (zh) * 2011-11-19 2012-06-27 江苏大学 基于椭圆拟合的猪行走姿态识别方法
CN106456057A (zh) * 2014-03-21 2017-02-22 凯耐特赛斯公司 用于评判哺乳动物动力学的动作捕获和分析系统
CN104881636A (zh) * 2015-05-08 2015-09-02 中国农业大学 识别猪躺卧行为的方法及装置
CN106295558A (zh) * 2016-08-08 2017-01-04 华南农业大学 一种猪只行为节律分析方法
CN108074224A (zh) * 2016-11-09 2018-05-25 环境保护部环境规划院 一种陆生哺乳动物与鸟类的监测方法及其监测装置
CN106778784A (zh) * 2016-12-20 2017-05-31 江苏大学 基于机器视觉的猪个体识别与饮水行为分析方法
CN106815579A (zh) * 2017-01-22 2017-06-09 深圳市唯特视科技有限公司 一种基于多区域双流卷积神经网络模型的动作检测方法
CN107527351A (zh) * 2017-08-31 2017-12-29 华南农业大学 一种融合fcn和阈值分割的哺乳母猪图像分割方法
CN107844797A (zh) * 2017-09-27 2018-03-27 华南农业大学 一种基于深度图像的哺乳母猪姿态自动识别的方法
CN108830144A (zh) * 2018-05-03 2018-11-16 华南农业大学 一种基于改进Faster-R-CNN的哺乳母猪姿态识别方法
CN108846326A (zh) * 2018-05-23 2018-11-20 盐城工学院 猪只体态识别方法、装置及电子设备
CN108921037A (zh) * 2018-06-07 2018-11-30 四川大学 一种基于BN-inception双流网络的情绪识别方法

Non-Patent Citations (2)

* Cited by examiner, † Cited by third party
Title
Automatic recognition of lactating sow postures from depth images by deep;Chan Zheng etc.;《Computers and Electronics in Agriculture》;20181231;51-63 *
基于改进Faster R-CNN 识别深度视频图像哺乳母猪姿态;薛月菊等;《农业工程学报》;20180531;第34卷(第9期);189-195 *

Also Published As

Publication number Publication date
CN109766856A (zh) 2019-05-17

Similar Documents

Publication Publication Date Title
CN109766856B (zh) 一种双流RGB-D Faster R-CNN识别哺乳母猪姿态的方法
Tian et al. Automated pig counting using deep learning
CN108830144B (zh) 一种基于改进Faster-R-CNN的哺乳母猪姿态识别方法
Wang et al. A deep learning approach incorporating YOLO v5 and attention mechanisms for field real-time detection of the invasive weed Solanum rostratum Dunal seedlings
Jay et al. In-field crop row phenotyping from 3D modeling performed using Structure from Motion
Nielsen et al. Vision-based 3D peach tree reconstruction for automated blossom thinning
Zhu et al. Automatic recognition of lactating sow postures by refined two-stream RGB-D faster R-CNN
CN107844797A (zh) 一种基于深度图像的哺乳母猪姿态自动识别的方法
CN110070571B (zh) 一种基于深度相机的毛竹形态参数检测方法
CN111507179A (zh) 一种生猪采食行为分析方法
CN114241031A (zh) 基于双视角融合的鱼类体尺测量和体重预测的方法及装置
Badhan et al. Real-time weed detection using machine learning and stereo-vision
CN116071424A (zh) 基于单目视觉的果实空间坐标定位方法
Wang et al. Using an improved YOLOv4 deep learning network for accurate detection of whitefly and thrips on sticky trap images
Buddha et al. Weed detection and classification in high altitude aerial images for robot-based precision agriculture
CN116012700A (zh) 一种基于YOLO-v5的鱼病实时检测系统
Amean et al. Automatic leaf segmentation and overlapping leaf separation using stereo vision
Wang et al. Accurate detection and precision spraying of corn and weeds using the improved YOLOv5 model
CN109166127B (zh) 一种可穿戴式植物表型感知系统
CN112883915A (zh) 一种基于迁移学习的麦穗自动识别方法和系统
CN111985472A (zh) 基于人工智能和主动球机的食槽干草温度图像处理方法
CN110969182A (zh) 基于农田图像的卷积神经网络构建方法及其系统
CN115457437A (zh) 作物识别方法、装置、系统及农药喷施机器人
CN108967246B (zh) 虾苗定位方法
CN112308002A (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
GR01 Patent grant
GR01 Patent grant