-
公开(公告)号:CN119251471A
公开(公告)日:2025-01-03
申请号:CN202411341307.8
申请日:2024-09-25
Applicant: 北京信息科技大学
IPC: G06V10/25 , G06N3/0464 , G06N3/09 , G06N3/045
Abstract: 本申请公开了一种基于YOLO架构的目标检测方法及装置,其中,基于初始学生模型对有标签图像和无标签图像进行目标检测,得到有标签图像的学生检测结果和无标签图像的学生检测结果;基于教师模型对无标签图像进行分类,得到无标签图像的伪标签;基于伪标签分配器对伪标签进行评分;利用第一有监督损失函数确定第一有监督损失;利用无监督损失函数确定无监督损失;基于第一有监督损失和无监督损失确定总损失;基于总损失迭代训练初始学生模型和教师模型,至总损失小于第二预设损失值,将此时的初始学生模型确定为目标学生模型;将待识别图像输入目标学生模型,得到待识别图像的识别结果。本申请可以提高模型的训练效率和目标检测的检测准确率。