[发明专利]一种训练样本的生成方法及装置

说明书

本申请提供一种训练样本的生成方法及装置，其中所述方法包括：建立不同类别的三维虚拟烟雾场景；针对每一类别的三维虚拟烟雾场景，将所述三维虚拟烟雾场景渲染为烟雾场景图像，并将该类别和该烟雾场景图像关联存储；将存储的各类别及其各类别对应的烟雾场景图像确定为训练样本，所述训练样本用于训练烟雾场景模型，所述烟雾场景模型用于确定输入的烟雾数据对应的烟雾场景类别，从而无需在现实中采集实际火灾的样本，极大地降低了数据采集的人力和物力成本，同时由于虚拟相机的高度可控性，无需依赖具体设备则可以实现任意拍摄角度的图像采集，提高了样本的丰富性，从而提升了模型的准确率。

技术领域

本申请涉及数据处理技术领域，尤其涉及一种训练样本的生成方法及装置。

背景技术

随着视频监控设备的大量普及、更高效图像处理相关的深度学习算法的出现和微处理器技术的发展，通过对图像信息的分析最终实现对火灾的检测逐渐成为一种新的有效技术。

在相关技术中，在火灾检测方面，大多侧重于机器学习算法上的研究，鲜少有人关注训练数据的采集。而在发生火灾时，训练数据又是十分难以采集的。

发明内容

有鉴于此，本申请提供一种训练样本的生成方法及装置。

具体地，本申请是通过如下技术方案实现的：

第一方面，本申请实施例提供了一种训练样本的生成方法，所述方法包括：

建立不同类别的三维虚拟烟雾场景；

针对每一类别的三维虚拟烟雾场景，将所述三维虚拟烟雾场景渲染为烟雾场景图像，并将该类别和该烟雾场景图像关联存储；

将存储的各类别及其各类别对应的烟雾场景图像确定为训练样本，所述训练样本用于训练烟雾场景模型，所述烟雾场景模型用于确定输入的烟雾数据对应的烟雾场景类别。

可选地，所述针对每一类别的三维虚拟烟雾场景，将所述三维虚拟烟雾场景渲染为烟雾场景图像的步骤包括：

针对每一类别的三维虚拟烟雾场景设置对应的虚拟相机，其中，所述虚拟相机具有对应的相机参数；

基于所述相机参数将所述三维虚拟烟雾场景渲染为烟雾场景图像。

可选地，所述相机参数包括相机内参以及相机外参；

所述相机内参为针对所述虚拟相机的内部设置的参数，不同类型的虚拟相机具有不同的相机内参；

所述相机外参为针对所述虚拟相机的外部姿态设置的参数。

可选地，所述相机内参采用四维参数矩阵表示，所述相机内参包括下述参数中的至少一个：y方向的视角、纵横比、近剪裁平面到原点的距离及远剪裁平面到原点的距离；

所述相机外参包括所述虚拟相机的位置信息以及朝向信息。

可选地，所述三维虚拟烟雾场景包括三维虚拟环境以及烟雾，所述建立不同类别的三维虚拟烟雾场景，包括：

获取不同类别的三维模型，并采用所述三维模型建立对应类别的三维虚拟环境；

针对各三维虚拟环境，设置该三维虚拟环境中的烟雾参数，并将所述烟雾参数输入至预设的虚拟烟雾生成组件，以通过所述虚拟烟雾生成组件为所述三维虚拟环境搭建对应的烟雾。

可选地，所述基于所述相机参数将所述三维虚拟烟雾场景渲染为烟雾场景图像，包括：

基于当前的相机参数，采用光线追踪算法将所述三维虚拟环境渲染成环境图像；

获取所述虚拟烟雾生成组件根据所述烟雾参数以及当前的相机参数输出的烟雾图像；

将所述烟雾图像叠加到所述环境图像的指定烟雾位置，生成烟雾场景图像。