YOLO POD:一种快速准确的高密度大豆豆荚计数模型


发布时间:

2023-02-24

来源:

植物表型资讯

作者:

PhenoTrait

大豆豆荚数是大豆产量的重要指标之一,荚果计数在产量估算、栽培管理、品种选育等方面具有重要意义。手动计数豆荚既缓慢又费力。在作物计数中,利用目标检测网络是常用的方法,但由于豆荚分散重叠,使得豆荚的检测和计数困难。

 

基于YOLO X框架,提出了一种命名为YOLO POD的模型(图2),在三个不同数据集(图1)及数据增强(图4)上进行试验。在YOLO X的基础上,增加了预测豆荚数量的模块,修改了损失函数,采用了空间金字塔池-快速(Spatial Pyramid Pooling-Fast,SPPF,图3)结构,并引入了卷积块注意力模块(Convolutional block Attention Module, CBAM,图5),实现了豆荚的准确识别和计数,而不降低推断的速度。结果(图6和7)表明,YOLO POD预测数与地面真实值的R2达到0.967,较YOLO X提高了0.049,而推理时间仅增加了0.08 s。此外,MAE、MAPE、RMSE分别仅为4.18、10.0%、6.48,偏差很小。

 

本文首次实现了大豆豆荚的精确计数,为高密度物体的检测和计数提出了一种新的解决方案。

 

图1 三个不同数据集的大豆图像示例

 

图2 YOLO X的整体结构和子模块说明

 

图3 说明空间金字塔池和空间金字塔池-快速的区别。SPPB使用3个不同内核大小的池化层,而SPPBF使用3个连续的内核大小= 5的池化层

 

图4 YOLO POD图像增强

 

图5 YOLO POD检测头示意图。在YOLO X头部的基础上,增加了自注意模块和预测豆荚数的新分支

 

图6 在Renshou 2021 数据集的YOLO系列的检测结果示例。绿色方框表示正确检测到的豆荚,红色方框表示不正确的结果,蓝色方框表示丢失的豆荚

 

图7 不同模型的热图示例:一个输入图像;b YOLO X;c无数字预测模块的YOLO POD;d YOLO POD

 

 
来 源
Xiang S, Wang S, Xu M, Wang W, Liu W. YOLO POD: a fast and accurate multi-task model for dense Soybean Pod counting. Plant Methods. 2023 Jan 28;19(1):8. doi: 10.1186/s13007-023-00985-4.
 
编辑
王春颖
 

扩展阅读

推荐新闻

石时之约|韩志国:透过表型数据,看见植物的喜怒哀乐!

本期石时之约,我们将对话慧诺瑞德(北京)科技有限公司总经理、国际植物表型学会(IPPN)执委会委员/工业分会副主席韩志国,一起从表型数据的科学角度,去读懂农作物的喜怒哀乐和前世今生。

慧科研、慧育种、慧种田——慧聚改变的力量

让我们“慧聚”在一起,为“慧科研、慧育种、慧种田”赋能。

高通量植物表型平台建设注意事项

育种,是在给定的环境条件下,选择各种表型指标(产量、品质、抗性)最优的基因型材料的过程(AI育种,从这里起步)。育种工作中大约70%的工作量来自表型观察测量和筛选,是最耗人力物力的过程。

作物生理表型测量基础原理

生理表型测量的核心在于“早、快”,要在肉眼可见之前就能测量并预判出变化趋势,才是这个技术的核心价值。叶绿素荧光成像,恰好满足了这个要求。