油气站压缩机气体泄漏图像生成方法、装置、设备及介质与流程

本发明属于图像处理，特别是涉及一种油气站压缩机气体泄漏图像生成方法、装置、设备及介质。

背景技术：

1、近年来，图像识别模型已经在各个领域的真实场景中得到了应用，而作为图像识别模型训练中非常重要的一个因素，图像数据的质量会对模型训练效果的造成非常大的影响。

2、在能源领域中，为得到精度较高的图像识别模型，需要大量的高质量训练数据，在现有技术中，一种方法是故意破坏压缩机，获取压缩机气体泄漏的图像作为图像识别模型的训练数据用于训练的图像数据，另一种方法是通过人为的修图操作，在已有的正常压缩机图像添加气体泄漏的图像特征作为图像识别模型的训练数据，上述现有技术普遍存在获取困难，或是与现实使用场景关联度不高的问题。

3、所以，目前亟需一种压缩机气体泄漏图像数据的获取方法，既可以保证图像数据更加贴近现实，也可以按照同样的标准批量生成足够数据的高质量图像训练数据，同时节约大量的人力资源和时间成本。

技术实现思路

1、有鉴于此，本发明提供一种油气站压缩机气体泄漏图像生成方法、装置、设备及介质，旨在解决如何快速高效的生成高质量的压缩机气体泄漏图像数据的问题。

2、依据本发明的第一方面，提供了一种油气站压缩机气体泄漏图像生成方法，包括：

3、获取待处理图像，所述待处理图像包含特定背景特征，所述特定背景特征为压缩机区域图像；

4、依据特定背景特征，将所述待处理图像划分为蒙版区域与非蒙版区域，所述蒙版区域为所述待处理图像中不包含所述特定背景特征的图像区域，所述非蒙版区域为所述待处理图像中包含所述特定背景特征的图像区域；

5、基于预设字母形式，随机生成目标字母图像，所述预设字母形式至少包括目标字母、字母字体和字母字体大小；

6、创建目标尺寸的空白图像，基于所述目标尺寸以及所述目标字母图像，生成所述目标字母图像的起点位置，并按照起点位置将所述目标字母图像绘制在所述空白图像上，作为蒙版图像；

7、将所述待处理图像、所述蒙版图像和提示词输入到图像生成模型，以将所述蒙版图像作为图像绘制的基础图素，基于所述提示词对所述蒙版区域进行局部重绘，生成目标图像，所述目标图像为在所述蒙版区域包含气体图像的待处理图像。

8、可选地，所述获取待处理图像，包括：

9、获取原始图像，所述原始图像包含至少一个所述特定背景特征；

10、以所述目标尺寸为目标，调整所述原始图像的尺寸，得到所述待处理图像。

11、可选地，所述基于预设字母形式，随机生成目标字母图像，包括：

12、获取字母形式集合，所述字母形式集合至少包括字母集合、字母字体集合以及字体大小区间；

13、基于所述字母集合，随机选取属于所述字母集合的任意一个字母，作为所述目标字母；

14、基于所述字母字体集合以及所述字体大小区间，随机生成所述目标字母对应的字母字体以及字母字体大小，得到所述预设字母形式；

15、按照所述预设字母形式对所述目标字母进行调整，生成所述目标字母图像。

16、可选地，所述基于所述目标尺寸以及所述目标字母图像，生成所述目标字母图像的起点位置，并按照所述起点位置将所述目标字母图像绘制在所述空白图像上，作为蒙版图像，包括：

17、对所述目标尺寸和所述字母字体大小做差值计算，得到所述目标字母图像的起点位置的最大坐标值；

18、获取所述目标字母图像的起点位置范围，所述起点位置范围的坐标值为坐标原点到所述起点位置的最大坐标值；

19、基于所述起点位置范围，随机生成所述目标字母图像的绘制起点，作为所述目标字母图像的起点位置；

20、将所述目标字母图像绘制到所述空白图像上，得到所述蒙版图像，所述目标字母图像起点位置的坐标与所述目标字母图像的顶点或中心点的坐标相同。

