买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:中电海康集团有限公司
摘要:本发明属于小目标检测领域,公开了一种基于高斯过程回归的小目标检测方法及装置,方法包括获取训练数据集;加载YOLO系列模型来样本图像输出检测框和检测框的中心坐标点;根据小目标检测模型的输出以及标签计算第一损失;将检测框的中心坐标点作为输入,使用高斯过程回归进行曲线拟合,得到回归函数;使用目标轮廓检测方法提取样本图像中的目标轮廓点,得到目标轮廓曲线方程;在目标轮廓曲线上随机选取若干个测试点,根据回归函数和目标轮廓曲线方程的输出计算第二损失;加权综合第一损失和第二损失得到总损失,根据总损失对小目标检测模型进行训练更新,直至达到训练结束条件。本发明旨在降低错检和漏检率,并提高检测的效率。
主权项:1.一种基于高斯过程回归的小目标检测方法,其特征在于,所述基于高斯过程回归的小目标检测方法,包括:获取训练数据集,所述训练数据集中包含若干样本图像,且样本图像标记有小目标及小目标位置点标签;加载YOLO系列模型作为小目标检测模型,将样本图像输入所述小目标检测模型,得到由所述小目标检测模型输出的样本图像中对应小目标的检测框和检测框的中心坐标点;根据所述小目标检测模型的输出以及标签计算第一损失;将所述小目标检测模型输出的检测框的中心坐标点作为输入,使用高斯过程回归进行曲线拟合,输出样本图像中小目标的回归函数;使用目标轮廓检测方法提取样本图像中的目标轮廓点,并基于目标轮廓点得到目标轮廓曲线方程;在目标轮廓曲线上随机选取若干个测试点,并将测试点分别输入所述回归函数和所述目标轮廓曲线方程,根据所述回归函数和所述目标轮廓曲线方程的输出计算第二损失;加权综合第一损失和第二损失得到总损失,根据总损失对所述小目标检测模型进行训练更新,直至达到训练结束条件;取训练结束的小目标检测模型完成小目标检测。
全文数据:
权利要求:
百度查询: 中电海康集团有限公司 基于高斯过程回归的小目标检测方法及装置
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。