目标检测方法、装置、电子设备及存储介质制造方法及图纸

技术编号:40055794 阅读:40 留言:0更新日期:2024-01-16 21:55
本申请涉及人工智能技术领域,提供了一种目标检测方法、装置、电子设备及存储介质。该方法通过将学生网络提取得到的第一特征图划分为N个学生网络候选区域,并在教师网络提取得到的第二特征图中确定与N个学生网络候选区域一一对应的N个教师网络候选区域,将每组对应的学生网络候选区域和教师网络候选区域中的特征进行对齐处理,从而解决了学生网络与教师网络的特征图差异较大导致的,教师网络无法较好指导学生网络学习的技术问题,降低了学生网络的学习难度,提升了知识迁移效果,进而使得使用该方法训练好的学生网络对待检测图像进行目标检测时,能够获得较高的检测精度。

【技术实现步骤摘要】

本申请涉及人工智能,尤其涉及一种目标检测方法、装置、电子设备及存储介质


技术介绍

1、基于目标检测的知识迁移算法可以将大模型知识迁移至小模型中,应用小模型时,从而在提升了推理速度的同时,保证了目标检测精度。

2、现有基于特征的目标检测知识迁移算法存在一个问题,即通过整体的特征图进行损失计算,因为学生模型的表达能力有限,所以两者的特征图存在较大差异,所以这种优化方式不能很好的指导学生模型学习。


技术实现思路

1、有鉴于此,本申请实施例提供了一种目标检测方法、装置、电子设备及存储介质,以解决现有技术中基于知识迁移算法进行目标检测时,学生网络学习难度大、迁移效果不佳的问题。

2、本申请实施例的第一方面,提供了一种目标检测方法,包括:

3、获取待检测图像;

4、将待检测图像分别输入学生网络和训练好的教师网络,学生网络对待检测图像进行特征提取得到第一特征图,教师网络对待检测图像进行特征提取得到第二特征图;

5、基于第一特征图确定n个候选区域,n为正整数;...

【技术保护点】

1.一种目标检测方法,其特征在于,包括:

2.根据权利要求1所述的方法,其特征在于,所述基于所述第一特征图确定N个候选区域,包括:

3.根据权利要求2所述的方法,其特征在于,所述在第一特征图中确定学生网络候选区域特征,包括:

4.根据权利要求1所述的方法,其特征在于,所述基于各学生网络候选区域特征以及对应的教师网络候选区域特征计算得到所述学生网络的对齐损失函数,包括:

5.根据权利要求4所述的方法,其特征在于,所述基于所述各候选区域的学生网络候选区域提取后的特征以及与其分别对应的教师网络候选区域特征计算得到所述学生网络的对齐损失函数,包括:...

【技术特征摘要】

1.一种目标检测方法,其特征在于,包括:

2.根据权利要求1所述的方法,其特征在于,所述基于所述第一特征图确定n个候选区域,包括:

3.根据权利要求2所述的方法,其特征在于,所述在第一特征图中确定学生网络候选区域特征,包括:

4.根据权利要求1所述的方法,其特征在于,所述基于各学生网络候选区域特征以及对应的教师网络候选区域特征计算得到所述学生网络的对齐损失函数,包括:

5.根据权利要求4所述的方法,其特征在于,所述基于所述各候选区域的学生网络候选区域提取后的特征以及与其分别对应的教师网络候选区域特征计算得到所述学生网络的对齐损失函数,包括:

6.根据权利要求4所述的方法,其特征在于,所述学生网络对所...

【专利技术属性】
技术研发人员:金毅勐蒋召周靖宇
申请(专利权)人:深圳须弥云图空间科技有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1