Extraction of agricultural plastic greenhouses based on a U-Net Convolutional Neural Network Coupled with edge expansion and loss function improvement

Abstract

ABSTRACT Agricultural plastic greenhouses (APGs) are crucial for sustainable agricultural planting, and accurate spatial distribution information acquisition is crucial. Deep learning network models can extract target features from remote sensing images more effectively than traditional interpretation methods, which face challenges like high workloads and poor repeatability. In this study, we aim to enhance the inventorying of Agricultural Plastic Greenhouses (APGs) by improving the extraction accuracy of their locations and numbers through remote sensing techniques. Utilizing GF-7 satellite imagery, we propose an enhanced U-Net convolutional neural network (CNN) model that incorporates edge information expansion and a joint loss function to optimize performance. The primary objective is to provide a rapid and accurate method for mapping APGs, which is crucial for effective agricultural management and environmental monitoring. The U-Net network’s accuracy was enhanced by 1.1% after expanding 3 × 3sample edge information, and further by 1.9% by combining edge extension and loss function constraints. Our results demonstrate that the modified U-Net model significantly improves extraction accuracy compared to traditional methods, thereby facilitating better inventory management and planning for agricultural cash crops. This advancement not only supports farmers in optimizing resources but also contributes to sustainable agricultural practices by enabling precise monitoring of APG distribution. Implication Statement Compared to traditional interpretation methods, which suffer problems such as heavy workloads, small adaptation ranges and poor repeatability, deep learning network models can better extract target features from remote sensing images. In this study, we used GF-7 image data to improve the traditional U-Net convolutional neural network (CNN) model. The Canny operator and Gaussian kernel (GK) function were used for sample edge expansion, and the binary cross-entropy and GK functions were used to jointly constrain the loss. Finally, APGs were accurately extracted and compared to those obtained with the original model. The results indicated that the APG extraction accuracy of the U-Net network was improved through the expansion of sample edge information and adoption of joint loss function constraints.