A vision transformer-based robotic perception for early tea chrysanthemum flower counting in field environments

文献类型： 外文期刊

第一作者： Qi, Chao

作者： Qi, Chao;Qi, Chao;Chen, Kunjie;Gao, Junfeng;Gao, Junfeng

作者机构：

关键词： agricultural robotics; density map estimation; high density object counting; visual transformer

期刊名称：JOURNAL OF FIELD ROBOTICS （ 影响因子：4.2； 五年影响因子：7.2 ）

ISSN： 1556-4959

年卷期： 2024 年

页码：

收录情况： SCI

摘要： The current mainstream approaches for plant organ counting are based on convolutional neural networks (CNNs), which have a solid local feature extraction capability. However, CNNs inherently have difficulties for robust global feature extraction due to limited receptive fields. Visual transformer (ViT) provides a new opportunity to complement CNNs' capability, and it can easily model global context. In this context, we propose a deep learning network based on a convolution-free ViT backbone (tea chrysanthemum-visual transformer [TC-ViT]) to achieve the accurate and real-time counting of TCs at their early flowering stage under unstructured environments. First, all cropped fixed-size original image patches are linearly projected into a one-dimensional vector sequence and fed into a progressive multiscale ViT backbone to capture multiple scaled feature sequences. Subsequently, the obtained feature sequences are reshaped into two-dimensional image features and using a multiscale perceptual field module as a regression head to detect the overall scale and density variance. The resulting model was tested on 400 field images in the collected TC test data set, showing that the proposed TC-ViT achieved the mean absolute error and mean square error of 12.32 and 15.06, with the inference speed of 27.36 FPS (512 x 512 image size) under the NVIDIA Tesla V100 GPU environment. It is also shown that light variation had the greatest effect on TC counting, whereas blurring had the least effect. This proposed method enables accurate counting for high-density and occlusion objects in field environments and this perception system could be deployed in a robotic platform for selective harvesting and flower phenotyping.

分类号：

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

微信小程序