-
公开(公告)号:CN115359295A
公开(公告)日:2022-11-18
申请号:CN202211012822.2
申请日:2022-08-23
Applicant: 华北电力大学(保定)
IPC: G06V10/764 , G06V10/774 , G06V10/762 , G06V10/82 , G06V20/17 , G06V20/60
Abstract: 本发明涉及一种解耦知识蒸馏金具目标检测方法及系统,方法包括:分别根据数据集标注框在第一模型和第二模型对标注数据集进行解耦,得到前景目标区域和背景区域,以确定金具的目标种类概率,并根据目标种类概率确定共有特征评分掩码;根据共有特征评分掩码和特征图分类得分图确定第一蒸馏损失函数;基于全局上下文模块,根据确定特征图构建第二蒸馏损失函数;根据各蒸馏损失函数确定总损失函数,并根据总损失函数对第二模型进行训练,以对待测金具图像进行目标检测。本发明利用不同种类金具的共有形状特征,采用解耦知识蒸馏方法,针对前景目标区域,将属于金具之间独特的共有特征用于目标分类的信息整合起来迁移到小模型,提升了金具检测精度。