An Improved UNet Lightweight Network for Semantic Segmentation of Weed Images in Corn Fields  

作　　者：Yu Zuo Wenwen Li 

机构地区：[1]School of Information and Control Engineering,Jilin Institute of Chemical Technology,Jilin,132022,China [2]School of Mechanical and Control Engineering,Baicheng Normal University,Baicheng,137000,China

出　　处：《Computers, Materials & Continua》2024年第6期4413-4431,共19页计算机、材料和连续体（英文）

摘　　要：In cornfields,factors such as the similarity between corn seedlings and weeds and the blurring of plant edge details pose challenges to corn and weed segmentation.In addition,remote areas such as farmland are usually constrained by limited computational resources and limited collected data.Therefore,it becomes necessary to lighten the model to better adapt to complex cornfield scene,and make full use of the limited data information.In this paper,we propose an improved image segmentation algorithm based on unet.Firstly,the inverted residual structure is introduced into the contraction path to reduce the number of parameters in the training process and improve the feature extraction ability;secondly,the pyramid pooling module is introduced to enhance the network’s ability of acquiring contextual information as well as the ability of dealing with the small target loss problem;and lastly,Finally,to further enhance the segmentation capability of the model,the squeeze and excitation mechanism is introduced in the expansion path.We used images of corn seedlings collected in the field and publicly available corn weed datasets to evaluate the improved model.The improved model has a total parameter of 3.79 M and miou can achieve 87.9%.The fps on a single 3050 ti video card is about 58.9.The experimental results show that the network proposed in this paper can quickly segment corn weeds in a cornfield scenario with good segmentation accuracy.

关 键 词：Semantic segmentation deep learning UNet pyramid pooling module 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]