基于YOLO架构的目标检测方法及装置

    公开(公告)号:CN119251471A

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

    申请号:CN202411341307.8

    申请日:2024-09-25

    Abstract: 本申请公开了一种基于YOLO架构的目标检测方法及装置,其中,基于初始学生模型对有标签图像和无标签图像进行目标检测,得到有标签图像的学生检测结果和无标签图像的学生检测结果;基于教师模型对无标签图像进行分类,得到无标签图像的伪标签;基于伪标签分配器对伪标签进行评分;利用第一有监督损失函数确定第一有监督损失;利用无监督损失函数确定无监督损失;基于第一有监督损失和无监督损失确定总损失;基于总损失迭代训练初始学生模型和教师模型,至总损失小于第二预设损失值,将此时的初始学生模型确定为目标学生模型;将待识别图像输入目标学生模型,得到待识别图像的识别结果。本申请可以提高模型的训练效率和目标检测的检测准确率。

Patent Agency Ranking