高级检索

      基于图像特征与机器学习的烤烟烟叶产量预测方法

      An Image Features Based Machine Learning Model for Flue-cured Tobacco Yield Prediction

      • 摘要: 为探讨基于RGB图像特征结合机器学习与深度学习算法预测烤烟产量的可行性,通过无人机获取烤烟中川208打顶后25 d的图像,提取颜色、纹理与形状共35个特征。采用随机森林算法优选特征,利用反向传播神经网络(BPNN)、基于遗传算法的反向传播神经网络(GA-BPNN)、极限学习机(ELM)、基于粒子群算法的极限学习机(PSO-ELM)、支持向量机(SVR)、基于遗传算法的支持向量机(GA-SVR)和随机森林(RF)7种传统机器学习算法和深度学习算法一维卷积神经网络(1D-CNN)构建产量预测模型。结果表明:基于RF优选组合特征(颜色、形状与纹理)所建立的RF预测模型准确度(R2=0.970)与泛化能力(R2=0.817)均高于其他6种机器学习模型与1D-CNN模型,产量预测值与实测值吻合性较高。特征优选结合随机森林算法构建烤烟产量预测模型可行性较高,可为烤烟产量预测提供新思路。

         

        Abstract: To explore the feasibility of predicting flue-cured tobacco yield based on RGB images in combination with machine learning and deep learning algorithms, a field experiment was carried out using flue-cured tobacco Zhongchuan 208. RGB images of flue-cured tobacco were obtained 25 days after topping using drones. Color, texture and shape of the images were extracted, totaling 35 features. Features selection were performed using the random forest algorithm, and a yield prediction model was constructed using seven machine learning algorithms (BPNN, GA-BPNN, ELM, PSO-ELM, SVR, GA-SVR, RF) and one deep learning algorithm (1D-CNN). The results showed as the follows: The accuracy (R2=0.970) and generalization ability (R2=0.817) of the random forest prediction model established by the combination of features (color, shape, and texture) obtained through the random forest algorithm are higher than those of the other six machine learning models and the convolutional neural network model. The predicted yield are in good accordance with the measured values. In summary, we constructed a tobacco yield prediction model through the combination of feature selection and the random forest algorithm and provided novel tools for tobacco yield prediction.

         

      /

      返回文章
      返回