• 基于条件生成对抗网络的目标流场散斑图像实时重建方法

    Real-time reconstruction of target flow field scattering image based on conditional generative adversarial network

    • 针对传统基于局部空间的数字图像相关(Digital Image Correlation, DIC)法以及基于时间域的光流法(Optical Flow Method, OFM)等散斑成像重建算法存在空间分辨率较低且难以实现实时重建等问题,提出一种基于条件生成对抗网络(Conditional Generative Adversarial Network, CGAN)的目标流场实时重建方法,称为快速散斑网络(Fast Scattering Network, FastScatNet)。该方法通过自设计的深度残差块和分组特征提取单元,优化了基于 U-Net 的生成器,并引入散斑图像对作为辅助信息以约束重建结果。在生成器与判别器的动态对抗中引入不同的损失函数指导各自的优化。以生成器为核心的损失函数中进一步融合了结构相似度指数、平均绝对损失和感知损失,实现了更优的图像重建效果和噪声抑制。仿真实验结果表明,在合成的目标流场图像数据集上,FastScatNet在不同重建分辨率下的结构相似度指数均高于0.956,推理速度在CPU和GPU下分别为30.2 frame/s和49.6 frame/s,相较传统DIC+OFM方法速度提升约9.44倍。 FastScatNet在对比原始含噪图像与降噪图像的实验中,SSIM 差异为 0.7% ~ 3.0%,证明其具备有效的噪声抑制能力。

       

      Abstract: For traditional scattering reconstruction algorithms such as digital image correlation based on local space and optical flow method based on the time domain have problems such as low spatial resolution and difficulty in real-time reconstruction. A real-time reconstruction method of the target flow field based on a conditional generative adversarial network called fast scattering network is proposed. This method optimizes the U-Net generator using a self-designed depth residual block and a grouping feature extraction unit and introduces speckle image pairs as auxiliary information to constrain the reconstruction results. In the dynamic interplay between the generator and discriminator, distinct loss functions are utilized to direct their optimization. The Structural Similarity Index (SSIM), average absolute loss, and perceived loss are further integrated into the loss function with the generator as the core to achieve better image reconstruction effect and noise suppression. The simulation results show that the SSIM of FastScatNet is higher than 0.956 at different reconstruction resolutions, and the inference speed is 30.2 frame/s on CPU and 49.6 frame/s on GPU, which is 9.44 times faster than the traditional DIC+OFM method. In the experiment of comparing the original noisy image and the denoised image, the SSIM difference of FastScatNet is 0.7% ~ 3.0%, which proves that FastScatNet has effective noise suppression ability.

       

    /

    返回文章
    返回