基于改进Cascade Mask R-CNN与协同注意力机制的群猪姿态识别

作者：王鲁; 刘晴; 曹月; 郝霞^*

来源：农业工程学报, 2023, 39(04): 144-153.

• 深度学习
• 图像识别
• 实例分割
• 群猪姿态识别
• 生猪个体提取

猪体姿态识别有助于实现猪只健康状况预警、预防猪病爆发，是当前研究热点。针对复杂场景下群猪容易相互遮挡、粘连，姿态识别困难的问题，该研究提出一种实例分割与协同注意力机制相结合的两阶段群猪姿态识别方法。首先，以Cascade Mask R-CNN作为基准网络，结合HrNetV2和FPN模块构建猪体检测与分割模型，解决猪体相互遮挡、粘连等问题，实现复杂环境下群猪图像的高精度检测与分割；在上述提取单只猪基础上，构建了基于协同注意力机制(coordinate attention,CA)的轻量级猪体姿态识别模型(CA-MobileNetV3)，实现猪体姿态的精准快速识别。最后，在自标注数据集上的试验结果表明，在猪体分割与检测环节，该研究所提模型与MaskR-CNN、MSR-CNN模型相比，在AP0.50、AP0.75、AP0.50:0.95和AP0.5:0.95-large指标上最多提升了1.3、1.5、6.9和8.8个百分点，表现出最优的分割与检测性能。而在猪体姿态识别环节，所提CA-MobileNetV3模型在跪立、站立、躺卧、坐立4种姿态类上的准确率分别为96.9%、99.1%、99.5%和98.6%，其性能优于主流的MobileNetV3、ResNet50、DenseNet121和VGG16模型，由此可知，该研究模型在复杂环境下群猪姿态识别具有良好的准确性和有效性，为实现猪体姿态的精准快速识别提供方法支撑。

• 出版日期2023
• 单位山东农业大学