arXiv:2501.15262cs.CVq-bio.QM2025-01被引 1

用AI精准计数茶花并判断开花阶段,助力茶叶育种研究

TflosYOLO+TFSC: An Accurate and Robust Model for Estimating Flower Count and Flowering Period

  • 基于YOLOv5改进的TflosYOLO模型,融合SE注意力机制提升检测精度
  • 在29份茶种上实现0.874 mAP50,花数预测相关系数达0.974
  • 可应对光照、修剪等复杂条件,适合茶树表型分析与育种应用

茶花在茶树分类与杂交育种中具有关键作用。传统观察方法耗时且不准确,本文提出TflosYOLO和TFSC模型,用于茶花数量估算与开花期识别。研究通过两年间采集29份茶种的花图像构建了高代表性和多样性数据集。在此基础上,基于YOLOv5架构并引入挤压-激励(SE)网络的TflosYOLO模型,首次实现了茶花检测与计数的可行方案,mAP50达到0.874,优于YOLOv5、YOLOv7和YOLOv8。该模型在涵盖26份茶种、五个开花阶段、不同光照条件及修剪/未修剪植株的34个数据集上均表现出强泛化能力,预测花数与实际值的相关系数(R²)为0.974。此外,设计了7层神经网络的TFSC模型,用于自动识别开花阶段,在两年数据上分别达到0.738和0.899的准确率。利用TflosYOLO+TFSC框架,可动态监测茶花发育过程,追踪不同茶种开花阶段变化,为茶树育种与种质资源表型分析提供重要支持。

原文摘要 · Abstract (English)

Tea flowers play a crucial role in taxonomic research and hybrid breeding for the tea plant. As traditional methods of observing tea flower traits are labor-intensive and inaccurate, we propose TflosYOLO and TFSC model for tea flowering quantifying, which enable estimation of flower count and flowering period. In this study, a highly representative and diverse dataset was constructed by collecting flower images from 29 tea accessions in 2 years. Based on this dataset, the TflosYOLO model was built on the YOLOv5 architecture and enhanced with the Squeeze-and-Excitation (SE) network, which is the first model to offer a viable solution for detecting and counting tea flowers. The TflosYOLO model achieved an mAP50 of 0.874, outperforming YOLOv5, YOLOv7 and YOLOv8. Furthermore, TflosYOLO model was tested on 34 datasets encompassing 26 tea accessions, five flowering stages, various lighting conditions, and pruned / unpruned plants, demonstrating high generalization and robustness. The correlation coefficient (R^2) between the predicted and actual flower counts was 0.974. Additionally, the TFSC (Tea Flowering Stage Classification) model, a 7-layer neural network was designed for automatic classification of the flowering period. TFSC model was evaluated on 2 years and achieved an accuracy of 0.738 and 0.899 respectively. Using the TflosYOLO+TFSC model, we monitored the tea flowering dynamics and tracked the changes in flowering stages across various tea accessions. The framework provides crucial support for tea plant breeding programs and phenotypic analysis of germplasm resources.

花期识别目标检测农业AI茶树育种

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。