农业机械学报2024,Vol.55Issue(11):154-159,319,7.DOI:10.6041/j.issn.1000-1298.2024.11.017
基于YOLO v8和CycleGAN的红掌植株表型参数自动提取方法
Automatic Extraction of Phenotypic Parameters from Anthurium andraeanum Linden Based on YOLO v8 and CycleGAN
摘要
Abstract
Phenotypic parameters of plants are quantitatively indicated,describing the morphology,structure,and physiological characteristics of plants,unveiling the growth patterns and relationships with environmental factors.Issues such as significant data errors,plant damage,high costs,and extensive data volume were exhibited by existing manual measurement and laser scanning-based methods for extracting plant phenotypic parameters.Therefore,an automatic extraction method for phenotypic parameters of Anthurium andraeanum Linden plants based on YOLO v8 and CycleGAN was proposed.The method included the follows:YOLO v8 was enhanced with the convolutional block attention module to improve the model's feature extraction capabilities for detecting and segmenting Anthurium andraeanum Linden leaves;the Grabcut algorithm was utilized to eliminate background features from segmented images,and the VGG model was employed for classification to distinguish intact and missing Anthurium andraeanum Linden leaves;the convolutional block attention module and feature pyramid network were introduced into the CycleGAN generator to enhance multi-scale feature extraction capabilities,incorporating the SmoothL1 loss function to enhance model stability and repair missing Anthurium andraeanum Linden leaves;a phenotypic parameters extraction algorithm(PPEA)was proposed to automatically extract leaf length,leaf width,and leaf area of Anthurium andraeanum Linden plants.The proposed methods were compared and analyzed by using a dataset of 650 self-collected images.Experimental results demonstrated the effectiveness of the proposed approach in automatically extracting phenotypic parameters of Anthurium andraeanum Linden plants.关键词
表型参数提取/红掌/目标检测/图像修复/YOLO v8/CycleGANKey words
phenotypic parameter extraction/Anthurium andraeanum Linden/target detection/image restoration/YOLO v8/CycleGAN分类
农业科技引用本文复制引用
卢鹏,孙天文,陈明,王振华,郑宗生..基于YOLO v8和CycleGAN的红掌植株表型参数自动提取方法[J].农业机械学报,2024,55(11):154-159,319,7.基金项目
上海市科技创新计划项目(20dz1203800) (20dz1203800)