西瓜计数模型

轻量高精度

在复杂农田环境下实现快速准确的西瓜检测与跟踪，同时大幅减小模型体积和计算量。

Huijun, Yin · 王宝丽 · 景运革 · 李菊霞 · Pengling, Wang · Gaoxiang, Quan · Tingting, Sun

Nongye Gongcheng Xuebao/Transactions of the Chinese Society of Agricultural Engineering 2024

参数信息

平均计数准确率

96.3 %

平均绝对误差

0.77

决定系数

0.93

平均精度 (mAP0.5)

98.2 %

模型尺寸减小量

18.88 MB

参数量减小量

0.57 M

多目标跟踪准确率

91.2 %

多目标跟踪精确率

89.6 %

技术优势

计数更准

采用掩膜碰撞线机制，计数结果与人工计数高度一致，平均计数准确率达96.3%。

模型更小

引入GhostConv和C2f模块，显著减少冗余信息，模型体积比原YOLOv7减小18.88MB，参数减少0.57M。

收敛更快

采用Focal EIoU损失函数替代CIoU，收敛速度显著提升，增强模型学习能力。

应用场景

• 果园产量估计：用无人机自动计数西瓜，替代人工估算产量。
• 农田作物监测：实时监测农田中西瓜数量与分布，指导田间管理。
• 无人机农业巡检：机载轻量化模型实现航拍视频实时计数，提升巡检效率。
• 边缘智能设备：大幅压缩模型体积和计算量，可部署于边缘设备离线运行。