21、可选地，所述提示词包括正面提示词和负面提示词，所述将所述待处理图像、所述蒙版图像和所述提示词输入到图像生成模型，以将所述蒙版图像作为图像绘制的基础图素，基于所述提示词对所述蒙版区域进行局部重绘，生成目标图像，包括：

22、分别按照所述正面提示词和所述负面提示词对所述图像生成模型进行参数设置，所述正面提示词用于指导所述图像生成模型生成需要的图像特征，所述负面提示词用于防止所述图像生成模型生成不需要的图像特征；

23、将所述蒙版图像作为图像重采样的基础图素，基于参数设置后的图像生成模型对所述待处理图像的蒙版区域进行局部图像重采样，得到所述待处理图像的蒙版区域重采样图像；

24、将所述待处理图像的蒙版区域重采样图像和所述待处理图像输入到所述参数设置后的图像生成模型，对所述蒙版区域重采样图像进行局部重绘，得到所述目标图像。

25、可选地，所述将所述蒙版图像作为图像重采样的基础图素，基于参数设置后的图像生成模型对所述待处理图像的蒙版区域进行局部图像重采样，得到所述待处理图像的蒙版区域重采样图像，包括：

26、获取噪声图素，并基于所述待处理图像的蒙版区域的位置范围，随机生成所述噪声图素的绘制起点；

27、将所述噪声图素绘制到所述待处理图像的蒙版区域上，得到所述待处理图像的蒙版区域随机噪声图像；

28、依据所述蒙版图像、所述正面提示词和所述负面提示词，分别生成需要去掉的噪声图素和需要增加的目标图素；

29、基于所述需要去掉的噪声元素和所述需要增加的目标图素，对所述待处理图像的蒙版区域进行采样，减少所述待处理图像蒙版区域中需要去掉的噪声图素，增加所述待处理图像蒙版区域中需要增加的目标图素，控制所述蒙版图像在所述蒙版区域中的稳定扩散，并保持所述待处理图像的非蒙版区域的图像保持不变，得到重采样迭代图像；

30、依据所述蒙版图像、所述正面提示词和所述负面提示词，对所述待处理图像的蒙版区域进行预设次数的重采样，将最后一次重采样得到的所述重采样迭代图像作为所述待处理图像的蒙版区域重采样图像。

31、可选地，在所述预设次数重采样的过程中，逐渐减少所述待处理图像蒙版区域中需要去掉的噪声图素，逐渐增加所述待处理图像蒙版区域中需要增加的目标图素，并保持所述待处理图像的非蒙版区域保持不变。

32、依据本发明的第二方面，提供了一种油气站压缩机气体泄漏图像生成装置，该装置可以包括：

33、待处理图像获取模块，用于获取待处理图像，所述待处理图像包含特定背景特征；

34、待处理图像区域划分模块，用于依据特定背景特征，将所述待处理图像划分为蒙版区域与非蒙版区域，所述蒙版区域为所述待处理图像中不包含所述特定背景特征的图像区域，所述非蒙版区域为所述待处理图像中包含所述特定背景特征的图像区域；

35、字母图像生成模块，用于基于预设字母形式，随机生成目标字母图像，所述预设字母形式至少包括目标字母、字母字体和字母字体大小；

36、蒙版图像生成模块，用于创建目标尺寸的空白图像，基于所述目标尺寸以及所述目标字母图像，生成所述目标字母图像的起点位置，并按照起点位置将所述目标字母图像绘制在所述空白图像上，作为蒙版图像；

37、目标图像生成模块，用于将所述待处理图像、所述蒙版图像和提示词输入到图像生成模型，以将所述蒙版图像作为图像绘制的基础图素，基于所述提示词对所述蒙版区域进行局部重绘，生成目标图像，所述目标图像为在所述蒙版区域包含气体图像的待处理图像。

38、可选地，所述待处理图像获取模块，包括：

39、原始图像获取子模块，用于获取原始图像，所述原始图像包含至少一个所述特定背景特征；

40、尺寸调整子模块，用于以所述目标尺寸为目标，调整所述原始图像的尺寸，得到所述待处理图像；

41、可选地，所述字母图像生成模块，包括：

42、字母形式集合获取子模块，用于获取字母形式集合，所述字母形式集合至少包括字母集合、字母字体集合以及字体大小区间；

43、目标字母选取子模块，用于基于所述字母集合，随机选取属于所述字母集合的任意一个字母，作为所述目标字母；

44、字母形式生成子模块，用于基于所述字母字体集合以及所述字体大小区间，随机生成所述目标字母对应的字母字体以及字母字体大小，得到所述预设字母形式；

45、目标字母调整子模块，用于按照所述预设字母形式对所述目标字母进行调整，生成所述目标字母图像。

46、可选地，所述蒙版图像生成模块，包括：

47、差值计算子模块，用于对所述目标尺寸和所述字母字体大小做差值计算，得到所述目标字母图像的起点位置的最大坐标值；

48、起点范围获取子模块，用于获取所述目标字母图像的起点位置范围，所述起点位置范围的坐标值为坐标原点到所述起点位置的最大坐标值；

49、绘制起点生成子模块，用于基于所述起点位置范围，随机生成所述目标字母图像的绘制起点，作为所述目标字母图像的起点位置；

50、绘制子模块，用于将所述目标字母图像绘制到所述空白图像上，得到所述蒙版图像，所述目标字母图像起点位置的坐标与所述目标字母图像的顶点或中心点的坐标相同。

51、可选地，目标图像生成模块，包括：

52、参数设置子模块，用于分别按照所述正面提示词和所述负面提示词对所述图像生成模型进行参数设置，所述正面提示词用于指导所述图像生成模型生成需要的图像特征，所述负面提示词用于防止所述图像生成模型生成不需要的图像特征；

53、重采样子模块，用于将所述蒙版图像作为图像重采样的基础图素，基于参数设置后的图像生成模型对所述待处理图像的蒙版区域进行局部图像重采样，得到所述待处理图像的蒙版区域重采样图像；

54、局部重绘子模块，用于将所述待处理图像的蒙版区域重采样图像和所述待处理图像输入到所述参数设置后的图像生成模型，对所述蒙版区域重采样图像进行局部重绘，得到所述目标图像。

55、噪声图素获取子模块，用于获取噪声图素，并基于所述待处理图像的蒙版区域的位置范围，随机生成所述噪声图素的绘制起点；

56、噪声图像生成子模块，用于将所述噪声图素绘制到所述待处理图像的蒙版区域上，得到所述待处理图像的蒙版区域随机噪声图像；

57、图素调整子模块，用于依据所述蒙版图像、所述正面提示词和所述负面提示词，分别生成需要去掉的噪声图素和需要增加的目标图素；

58、重采样迭代子模块，用于基于所述需要去掉的噪声元素和所述需要增加的目标图素，对所述待处理图像的蒙版区域进行采样，减少所述待处理图像蒙版区域中需要去掉的噪声图素，增加所述待处理图像蒙版区域中需要增加的目标图素，控制所述蒙版图像在所述蒙版区域中的稳定扩散，并保持所述待处理图像的非蒙版区域的图像保持不变，得到重采样迭代图像；

59、重采样输出子模块，用于依据所述蒙版图像、所述正面提示词和所述负面提示词，对所述待处理图像的蒙版区域进行预设次数的重采样，将最后一次重采样得到的所述重采样迭代图像作为所述待处理图像的蒙版区域重采样图像。

60、重采样控制子模块，用于在所述预设次数重采样的过程中，逐渐减少所述待处理图像蒙版区域中需要去掉的噪声图素，逐渐增加所述待处理图像蒙版区域中需要增加的目标图素，并保持所述待处理图像的非蒙版区域保持不变。

61、第三方面，本发明实施例提供了一种电子设备，所述电子设备包括：

62、存储器，用于存储程序指令；

63、处理器，用于调用所述存储器中存储的程序指令，按照获得的程序指令执行如第一方面所述的油气站压缩机气体泄漏图像生成方法中包括的步骤。

64、第四方面，本发明实施例提供了一种计算机存储介质，其特征在于，所述计算机存储介质上存储程序指令，所述程序指令被处理器执行时实现如第一方面所述的油气站压缩机气体泄漏图像生成方法中包括的步骤。

