IET Image Processing (Jul 2023)

Research on identification and classification of grassland forage based on deep learning and attention mechanisms

  • Yilei Liu,
  • Jiangping Liu,
  • Xuanhe Zhao,
  • Xin Pan,
  • Weihong Yan

DOI
https://doi.org/10.1049/ipr2.12808
Journal volume & issue
Vol. 17, no. 9
pp. 2628 – 2639

Abstract

Read online

Abstract Grassland is an important resource for China's economic development and the main economic source of animal husbandry. The identification and classification of grassland forage is an important part of the improvement of forage varieties and the monitoring of germplasm resources, which can fundamentally solve the problems of poor forage quality and low reproduction rate. For the problem of low accuracy of forage identification and classification, the authors put forward a new 3DSECNN model to remove the preprocessing operation and directly study the images. The authors took forage hyperspectral image (HSI) images on the field and built dataset, used 3DSECNN to train the images to improve the classification effect. The outstanding contributions of this paper are: (1) The authors took high‐precision forage HSI images in the field, established a dedicated database of forage HSIs, and expanded the datasets; (2) the process of integrating preprocessing ideas into the network and replacing the traditional method of preprocessing the data and then extracting features; (3) proposing the 3DSECNN model, which adds SENet on the basis of the traditional 3DCNN, strengthens the correlation of the spatial dimension, selects the key features for the classification by calculating the channel weight, inhibits the unimportant information, and achieves the purpose of integrating the preprocessing ideas into the network. The experimental results show the overall accuracy (OA) of 3DSECNN is 94.36%, Precision, Recall, F1‐score, Kappa, and Time also showed good levels. The experimental results prove that the 3DSECNN strengthens the correlation between image channels, enhances the performance ability of features, and provides a new method for the identification and classification research of forage.

Keywords