一种基于半监督学习的火焰分割数据集构建方法及装置

本发明公开了一种基于半监督学习的火焰分割数据集构建方法及装置，方法包括：采集原始火焰燃烧图像；将原始火焰燃烧图像按照不同的照度和场景进行分类存储，获得分类图像数据集；对各个分类图像数据集中的部分原始火焰燃烧图像进行处理和标注形成多个标注图像对，获得火焰标注数据集；从火焰标注数据集中选取标注图像对进行无监督数据增强处理，形成训练集；根据训练集对火焰分割模型进行训练，选取未标注的原始火焰燃烧图像输入至火焰分割模型，获得分割结果图像；对分割结果图像进行优选，将优选后的分割结果图像加入至训练集中，直到训练集的规模满足预设值，获得火焰分割数据集；该方法可构建出大规模、场景数量均衡的火焰分割数据集。