一种轻量化目标检测方法、装置、计算机程序产品制造方法及图纸

技术编号:41217586 阅读:27 留言:0更新日期:2024-05-09 23:38
本发明专利技术公开了一种轻量化目标检测方法、装置、计算机程序产品,包括:对图像数据进行增强处理,得到增强后的图像数据;利用教师网络和学生网络分别对所述增强后的图像数据进行特征提取,得到教师特征图和学生特征图;利用区域候选网络分别处理教师特征图和学生特征图,通过特征蒸馏针对所述教师特征图和学生特征图进行焦点蒸馏和全局蒸馏,得到学生特征图对应的候选框;根据所述学生特征图对应的候选框,利用分类卷积网络和定位卷积网络生成目标检测结果,其中,所述目标检测结果包括目标分类概率和目标预测框定位信息。本发明专利技术能够实现高精度、高效的轻量化目标检测需求,提高嵌入式设备下目标检测的精确性和泛用性。

【技术实现步骤摘要】

本专利技术涉及一种轻量化目标检测方法、装置、计算机程序产品,属于目标检测。


技术介绍

1、目标检测是计算机视觉中的一个基础任务,图像分割、物体追踪等都要依赖目标检测。随着基于深度学习的目标检测技术的出现,卷积神经网络的应用使得目标检测的效果大大提升。为了获得更好的性能,通常使用更大的主干,这需要更多的计算资源并且推理速度更慢。当目标检测技术运用到无人机航拍技术、医学影像病灶检测等领域时,往往需要将目标检测模型部署在算力小、电池容量小的嵌入式设备中,这时候传统的目标检测方法因其高耗电量、高算力要求、较慢推理速度等而不再适用,为了克服这些缺点,在有限的推理时间内实现强大的目标检测性能,需要对目标检测模型进行轻量化,以便适用目标检测模型部署到嵌入式设备的应用场景。

2、知识蒸馏是一种将大型教师网络中的信息集成到一个紧凑的学生网络中,并在推理过程中不增加额外成本而获得强大性能的方法。主要分为逻辑蒸馏和特征蒸馏,前者是直接匹配网络输出的概率分布,后者是直接匹配中间的特征或者学习特征之间的转换关系。知识蒸馏技术能够完成目标检测的轻量化工作,但是大多数蒸馏方本文档来自技高网...

【技术保护点】

1.一种轻量化目标检测方法,其特征在于,包括如下步骤:

2.根据权利要求1所述的轻量化目标检测方法,其特征在于,所述增强处理包括:

3.根据权利要求1所述的轻量化目标检测方法,其特征在于,所述教师网络的骨干网络采用101层残差网络,所述学生网络的骨干网络采用50层残差网络。

4.根据权利要求1所述的轻量化目标检测方法,其特征在于,对所述教师特征图和学生特征图进行焦点蒸馏,包括:

5.根据权利要求4所述的轻量化目标检测方法,其特征在于,针对特征图F,其对应的注意力特征图的计算公式如下:

6.根据权利要求1所述的轻量化目标检测方法,...

【技术特征摘要】

1.一种轻量化目标检测方法,其特征在于,包括如下步骤:

2.根据权利要求1所述的轻量化目标检测方法,其特征在于,所述增强处理包括:

3.根据权利要求1所述的轻量化目标检测方法,其特征在于,所述教师网络的骨干网络采用101层残差网络,所述学生网络的骨干网络采用50层残差网络。

4.根据权利要求1所述的轻量化目标检测方法,其特征在于,对所述教师特征图和学生特征图进行焦点蒸馏,包括:

5.根据权利要求4所述的轻量化目标检测方法,其特征在于,针对特征图f,其对应的注意力特征图的计算公式如下:

6.根据权利要求1所述的轻量化目标检测方法,其特征在于,...

【专利技术属性】
技术研发人员:徐小龙徐国庆张用明
申请(专利权)人:南京邮电大学
类型:发明
国别省市:

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

1