基于Openpose改进的苹果生长方向检测

作者：李会宾; 史云^*; 刘怀洋; 王文昊; 刘万福; 杨鹏

来源：中国农业信息, 2022, 34(06): 34-48.

• 苹果
• 生长方向
• Openpose
• ShuffleNet V2,CA

【目的】苹果采摘机器人在果园作业过程中，不仅需要精准地确定果实的位置，还需要对苹果的生长方向进行精准检测，这样才能够实现机械爪规划出仿人式采摘的抓取方式，实现果实的高效采摘。【方法】文章提出了一种基于改进Openpose的方法，实现了对自然生长状态下果实生长方向的精准检测。改进方案主要从两个方面进行，首选是通过ShuffleNet V2和CA注意力机制相融合的方式替换原VGG19结构，降低主干的参数量，然后是结合单个苹果关键点的特征对该模型进行了去除部分PAF分支的优化改进策略。【结果】改进后的Openpose在各个指标上，全面超过了原Openpose，其运行速度是改进前的6.56倍，对于mAP、m AP-s、mAP-b、AP50和AP75这些参数，分别增长9.18%、8.50%、11.56%、1.67%和6.35%，并且在各项指标上超越了AlphaPose和CFA算法。【结论】经过对Openpose模型改进前后的对比和现有算法对比，证明了该文算法在自然场景下对果实生长方向检测的优越性和鲁棒性。

• 单位
  苏州大学; 中国农业科学院农业资源与农业区划研究所; 河海大学; 机电工程学院