一种1+2+1蒸馏模式的知识蒸馏猕猴桃视觉计数方法

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种1+2+1蒸馏模式的知识蒸馏猕猴桃视觉计数方法
申请号:CN202510336159
申请日期:2025-03-21
公开号:CN119887743B
公开日期:2025-05-27
类型:发明专利
摘要
本发明属于图像识别技术领域,具体涉及一种1+2+1蒸馏模式的知识蒸馏猕猴桃视觉计数方法,该方法以YOLOv11网络模型构建学生模型;将SEAM注意力模块作为YOLOv11网络模型的检测头,得到教师模型;在YOLOv11网络模型的颈部网络添加SBA自适应注意力模块和SOEP模块,两个助教模型;使用教师模型对两个助教模型进行蒸馏,得到两个蒸馏损失函数;用两个蒸馏损失函数之和对学生模型进行蒸馏训练,然后对学生模型进行自蒸馏操作,使用蒸馏训练后的学生模型对输入的猕猴桃图像进行检测并计数。本发明可解决教师模型和学生模型之间的差距过大的问题,提高模型的可靠性和精确性,实现高效的猕猴桃计数。
技术关键词
计数方法 蒸馏 注意力 教师 学生 预测特征 视觉 网络 生成知识 模块 标签 校准 图像识别技术 检测头 层级 模式 数据