Journal of Integrative Agriculture ›› 2020, Vol. 19 ›› Issue (8): 1998-2008.DOI: 10.1016/S2095-3119(19)62803-0

所属专题: 麦类耕作栽培合辑Triticeae Crops Physiology · Biochemistry · Cultivation · Tillage

• 论文 • 上一篇    下一篇

  

  • 收稿日期:2019-04-24 出版日期:2020-08-01 发布日期:2020-06-21

Detection and enumeration of wheat grains based on a deep learning method under various scenarios and scales

WU Wei, YANG Tian-le, LI Rui, CHEN Chen, LIU Tao, ZHOU Kai, SUN Cheng-ming, LI Chun-yan, ZHU Xin-kai, GUO Wen-shan
  

  1. Jiangsu Key Laboratory of Crop Genetics and Physiology/Co-Innovation Center for Modern Production Technology of Grain Crops, Yangzhou University, Yangzhou 225009, P.R.China   
  • Received:2019-04-24 Online:2020-08-01 Published:2020-06-21
  • Contact: Correspondence SUN Cheng-ming, Tel: +86-514-879793381, E-mail: cmsun@yzu.edu.cn; GUO Wen-shan, Tel: +86-514-87979339, E-mail: guows@yzu.edu.cn
  • About author: WU Wei, E-mail: 435208450@qq.com;
  • Supported by:
    This research was mainly supported by the National Key Research and Development Program of China (2017YFD0301205), the Postgraduate Research & Practice Innovation Program of Jiangsu Province, China (KYCX18_2371), the National Natural Science Foundation of China (31701355 and 31671615), the China Postdoctoral Science Foundation, China (2016M600448), the Priority Academic Program Development of Jiangsu Higher Education Institutions, China (PAPD), the Yangzhou Science Foundation for Excellent Youths, China (YZ2017098) and the Science and Technology Plan Projects of Yangzhou, China (YZ2016251).

Abstract:

Grain number is crucial for analysis of yield components and assessment of effects of cultivation measures.  The grain number per spike and thousand-grain weight can be measured by counting grains manually, but it is time-consuming, tedious and error-prone.  Previous image processing algorithms cannot work well with different backgrounds and different sizes.  This study used deep learning methods to resolve the limitations of traditional image processing algorithms.  Wheat grain image datasets were collected in the scenarios of three varieties, six background and two image acquisition devices with different heights, angles and grain numbers, 1 748 images in total.  All images were processed through color space conversion, image flipping and rotation.  The grain was manually annotated, and the datasets were divided into training set, validation set and test set.  We used the TensorFlow framework to construct the Faster Region-based Convolutional Neural Network Model.  Using the transfer learning method, we optimized the wheat grain detection and enumeration model.  The total loss of the model was less than 0.5 and the mean average precision was 0.91.  Compared with previous grain counting algorithms, the grain counting error rate of this model was less than 3% and the running time was less than 2 s.  The model can be effectively applied under a variety of backgrounds, image sizes, grain sizes, shooting angles, and shooting heights, as well as different levels of grain crowding.  It constitutes an effective detection and enumeration tool for wheat grain.  This study provides a reference for further grain testing and enumeration applications.
 

Key words: wheat grain ,  deep learning , Faster R-CNN ,  object detection ,  counting