Generation of Smoke Dataset for Power Equipment and Study of Image Semantic Segmentation

Abstract

Fire in power equipment has always been one of the main hazards of power equipment. Smoke detection and recognition have always been extremely important in power equipment, as they can provide early warning before a fire breaks out. Compared to relying on smoke concentration for recognition, image-based smoke recognition has the advantage of being unaffected by indoor and outdoor environments. This paper addresses the problems of limited smoke data, difficult labeling, and insufficient research on recognition algorithms in power systems. We propose using three-dimensional virtual technology to generate smoke and image masks and using environmental backgrounds such as HDR (high dynamic range imaging) lighting to realistically combine smoke and background. In addition, to address the characteristics of smoke in power equipment, a dual UNet model named DS-UNet is proposed. The model consists of a deep and a shallow network structure, which can effectively segment the details of smoke in power equipment and handle partial occlusion. Finally, DS-UNet is compared with other smoke segmentation networks with similar structures, and it demonstrates better smoke segmentation performance.