用掩码引导生成逼真烟雾图像,提升林火检测效果
MFGDiffusion: Mask-Guided Smoke Synthesis for Enhanced Forest Fire Detection
- 结合分割与多模态模型生成烟雾掩码和描述,指导图像生成
- 提出新损失函数,使烟雾边缘与背景更一致,提升合成质量
- 用大语言模型筛选高质量烟雾图像,适合林火检测数据增强
烟雾是野火最早可见的征兆。随着深度学习发展,基于图像的烟雾检测已成为预防森林火灾的关键手段。然而,真实林火烟雾图像数据稀缺,严重制约检测性能。图像生成模型可有效合成逼真烟雾图像,但现有修复模型在烟雾与背景一致性方面表现不佳。为此,本文提出一个综合框架生成林火烟雾图像:首先利用预训练分割模型和多模态模型获取烟雾掩码与图像描述;为解决修复模型对掩码和掩蔽图像特征利用不足的问题,设计了基于掩码与掩蔽图像特征的网络结构;并提出一种新的损失函数——掩码随机差值损失(mask random difference loss),通过随机扩展和侵蚀掩码边缘,增强生成结果在掩码区域周围的连贯性;最后,为构建可用于后续检测任务的烟雾图像数据集,采用随机掩码生成,并结合烟雾特征与多模态大语言模型作为过滤工具,筛选多样且合理的烟雾图像,显著提升合成数据质量。实验表明,生成的烟雾图像具有高度真实性和多样性,能有效提升林火烟雾检测模型性能。代码已公开于 https://github.com/wghr123/MFGDiffusion。
原文摘要 · Abstract (English)
Smoke is the first visible indicator of a wildfire.With the advancement of deep learning, image-based smoke detection has become a crucial method for detecting and preventing forest fires. However, the scarcity of smoke image data from forest fires is one of the significant factors hindering the detection of forest fire smoke. Image generation models offer a promising solution for synthesizing realistic smoke images. However, current inpainting models exhibit limitations in generating high-quality smoke representations, particularly manifesting as inconsistencies between synthesized smoke and background contexts. To solve these problems, we proposed a comprehensive framework for generating forest fire smoke images. Firstly, we employed the pre-trained segmentation model and the multimodal model to obtain smoke masks and image captions.Then, to address the insufficient utilization of masks and masked images by inpainting models, we introduced a network architecture guided by mask and masked image features. We also proposed a new loss function, the mask random difference loss, which enhances the consistency of the generated effects around the mask by randomly expanding and eroding the mask edges.Finally, to generate a smoke image dataset using random masks for subsequent detection tasks, we incorporated smoke characteristics and use a multimodal large language model as a filtering tool to select diverse and reasonable smoke images, thereby improving the quality of the synthetic dataset. Experiments showed that our generated smoke images are realistic and diverse, and effectively enhance the performance of forest fire smoke detection models. Code is available at https://github.com/wghr123/MFGDiffusion.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。