基于PSA-YOLO网络的苹果叶片病斑检测
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家重点研发计划项目(2020YFD1100601)


Apple Leaf Lesion Detection Based on PSA-YOLO Network
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为提高YOLOv4目标检测算法对苹果叶片小型病斑的检测性能,提出了一种PSA(金字塔压缩注意力)-YOLO算法。在CSPDarknet53的基础上融合了Focus结构和PSA机制,并采用网络深度减小策略,构建了参数量小、精确度高的PSA-CSPDarknet-1轻量化主干网络。其次在网络颈部,搭建了空间金字塔卷积池化模块,用极小的计算代价增强了对深层特征图的空间信息提取能力,并采用α-CIoU损失函数作为边界框损失函数,提高网络对高IoU阈值下目标的检测精度。根据实验结果,PSA-YOLO网络在苹果叶片病斑识别任务中的AP50达到88.2%。COCO AP@[0.5∶0.05∶0.95]达到49.8%,比YOLOv4提升3.5个百分点。网络对于小型病斑的特征提取能力提升幅度更大,小型病斑检测AP比YOLOv4提升3.9个百分点。在单张NVIDIA GTX TITAN V显卡上的实时检测速度达到69帧/s,相较于YOLOv4网络提升13帧/s。

    Abstract:

    In order to improve the detection performance of YOLOv4 object detection algorithm for small apple leaf lesions, a PSA-YOLO network with low computational cost and high accuracy was proposed, which integrated a Focus layer and the pyramid squeeze attention block in the CSPDarknet, and the strategy of network depth reduction was adopted. Finally, the PSA-CSPDarknet-1 was built on the basis of CSPDarknet53. The experimental results showed that the computational complexity of PSA-CSPDarknet-1 was reduced by 30.4% compared with the CSPDarknet53 and the detection accuracy of the network for small lesions (covering area less than 32 pixels×32 pixels) was improved by 2.9 percentage points. In the neck, a spatial pyramid convolution and pooling module was built to enhance multi-scale information extraction in spatial dimensions with a small computational cost, and α-CIoU loss function for the bounding box was used to improve the detection accuracy of bounding boxes for improving the detection accuracy of lesions under the high IoU threshold. According to the experimental results, the proposed PSA-YOLO network achieved 88.2% AP50 and it achieved 49.8% COCO AP@[0.5∶0.05∶0.95] in the apple leaf lesion dataset, which was 3.5 percentage points higher than that of YOLOv4. At the same time, the feature extraction ability of the network for small lesions was more improved, and APS was 3.9 percentage points higher than that of YOLOv4, respectively. The detection speed on a single NVIDIA GTX TITAN V reached 69 frames per second, which was 13 frames per second faster than that of YOLOv4.

    参考文献
    相似文献
    引证文献
引用本文

晁晓菲,池敬柯,张继伟,王孟杰,陈尧,刘斌.基于PSA-YOLO网络的苹果叶片病斑检测[J].农业机械学报,2022,53(8):329-336. CHAO Xiaofei, CHI Jingke, ZHANG Jiwei, WANG Mengjie, CHEN Yao, LIU Bin. Apple Leaf Lesion Detection Based on PSA-YOLO Network[J]. Transactions of the Chinese Society for Agricultural Machinery,2022,53(8):329-336.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2022-05-13
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2022-06-07
  • 出版日期: