Industrial smoke image segmentation based on frequency domain multi-directional C-UNet and dynamic loss
The accurate segmentation of the smoke in industrial smoke pollution level monitoring is an important pre-requisite for pollution level determination.The typical challenges in feature extraction of smoke include blurred edges,difficult extraction of edge directional detail information and inaccurate segmentation.In this study,a frequency domain multi-directional C-UNet(Contourlet U-Net)and dynamic loss industrial smoke image segmentation method is proposed,aiming to provide support to overcome these problems.Firstly,the contourlet multi-directional decomposition down-sampling structure is constructed to enhance the ability to extract edge direction information of smoke in the encoding stage.Secondly,the contourlet transform is used to extract detailed information on the eight edge directions of smoke for skip connections,improving the accuracy of detail information expression during continuous sampling.Then,the con-tourlet detail reconstruction up-sampling structure is constructed to enhance the recovery ability of edge detail information of smoke in the decoding stage.Finally,a dynamic weighting strategy is proposed to construct a combined loss function to optimize the training network and enhance the network's ability to extract smoke edge features.The results show that compared with U-Net and other similar methods,the proposed method has a better improvement in indicators,improves the accuracy of smoke edge segmentation,and the segmentation effect on different smoke scenes is better than the existing segmentation model.
industrial smokeimage segmentationcontourlet transformfeature extractiondynamic loss function
国家科技期刊平台
NSTL
万方数据
