基于数据剪枝的智能烟叶识别模型微调训练方法及系统

    公开(公告)号:CN119251616A

    公开(公告)日:2025-01-03

    申请号:CN202411310029.X

    申请日:2024-09-19

    Abstract: 本发明公开了基于数据剪枝的智能烟叶识别模型微调训练方法及系统,方法包括:收集待剪枝的视频数据集;预处理所收集的视频数据集;将预处理得到的烟叶图像训练集S输入烟叶识别基模型Wt进行预训练;利用预训练后的烟叶识别基模型Wt对烟叶图片训练集S中新的烟叶样本x进行剪枝,形成剪枝后的烟叶样本数据集#imgabs0#以用于后续的模型微调训练;将剪枝后的烟叶样本数据集#imgabs1#中的烟叶样本x’输入至预训练后的烟叶识别基模型Wt,对烟叶识别基模型Wt进行微调训练,得到烟叶测试模型Wt”;将预处理得到的烟叶图片测试集C输入至烟叶测试模型Wt”进行测试。本发明利用数据集剪枝法能够减小训练数据集的大小来提高训练和模型识别效率。

Patent Agency Ranking