目标检测模型的生成方法、装置、设备以及存储介质与流程

本公开涉及人工智能，尤其涉及计算机视觉、深度学习、大模型等，具体涉及目标检测模型的生成方法、装置、设备以及存储介质。

背景技术：

1、目标检测是计算机视觉中的一项基础任务，是很多计算机视觉任务的前提。少数类别的图像含有大量的样本(定义为头部类别)，而大多数类别的图像只包含少量甚至极少的样本(定义为尾部类别)。由于存在样本分布不平衡的问题，导致目标检测模型在头部类别的图像识别上准确率较高，而在尾部类别的图像识别上准确率较低。

技术实现思路

1、本公开提供了一种目标检测模型的生成方法、装置、设备以及存储介质。

2、根据本公开的一方面，提供了一种目标检测模型的生成方法，包括：

3、基于经选择的初始大模型进行数据生产，得到用于长尾目标检测场景的样本数据；

4、基于样本数据对初始大模型进行训练，得到可用于长尾目标检测场景的目标检测大模型；

5、基于目标检测大模型得到可用于长尾目标检测场景的目标检测小模型。

6、根据本公开的另一方面，提供了一种目标检测模型的生成装置，包括：

7、数据生产模块，用于基于经选择的初始大模型进行数据生产，得到用于长尾目标检测场景的样本数据；

8、模型训练模块，用于基于样本数据对初始大模型进行训练，得到可用于长尾目标检测场景的目标检测大模型；

9、模型压缩模块，用于基于目标检测大模型得到可用于长尾目标检测场景的目标检测小模型。

10、根据本公开的另一方面，提供了一种电子设备，包括：

11、至少一个处理器；以及

12、与该至少一个处理器通信连接的存储器；其中，

13、该存储器存储有可被该至少一个处理器执行的指令，该指令被该至少一个处理器执行，以使该至少一个处理器能够执行本公开实施例中任一的方法。

14、根据本公开的另一方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，其中，该计算机指令用于使该计算机执行根据本公开实施例中任一的方法。

15、根据本公开的另一方面，提供了一种计算机程序产品，包括计算机程序，该计算机程序在被处理器执行时实现根据本公开实施例中任一的方法。

16、根据本公开的方案，至少能够提高用于长尾目标检测场景的目标检测小模型的生成速度和识别准确率。

17、应当理解，本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征，也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。

技术特征：

1.一种目标检测模型的生成方法，包括：

2.根据权利要求1所述的方法，还包括：

3.根据权利要求1或2所述的方法，还包括：

4.根据权利要求1所述的方法，其中，所述基于经选择的初始大模型进行数据生产，得到用于长尾目标检测场景的样本数据，包括以下至少之一：

5.根据权利要求1所述的方法，其中，所述基于所述目标检测大模型得到可用于长尾目标检测场景的目标检测小模型，包括：

6.根据权利要求1所述的方法，其中，所述基于所述目标检测大模型得到可用于长尾目标检测场景的目标检测小模型，包括：

7.根据权利要求6所述的方法，还包括：

8.根据权利要求6所述的方法，还包括：

9.根据权利要求1所述的方法，还包括：

10.根据权利要求1所述的方法，其中，数据生产的处理方式，包括：

11.一种目标检测模型的生成装置，包括：

12.根据权利要求11所述的装置，还包括：

13.根据权利要求11或12所述的装置，还包括：

14.根据权利要求11所述的装置，其中，所述数据生产模块，用于执行以下至少之一：

15.根据权利要求11所述的装置，其中，所述模型压缩模块，用于：

16.根据权利要求11所述的装置，其中，所述模型压缩模块，用于：

17.根据权利要求16所述的装置，其中，所述模型压缩模块，还用于：

18.根据权利要求16所述的装置，其中，所述模型压缩模块，还用于：

19.根据权利要求11所述的装置，还包括：

20.根据权利要求11所述的装置，其中，所述数据生产模块，还用于：

21.一种电子设备，包括：

22.一种存储有计算机指令的非瞬时计算机可读存储介质，其中，计算机指令用于使计算机执行根据权利要求1-10中任一项的方法。

23.一种计算机程序产品，包括存储在存储介质上的计算机程序，计算机程序在被处理器执行时实现根据权利要求1-10中任一项的方法。

技术总结本公开提供了目标检测模型的生成方法、装置、设备以及存储介质，涉及人工智能技术领域，尤其涉及计算机视觉、深度学习、大模型等技术领域。具体实现方案为：基于经选择的初始大模型进行数据生产，得到用于长尾目标检测场景的样本数据；基于样本数据对初始大模型进行训练，得到可用于长尾目标检测场景的目标检测大模型；基于目标检测大模型得到可用于长尾目标检测场景的目标检测小模型。技术研发人员：希滕,张刚,姚锟,丁二锐受保护的技术使用者：北京百度网讯科技有限公司技术研发日：技术公布日：2024/11/14