文献类型：期刊文章

作　　者：王立舒[1] 秦铭霞[1] 雷洁雅[1] 王小飞[1] 谭克竹[1]

Wang Lishu;Qin Mingxia;Lei Jieya;Wang Xiaofei;Tan Kezhu(Institute of Electrical and Information,Northeast Agricultural University,Harbin 150030,China)

机构地区：[1]东北农业大学电气与信息学院,哈尔滨150030

出　　处：《农业工程学报》

基　　金：黑龙江省教育厅科技课题(12521038);黑龙江省自然科学基金联合引导项目(LH2020C003)。

年　　份：2021

卷　　号：37

期　　号：18

起止页码：170-178

语　　种：中文

收录情况：AJ、BDHX、BDHX2020、CAB、CAS、CSCD、CSCD2021_2022、EAPJ、EI、IC、JST、PROQUEST、RCCSE、SCOPUS、ZGKJHX、核心刊

摘　　要：为实现自然环境下蓝莓果实成熟度的精确快速识别,该研究对YOLOv4-Tiny网络结构进行改进,提出一种含有注意力模块的目标检测网络(I-YOLOv4-Tiny)。该检测网络采用CSPDarknet53-Tiny网络模型作为主干网络,将卷积注意力模块(Convolution Block Attention Module,CBAM)加入到YOLOv4-Tiny网络结构的特征金字塔(Feature Pyramid Network,FPN)中,通过对每个通道的特征进行权重分配来学习不同通道间特征的相关性,加强网络结构深层信息的传递,从而降低复杂背景对目标识别的干扰,且该检测网络的网络层数较少,占用内存低,以此提升蓝莓果实检测的精度与速度。对该研究识别方法进行性能评估与对比试验的结果表明,经过训练的I-YOLOv4-Tiny目标检测网络在验证集下的平均精度达到97.30%,能有效地利用自然环境中的彩色图像识别蓝莓果实并检测果实成熟度。对比YOLOv4-Tiny、YOLOv4、SSD-MobileNet、Faster R-CNN目标检测网络,该研究在遮挡与光照不均等复杂场景中,平均精度能达到96.24%。平均检测时间为5.723 ms,可以同时满足蓝莓果实识别精度与速度的需求。I-YOLOv4-Tiny网络结构占用内存仅为24.20 M,为采摘机器人与早期产量预估提供快速精准的目标识别指导。

关 键 词：机器视觉 图像识别 目标检测网络  深度学习  蓝莓 卷积注意力块  

分 类 号：TP391.4]