Affiliation:
1. College of Mechanical and Electronic Engineering, Inner Mongolia Agricultural University, Hohhot / China
Abstract
In order to realize the automatic driving of agricultural machinery in greenhouse, this paper uses image acquisition equipment to collect road images in greenhouse and makes data sets, builds SETR model based on Transformer framework and DeepLabv3+ model based on convolution neural network for semantic segmentation of road images in greenhouse, and verifies the semantic segmentation ability of the two models to road images in greenhouse. Several groups of training periods are set as observation points to observe the semantic segmentation effect of the two models on the greenhouse road image, and the test set which has not been trained by the model is used as the prediction object to verify the performance of the two models on the semantic segmentation of greenhouse road image. The SETR model reached 94.64% PA on the greenhouse road data set, and 82.72% mIoU, DeepLabv3+ model reached 90.80% PA and 72.35% mIoU on the greenhouse road data set. Both models have excellent performance in semantic segmentation of greenhouse road images, and the performance of SETR model is slightly better than that of DeepLabv3+ model. The semantic segmentation performance of the two models for greenhouse road images can meet the needs of actual deployment.