物体检测模型处理方法、装置、设备及存储介质与流程

文档序号:29575307发布日期:2022-04-09 06:06阅读:来源:国知局

技术特征:
1.一种物体检测模型处理方法,其特征在于,应用于边缘设备,包括:获取待检测数据,并将所述待检测数据输入初始物体检测模型,得到模型检测结果,其中待检测数据为视频;基于视频流对所述模型检测结果进行难例挖掘,得到难例数据;将得到的所述难例数据上传至服务器,以使所述服务器在接收到的所述难例数据的数量达到预设标准时,根据所述难例数据和基础数据创建数据集,其中所述基础数据为所述服务器预先获取的数据,并采用所述数据集进行模型训练,得到训练好的物体检测模型;接收所述服务器发送的所述训练好的物体检测模型。2.根据权利要求1所述的方法,其特征在于,所述基于视频流对所述模型检测结果进行难例挖掘,得到难例数据,包括:基于视频流提取所述模型检测结果的视频中各帧的检测框;对所述模型检测结果中各帧进行目标追踪,以对所述各帧的检测框添加标识,得到追踪结果;对所述追踪结果进行难例挖掘,得到所述难例数据。3.根据权利要求2所述的方法,其特征在于,所述对所述追踪结果进行难例挖掘,得到所述难例数据,包括:若检测到所述追踪结果中任一帧的前后第一预设数量的帧均含有相同标识对应的检测框,且所述任一帧不含有所述相同标识对应的检测框,则将所述任一帧确定为待确定漏检难例帧;将所述待确定漏检难例帧前后第二预设数量帧中的所有的检测框求并集,得到待映射检测框;将所述待映射检测框映射至所述待确定漏检难例帧,得到所述待确定漏检难例帧的漏检难例候选区;提取所述漏检难例候选区内图像的第一图像特征点;提取所述待确定漏检难例帧前后第三预设数量帧中各帧检测框内图像的图像特征点;将所述第一图像特征点与所述待确定漏检难例帧前后第三预设数量帧中所有帧检测框内图像的图像特征点做匹配,确定多个第一匹配项;若所述多个第一匹配项的数量均大于第四预设数量或大于第一预设比例,则根据所述第一图像特征点确定第一外接矩形,其中所述第一外接矩形是所述第一图像特征点的外接矩形;将所述第一外接矩形对应的图像内容和所述第一外接矩形的属性确定为漏检难例,其中所述外接矩形的属性为外接矩形内物体的类别。4.根据权利要求2所述的方法,其特征在于,所述对所述追踪结果进行难例挖掘,得到所述难例数据,包括:若检测到所述追踪结果中任一帧含有任一标识对应的检测框,且所述任一帧的前后第五预设数量的帧均不含有所述任一标识对应的检测框,则将所述任一帧确定为待确定误检难例帧;将所述待确定误检难例帧中的所述检测框扩大预设倍,得到误检难例候选区;将所述误检难例候选区映射至所述待确定误检难例帧的前后第六预设数量帧中,得到
多个对比区;提取所述误检难例候选区内图像的第二图像特征点;提取所述待确定误检难例帧的前后第六预设数量帧中各帧的所述多个对比区内图像的图像特征点;将所述第二图像特征点与所述待确定误检难例帧的第六预设数量帧中每帧的图像特征点做匹配,得到多个第二匹配项;若所述多个第二匹配项的数量均大于第七预设数量或大于第二预设比例,则根据所述第二图像特征点确定第二外接矩形,其中所述第二外接矩形是所述第二图像特征点的外接矩形;将所述第二外接矩形对应的图像内容和所述第二外接矩形的属性确定为误检难例,其中所述外接矩形的属性为外接矩形内物体的类别。5.一种物体检测模型处理方法,其特征在于,应用于服务器,包括:接收边缘设备发送的难例数据;当接收到的所述难例数据的数量达到预设标准时,根据所述难例数据和基础数据创建数据集,其中所述基础数据为所述服务器预先获取的数据;采用所述数据集进行模型训练,得到训练好的物体检测模型;将所述训练好的物体检测模型发送至所述边缘设备。6.根据权利要求5所述的方法,其特征在于,所述当接收到的所述难例数据的数量达到预设标准时,根据所述难例数据和基础数据创建数据集,包括:当接收到的所述难例数据的数据量达到预设值,或难例数据的数量与所述基础数据的数量的比例达到第三预设比例时,将所述难例数据和所述基础数据混合后划分为训练集和测试集,其中所述训练集和所述测试集组成所述数据集。7.根据权利要求6所述的方法,其特征在于,所述采用所述数据集进行模型训练,得到训练好的物体检测模型,包括:基于所述训练集,采用迁移学习和/或早停策略对初始物体检测模型进行训练,得到待验证的物体检测模型;基于所述测试集,计算所述待验证的物体检测模型的性能指标,若所述性能指标达到预设标准则将所述待验证的物体检测模型确定为训练好的模型。8.根据权利要求5至7任一项所述的方法,其特征在于,所述采用所述数据集进行模型训练,得到训练好的物体检测模型之后,还包括:采用网络剪枝和/或量化技术将所述训练好的物体检测模型简化,得到简化后的物体检测模型;相应地,所述将所述训练好的物体检测模型发送至所述边缘设备,还包括:将所述简化后的物体检测模型发送至所述边缘设备。9.一种物体检测模型处理装置,其特征在于,包括:检测结果获得模块,用于获取待检测数据,并将所述待检测数据输入初始物体检测模型,得到模型检测结果,其中待检测数据为视频;难例数据获得模块,用于基于视频流对所述模型检测结果进行难例挖掘,得到难例数据;
难例数据上传模块,用于将得到的所述难例数据上传至服务器,以使所述服务器在接收到的所述难例数据的数量达到预设标准时,根据所述难例数据和基础数据创建数据集,其中所述基础数据为所述服务器预先获取的数据,并采用所述数据集进行模型训练,得到训练好的物体检测模型;检测模型接收模块,用于接收所述服务器发送的所述训练好的物体检测模型。10.一种物体检测模型处理装置,其特征在于,包括:难例数据接收模块,用于接收边缘设备发送的难例数据;数据集创建模块,用于当接收到的所述难例数据的数量达到预设标准时,根据所述难例数据和基础数据创建数据集,其中所述基础数据为所述服务器预先获取的数据;检测模型获取模块,用于采用所述数据集进行模型训练,得到训练好的物体检测模型;检测模型发送模块,用于将所述训练好的物体检测模型发送至所述边缘设备。11.一种电子设备,其特征在于,包括:处理器,以及与所述处理器通信连接的存储器;所述存储器存储计算机执行指令;所述处理器执行所述存储器存储的计算机执行指令,以实现如权利要求1至4中任一项所述的物体检测模型处理方法。12.一种电子设备,其特征在于,包括:处理器,以及与所述处理器通信连接的存储器;所述存储器存储计算机执行指令;所述处理器执行所述存储器存储的计算机执行指令,以实现如权利要求5至8中任一项所述的物体检测模型处理方法。13.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质中存储有计算机执行指令,所述计算机执行指令被处理器执行时用于实现如权利要求1至4中任一项所述的物体检测模型处理方法。14.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质中存储有计算机执行指令,所述计算机执行指令被处理器执行时用于实现如权利要求5至8中任一项所述的物体检测模型处理方法。15.一种计算机程序产品,其特征在于,包括计算机程序,该计算机程序被处理器执行时实现权利要求1至4中任一项所述的物体检测模型处理方法。16.一种计算机程序产品,其特征在于,包括计算机程序,该计算机程序被处理器执行时实现权利要求5至8中任一项所述的物体检测模型处理方法。

技术总结
本申请提供一种物体检测模型处理方法、装置、设备及存储介质,涉及数据处理领域。该方法包括:获取待检测数据,并将所述待检测数据输入初始物体检测模型,得到模型检测结果,其中待检测数据为视频;基于视频流对所述模型检测结果进行难例挖掘,得到难例数据;将得到的所述难例数据上传至服务器,以使所述服务器在接收到的所述难例数据的数量达到预设标准时,根据所述难例数据和基础数据创建数据集,其中所述基础数据为所述服务器预先获取的数据,并采用所述数据集进行模型训练,得到训练好的物体检测模型;接收所述服务器发送的所述训练好的物体检测模型。本申请的方法,增加了边缘设备在检测物体时的准确性。在检测物体时的准确性。在检测物体时的准确性。


技术研发人员:王晔 杨凯 丁平
受保护的技术使用者:中国建设银行股份有限公司
技术研发日:2021.12.29
技术公布日:2022/4/8
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1