一种基于改进YOLOv5s的轻量化番茄成熟度检测系统及方法
摘要:
本发明属于果蔬检测特定计算机模型的技术领域,更具体地,涉及一种基于改进YOLOv5s的轻量化番茄成熟度检测系统及方法。所述系统包括骨干网络、颈部结构和头部结构;所述骨干网络中包括Ghostconv模块、C3Ghost模块和SPPF模块,将原C3模块内部的Bott leneck模块替换为GhostBott leneck模块;所述颈部结构的PANet使用加权双向特征金字塔网络BiFPN进行代替;所述头部结构使用EIoU损失函数替换CIoU损失函数。本发明解决了现有技术中YOLOv5s模型的PANet结构在特征融合过程中未考虑到不同特征信息在融合过程中的不同贡献,且没有横向连接充分融合特征信息,导致多尺度特征的表达能力减弱,检测精度降低的问题。
0/0