Liu Kaixuan, Wang Jie, Zhang Kai, Chen Minhui, Zhao Haonan, Liao Juan
College of Engineering, Anhui Agricultural University, Hefei 230036, China.
Anhui Provincial Rural Comprehensive Economic Information Center, Hefei 230031, China.
Sensors (Basel). 2023 Jul 27;23(15):6738. doi: 10.3390/s23156738.
The identification of the growth and development period of rice is of great significance to achieve high-yield and high-quality rice. However, the acquisition of rice growth period information mainly relies on manual observation, which has problems such as low efficiency and strong subjectivity. In order to solve these problems, a lightweight recognition method is proposed to automatically identify the growth period of rice: Small-YOLOv5, which is based on improved YOLOv5s. Firstly, the new backbone feature extraction network MobileNetV3 was used to replace the YOLOv5s backbone network to reduce the model size and the number of model parameters, thus improving the detection speed of the model. Secondly, in the feature fusion stage of YOLOv5s, we introduced a more lightweight convolution method, GsConv, to replace the standard convolution. The computational cost of GsConv is about 60-70% of the standard convolution, but its contribution to the model learning ability is no less than that of the standard convolution. Based on GsConv, we built a lightweight neck network to reduce the complexity of the network model while maintaining accuracy. To verify the performance of Small-YOLOv5s, we tested it on a self-built dataset of rice growth period. The results show that compared with YOLOv5s (5.0) on the self-built dataset, the number of the model parameter was reduced by 82.4%, GFLOPS decreased by 85.9%, and the volume reduced by 86.0%. The (0.5) value of the improved model was 98.7%, only 0.8% lower than that of the original YOLOv5s model. Compared with the mainstream lightweight model YOLOV5s- MobileNetV3-Small, the number of the model parameter was decreased by 10.0%, the volume reduced by 9.6%, and the (0.5:0.95) improved by 5.0%-reaching 94.7%-and the recall rate improved by 1.5%-reaching 98.9%. Based on experimental comparisons, the effectiveness and superiority of the model have been verified.
识别水稻的生长发育期对于实现水稻高产和优质具有重要意义。然而,水稻生育期信息的获取主要依靠人工观测,存在效率低、主观性强等问题。为了解决这些问题,提出了一种轻量级识别方法来自动识别水稻的生育期:基于改进的YOLOv5s的Small-YOLOv5。首先,使用新的骨干特征提取网络MobileNetV3替换YOLOv5s骨干网络,以减小模型大小和模型参数数量,从而提高模型的检测速度。其次,在YOLOv5s的特征融合阶段,引入了一种更轻量级的卷积方法GsConv来替代标准卷积。GsConv的计算成本约为标准卷积的60%-70%,但其对模型学习能力的贡献不低于标准卷积。基于GsConv,构建了一个轻量级颈部网络,在保持准确性的同时降低了网络模型的复杂度。为了验证Small-YOLOv5s的性能,我们在自建的水稻生育期数据集上对其进行了测试。结果表明,与自建数据集上的YOLOv5s(5.0)相比,模型参数数量减少了82.4%,GFLOPS下降了85.9%,体积减少了86.0%。改进模型的(0.5)值为98.7%,仅比原始YOLOv5s模型低0.8%。与主流轻量级模型YOLOV5s-MobileNetV3-Small相比,模型参数数量减少了10.0%,体积减少了9.6%,(0.5:0.95)提高了5.0%,达到94.7%,召回率提高了1.5%,达到98.9%。通过实验比较,验证了该模型的有效性和优越性。