高级检索+

松树株数识别的YOLOv5轻量化算法研究

Research on YOLOv5 Lightweight Algorithm for Pine Tree Strain Identification

  • 摘要: 随着5G技术的发展,其高带宽、低时延和高密度接入特点,促使云计算模式向“云-管-端”模式改变,边缘计算作为终端关键技术对人工智能算法在算力有限的终端上的部署成为关键。以苗圃验收环节中松树株数识别的视频检索算法为例,提出一种适用于人工智能算法在终端部署的轻量级苗圃松树苗检测计数算法。算法通过在YOLOv5网络的基础上引入MobileNet v3特征提取机制来实现网络的轻量化,将压缩激励网络(Squeeze-and-Excitation Networks, SENet)中的轻量级注意模块集成作为bneck基本块,提高网络对于特征通道的敏感程度,增强网络的特征提取能力;在IoU(Intersection over Union, IoU)基础上进一步考虑目标框和预测框的向量角度,使用SIoU损失函数作为预测函数,重新定义相关损失函数,从而使苗圃树苗预测框更加接近真实框。研究结果表明,改进后的模型参数量明显减少,改进后的网络模型大小与对比试验中的方法相比,模型在准确率(Precision)降低3.26%、平均精确率均值(Mean Average Precision,mAP)降低1.03%的情况下,帧率(Frame Per Second, FPS)提升了21.48%,达到71.43帧/s,计算量较原YOLOv5s减少了148.44%。证明该算法具有高效性和轻量性,为边缘计算终端人工智能算法移植提供算法原型。

     

    Abstract: With the development of 5G technology, its high bandwidth, low latency and high density access features have led to a change in the cloud computing model to a’cloud-management-end’ model, and edge computing as a key terminal technology has become critical to the deployment of AI algorithms on terminals with limited computing power. Taking the video retrieval algorithm for pine tree plant identification in nursery acceptance as an example, a lightweight algorithm for pine sapling detection and counting in nurseries suitable for terminal deployment of AI algorithms in proposed. The algorithm achieves network lightweighting by introducing the MobileNet v3 feature extraction mechanism on the basis of the YOLOv5 network, integrating the lightweight attention module in Squeeze-and-Excitation Networks(SENet) as a bneck basic block to improve the network’s sensitivity to feature channels and enhance the network’s feature extraction capability. The vector angles of the target and prediction frames are further considered on the IoU basis. The SIoU loss function is used as the prediction function and the associated loss function is redefined, thus making the nursery sapling prediction frame closer to the real frame. The results of the study show that the number of parameters of the improved model is significantly reduced, and the size of the improved network model is compared with the method in the comparison experiment, the model has a 21.48% improvement in frame rate(FPS) to reach to 71.43 frames per second with a 3.26% reduction in accuracy(Precision) and a 1.03% reduction in mean average precision(mAP), and the computational effort is reduced from the original YOLOv5s reduced 148.44%, proving that the algorithm is highly efficient and lightweight, providing an algorithm prototype for the porting of artificial intelligence algorithms to edge computing terminals.

     

/

返回文章
返回