猪三维点云体尺自动计算模型Pig Back Transformer

• Published in: 智慧农业（中英文） Vol. 6; no. 4; pp. 76 - 90
• Main Authors: 王宇啸, 石源源, 陈招达, 吴珍芳, 蔡更元, 张素敏, 尹令
• Format: Journal Article
• Language: Chinese
• Published: 中国农业科学院农业信息研究所 30.07.2024; 国家生猪种业工程技术研究中心,广东广州 510642,中国; 国家生猪种业工程技术研究中心,广东广州 510642,中国%华南农业大学 数学与信息学院,广东广州 510642,中国%国家生猪种业工程技术研究中心,广东广州 510642,中国; 猪禽种业全国重点实验室,广东广州 510640,中国%华南农业大学 数学与信息学院,广东广州 510642,中国; 华南农业大学 动物科学学院,广东广州 510642,中国; 猪禽种业全国重点实验室,广东广州 510640,中国
• Subjects: 三维点云; 自注意力机制; Transformer; Back; 体尺自动测量; 深度相机; Pig; 测量关键点定位; key point positioning; body size automic measurement; 3D point cloud; self-atten-tion mechanism; Pig Back Transformer; depth camera
• ISSN: 2096-8094
• DOI: 10.12133/j.smartag.SA202401023

[目的/意义]为了提高体尺关键点定位准确率，猪三维点云体尺自动测量方法会采用点云分割，在各个分割后局部点云定位测量关键点，以减少点云之间相互干扰。然而点云分割网络通常需要消耗较大计算资源，且现有测量点定位效果仍有待提升空间。本研究旨在通过设计关键点生成网络从猪体点云中提取出各体尺测量所需关键点。在降低显存资源需求的同时提高测量关键点定位效果，提高体尺测量的效率和精度。[方法]针对猪三维表面点云进行体尺测量，提出了一种定位猪体尺关键点的模型Pig Back Transformer。模型分为两个模块，分别设计了两种改进的Transformer自注意力编码器，第一模块为全局关键点模块，首先设计了一种猪背部边缘点提取算法用于获取边缘点，再使用edge encoder编码器以边缘点集合作为输入，edge encoder的edge attention中加入了边缘点和质点的偏移距离信息；第二模块为关键点生成模块，使用了back attention机制的back encoder，其中加入了与质心和第一模块生成的全局关键点的偏移量，并将偏移量与点云注意力通过按位max pooling操作结合，最后通过生成猪的体尺测量关键点和背脊走向点。最后设计了使用关键点和背脊走向点作为输入的体尺算法。[结果和讨论] 对比关键点和背脊走向点生成任务上Pig Back Transformer表现最佳，并对比体尺计算结果与人工测量结果，体长相对误差为0.63%，相对PointNet++、Point Transformer V2、Point Cloud Transforme、OctFormer PointTr等模型有较大提升。[结论]Pig Back Transformer能相对准确地生成猪体尺关键点，提高体尺测量数据准确度，并且通过点云特征定位体尺关键点节省了计算资源，为无接触牲畜体尺测量提供了新思路。