65、本发明提供了一种油气站压缩机气体泄漏图像生成方法、装置、设备及介质，包括：获取待处理图像，基于预设字母形式，随机生成目标字母图像，按照所述起点位置将所述目标字母图像绘制在所述空白图像上，作为蒙版图像，将蒙版图像作为图像绘制的基础图素，基于提示词对蒙版区域进行局部重绘，生成目标图像。本发明具有如下有益效果：

66、(1)以随机生成目标字母图像作为蒙版图像，对待处理图像的蒙版区域进行局部重绘，从而可以确保生成的图像像素之间具有自然的关联，没有违和感。

67、(2)在对蒙版区域进行局部重绘的过程中加入了提示词的指导，控制图像生成过程中的图素的表现效果，从而确保生成的图像具备与实际场景相符的特征，可以准确反映出气体泄漏的真实情况，进而为能源领域的压缩机气体泄漏图像数据模拟生成提供了一种高效可行的解决方案。

68、上述说明仅是本发明技术方案的概述，为了能够更清楚了解本发明的技术手段，而可依照说明书的内容予以实施，并且为了让本发明的上述和其它目的、特征和优点能够更明显易懂，以下特举本发明的具体实施方式。

技术特征：

1.一种油气站压缩机气体泄漏图像生成方法，其特征在于，所述方法包括：

2.根据权利要求1所述的油气站压缩机气体泄漏图像生成方法，其特征在于，所述基于目标尺寸，对所述原始图像的特定背景特征进行调整，得到目标尺寸的待处理图像，包括：

3.根据权利要求1所述的油气站压缩机气体泄漏图像生成方法，其特征在于，所述基于预设字母形式，随机生成目标字母图像，包括：

4.根据权利要求1所述的油气站压缩机气体泄漏图像生成方法，其特征在于，所述提示词包括正面提示词和负面提示词，所述将所述待处理图像、所述蒙版图像和所述提示词输入到图像生成模型，以将所述蒙版图像作为图像绘制的基础图素，基于所述提示词对所述蒙版区域进行局部重绘，生成目标图像，包括：

5.根据权利要求4所述的油气站压缩机气体泄漏图像生成方法，其特征在于，所述将所述蒙版图像作为图像重采样的基础图素，基于参数设置后的图像生成模型对所述待处理图像的蒙版区域进行局部图像重采样，得到所述待处理图像的蒙版区域重采样图像，包括：

6.根据权利要求5所述的油气站压缩机气体泄漏图像生成方法，其特征在于，在所述预设次数重采样的过程中，逐渐减少所述待处理图像蒙版区域中需要去掉的噪声图素，逐渐增加所述待处理图像蒙版区域中需要增加的目标图素，并保持所述待处理图像的非蒙版区域保持不变。

7.一种油气站压缩机气体泄漏图像生成装置，其特征在于，所述装置包括：

8.一种电子设备，其特征在于，所述电子设备包括：

9.一种计算机存储介质，其特征在于，所述计算机存储介质上存储程序指令，所述程序指令被处理器执行时实现如权利要求1至6中任一所述的油气站压缩机气体泄漏图像生成方法包括的步骤。

技术总结
本申请提供了一种油气站压缩机气体泄漏图像生成方法、装置、设备及介质，涉及图像处理技术领域，包括：获取待处理图像；基于预设字母形式，随机生成目标字母图像；创建目标尺寸的空白图像，基于目标尺寸以及目标字母图像，生成目标字母图像的起点位置，并按照起点位置将目标字母图像绘制在空白图像上，作为蒙版图像；将待处理图像、蒙版图像和提示词输入到图像生成模型，以将蒙版图像作为图像绘制的基础图素，基于提示词对蒙版区域进行局部重绘，生成目标图像。本发明以随机生成目标字母图像作为蒙版图像，对待处理图像的蒙版区域进行局部重绘，从而确保生成的图像像素之间具有自然的关联，进而稳定高效地获得压缩机气体泄漏图像数据。

技术研发人员：陈友明,陈思竹,姜超
受保护的技术使用者：四川弘和数智集团有限公司
技术研发日：
技术公布日：2024/9/23