一种基于深度学习的木皮颜色自动分选方法

本发明的一种基于深度学习的木皮颜色自动分选方法，包括如下步骤：(1)采集不同颜色的木皮图像数据集，将处理后的数据集以8：2的比例划分，作为模型的训练集和测试集；(2)使用多种图像预处理方法进行图像预处理；(3)对深度学习算法进行模型选择分析，选择MobileViT模型作为木皮颜色分选的模型；(4)对选择的模型进行提升效率、精度、稳定性层面的优化。本发明结合图像预处理方法，通过深度学习算法对木皮颜色进行预测，探索非约束条件下模型对不同颜色木皮识别的有效性。采用本发明利于协助工人进行木皮颜色分选，提高分选效率和准确性，减轻工人的工作负担。