计算机与现代化 ›› 2023, Vol. 0 ›› Issue (07): 93-98.doi: 10.3969/j.issn.1006-2475.2023.07.016

• 图像处理 • 上一篇    下一篇

基于改进SegFormer模型的棉田地表残膜图像分割方法

  

  1. (新疆农业大学计算机与信息工程学院,新疆 乌鲁木齐 830052)
  • 出版日期:2023-07-26 发布日期:2023-07-27
  • 作者简介:牛玉珩(1995—),男,山西晋城人,硕士研究生,研究方向:图像处理,E-mail: 1063093035@qq.com; 通信作者:李永可(1985—),男,河南许昌人,副教授,研究方向:智慧农业,E-mail: 553667423@qq.com。
  • 基金资助:
    新疆维吾尔自治区重大科技专项(2020A01002-4)

Image Segmentation Method of Residual Film on Cotton Field Surface Based on Improved SegFormer Model#br#

  1. (College of Computer and Information Engineering, Xinjiang Agricultural University, Urumqi 830052, China)
  • Online:2023-07-26 Published:2023-07-27

摘要: 为了解决棉花种植过程中残留地膜造成的严重污染问题,提出一种基于改进SegFormer模型的残膜目标快速识别分割的方法。以新疆维吾尔族自治区昌吉市棉田(坐标44°23′1″ N,87°30′23″ E)的收后地表残膜为研究对象,在雪后晴天中午时间段使用无人机采集图像共计1047幅并制作成数据集。对SegFormer模型增加更深的特征图层级,使其能获取更细微的特征以解决残膜形态多变和目标较小的问题。SegFormer原始模型平均交并比已达到83.00%,改进SegFormer模型较原始模型平均交并比提升0.42个百分点,骰子系数提升0.3个百分点,单幅检测时间为51.13 ms。实验结果表明,改进SegFormer模型基本能满足快速分割任务的要求,为棉田残膜污染情况的快速评估提供了理论基础。

关键词: 棉田残膜, 无人机, 神经网络, 语义分割, SegFormer, 特征图层级

Abstract: In order to solve the problem of serious pollution caused by residual plastic film during cotton planting, a fast recognition and segmentation method based on improved SegFormer model is proposed. Taking the collected surface residual film of cotton field in Changji City, Xinjiang Uygur Autonomous Region (coordinates 44 ° 23 ′ 1 ″ N, 87 ° 30 ′ 23 ″ E) as the research object, 1047 images are collected at noon on a sunny day after snow and made into a data set. Based on the SegFormer model, a deeper feature layer level is added to obtain more subtle features to solve the problem of the morphologic variation of the residual film and the smaller target. The average crossing and merging ratio of the original SegFormer model has reached 83.00%, the average crossing and merging ratio of the improved SegFormer model has increased by 0.42 percentage points compared with the original model, the die coefficient has increased by 0.3 percentage points and the single detection time is 51.13 ms. The experimental results show that the improved SegFormer model can basically meet the requirements of fast segmentation tasks, and provide a theoretical basis for rapid assessment of residual film pollution in cotton fields.

Key words: residual film in cotton fields, UAV, neural networks, semantic segmentation, SegFormer, feature map level

中图分类号: