Affiliations: [a] Electromechanic Engineering College, Northeast Forestry University, Harbin, P.R. China | [b] School of Artificial Intelligence, Wuxi Vocational College of Science and Technology, Wuxi, P.R. China
Correspondence:
[*]
Corresponding author. Kexin Li, School of artificial intelligence, Wuxi Vocational College of Science and Technology, Wuxi, P.R. China. E-mail: dillonlkx@163.com.
Abstract: In agricultural production, weed removal is an important part of crop cultivation, but inevitably, other plants compete with crops for nutrients. Only by identifying and removing weeds can the quality of the harvest be guaranteed. Therefore, the distinction between weeds and crops is particularly important. Recently, deep learning technology has also been applied to the field of botany, and achieved good results. Convolutional neural networks are widely used in deep learning because of their excellent classification effects. The purpose of this article is to find a new method of plant seedling classification. This method includes two stages: image segmentation and image classification. The first stage is to use the improved U-Net to segment the dataset, and the second stage is to use six classification networks to classify the seedlings of the segmented dataset. The dataset used for the experiment contained 12 different types of plants, namely, 3 crops and 9 weeds. The model was evaluated by the multi-class statistical analysis of accuracy, recall, precision, and F1-score. The results show that the two-stage classification method combining the improved U-Net segmentation network and the classification network was more conducive to the classification of plant seedlings, and the classification accuracy reaches 97.7%.
Keywords: Deep learning, plant seedlings classification, machine learning, U-Net
