[发明专利]基于泊松融合的烟雾数据增强方法、增强设备和存储介质

权利要求书

1.一种基于泊松融合的烟雾数据增强方法，其特征在于，包括：

获取图像数据集；

获取待增强图片；

判断所述待增强图片中的目标烟囱是否为无烟状态，生成第一烟雾判断结果；

若所述第一烟雾判断结果为是，根据所述图像数据集，确定烟雾图像数据；

通过泊松融合方法，将所述烟雾图像数据与所述待增强图片进行泊松融合，生成增强图片。

2.根据权利要求1所述的基于泊松融合的烟雾数据增强方法，其特征在于，所述判断所述待增强图片中的目标烟囱是否为无烟状态，生成第一烟雾判断结果，具体包括：

获取所述待增强图片对应的图像标签文件；

通过所述图像标签文件，确定所述待增强图片中的图片烟囱数据及图片烟雾数据；

根据所述图片烟囱数据及所述图片烟雾数据，判断所述待增强图片中的所述目标烟囱是否为所述无烟状态，确定所述第一烟雾判断结果。

3.根据权利要求2所述的基于泊松融合的烟雾数据增强方法，其特征在于，所述若所述第一烟雾判断结果为是，根据所述图像数据集，确定烟雾图像数据，具体包括：

对所述图像数据集进行随机抽取，确定所述烟雾图像数据。

4.根据权利要求2所述的基于泊松融合的烟雾数据增强方法，其特征在于，所述根据所述图片烟囱数据及所述图片烟雾数据，判断所述待增强图片中的所述目标烟囱是否为所述无烟状态，确定所述第一烟雾判断结果，具体包括：

获取所述图片烟雾数据的烟雾最高位置和所述图片烟囱数据的烟囱最低位置，以及最大允许距离；

根据所述烟囱最低位置、所述烟雾最高位置，以及所述最大允许距离，确定第一烟雾判断结果。

5.根据权利要求2所述的基于泊松融合的烟雾数据增强方法，其特征在于，所述根据所述图片烟囱数据及所述图片烟雾数据，判断所述待增强图片中的所述目标烟囱是否为所述无烟状态，确定所述第一烟雾判断结果，还包括：

获取所述图片烟囱数据的烟囱最左位置和烟囱最右位置；

获取所述图片烟雾数据的烟雾最右位置；

根据所述烟囱最左位置和所述烟囱最右位置，以及所述烟雾最右位置，确定第一烟雾判断结果。

6.根据权利要求2所述的基于泊松融合的烟雾数据增强方法，其特征在于，所述根据所述图片烟囱数据及所述图片烟雾数据，判断所述待增强图片中的所述目标烟囱是否为无烟状态，确定所述第一烟雾判断结果，还包括：

获取所述图片烟囱数据的烟囱最左位置和烟囱最右位置；

获取所述图片烟雾数据的烟雾最左位置；

根据所述烟囱最左位置和所述烟囱最右位置，以及所述烟雾最左位置，确定第一烟雾判断结果。

7.根据权利要求1所述的基于泊松融合的烟雾数据增强方法，其特征在于，所述判断所述待增强图片中的目标烟囱是否为无烟状态，生成第一烟雾判断结果之前，还包括：

根据图片烟囱数据，判断所述待增强图片中的所述目标烟囱的数量是否小于第一烟囱数量，生成烟囱数量判断结果；

若所述烟囱数量判断结果为是，则对所述图像数据集进行随机抽取，确定烟囱图像数据；

将所述烟囱图像数据与所述待增强图片进行融合，对所述待增强图片增加所述目标烟囱。

8.根据权利要求1所述的基于泊松融合的烟雾数据增强方法，其特征在于，所述通过泊松融合方法，将所述烟雾图像数据与所述待增强图片进行泊松融合，生成增强图片，具体包括：

获取所述烟雾图像数据的烟雾类别和图像参数；

根据所述烟雾类别、所述图像参数和待增强图片确定融合位置中心；

根据所述待增强图片和所述融合位置中心，生成所述增强图片。