您好,欢迎访问安徽省农业科学院 机构知识库!

Rice Grain Detection and Counting Method Based on TCLE-YOLO Model

文献类型: 外文期刊

作者: Zou, Yu 1 ; Tian, Zefeng 2 ; Cao, Jiawen 2 ; Ren, Yi 3 ; Zhang, Yaping 4 ; Liu, Lu 4 ; Zhang, Peijiang 1 ; Ni, Jinlong 1 ;

作者机构: 1.Anhui Acad Agr Sci, Rice Res Inst, Hefei 230031, Peoples R China

2.Anhui Agr Univ, Coll Engn, Hefei 230036, Peoples R China

3.Anhui Sci & Technol Univ, Coll Agr, Chuzhou 239000, Peoples R China

4.Chinese Acad Sci, Hefei Inst Technol Innovat Engn, Hefei 230094, Peoples R China

关键词: rice grain detection and counting; YOLOv5; coordinate attention module; transform

期刊名称:SENSORS ( 影响因子:3.9; 五年影响因子:4.1 )

ISSN:

年卷期: 2023 年 23 卷 22 期

页码:

收录情况: SCI

摘要: Thousand-grain weight is the main parameter for accurately estimating rice yields, and it is an important indicator for variety breeding and cultivation management. The accurate detection and counting of rice grains is an important prerequisite for thousand-grain weight measurements. However, because rice grains are small targets with high overall similarity and different degrees of adhesion, there are still considerable challenges preventing the accurate detection and counting of rice grains during thousand-grain weight measurements. A deep learning model based on a transformer encoder and coordinate attention module was, therefore, designed for detecting and counting rice grains, and named TCLE-YOLO in which YOLOv5 was used as the backbone network. Specifically, to improve the feature representation of the model for small target regions, a coordinate attention (CA) module was introduced into the backbone module of YOLOv5. In addition, another detection head for small targets was designed based on a low-level, high-resolution feature map, and the transformer encoder was applied to the neck module to expand the receptive field of the network and enhance the extraction of key feature of detected targets. This enabled our additional detection head to be more sensitive to rice grains, especially heavily adhesive grains. Finally, EIoU loss was used to further improve accuracy. The experimental results show that, when applied to the self-built rice grain dataset, the precision, recall, and mAP@0.5 of the TCLE-YOLO model were 99.20%, 99.10%, and 99.20%, respectively. Compared with several state-of-the-art models, the proposed TCLE-YOLO model achieves better detection performance. In summary, the rice grain detection method built in this study is suitable for rice grain recognition and counting, and it can provide guidance for accurate thousand-grain weight measurements and the effective evaluation of rice breeding.

  • 相关文献
作者其他论文 更多>>