| Literature DB >> 35096042 |
Wenbo Jiang1,2, Anshun Liu1,2.
Abstract
A network structure (DRSN-GAN) is proposed for image motion deblurring that combines a deep residual shrinkage network (DRSN) with a generative adversarial network (GAN) to address the issues of poor noise immunity and low generalizability in deblurring algorithms based solely on GANs. First, an end-to-end approach is used to recover a clear image from a blurred image, without the need to estimate a blurring kernel. Next, a DRSN is used as the generator in a GAN to remove noise from the input image while learning residuals to improve robustness. The BN and ReLU layers in the DRSN were moved to the front of the convolution layer, making the network easier to train. Finally, deblurring performance was verified using the GoPro, Köhler, and Lai datasets. Experimental results showed that deblurred images were produced with more subjective visual effects and a higher objective evaluation, compared with algorithms such as MPRNet. Furthermore, image edge and texture restoration effects were improved along with image quality. Our model produced slightly higher PSNR and SSIM values than the latest MPRNet, as well as increased YOLO detection accuracy. The number of required parameters in the DRSN-GAN was also reduced by 21.89%.Entities:
Mesh:
Year: 2022 PMID: 35096042 PMCID: PMC8799329 DOI: 10.1155/2022/5605846
Source DB: PubMed Journal: Comput Intell Neurosci
Comparison of common GAN models.
| Model | Mechanisms | Advantages | Disadvantages | Applicable scenarios |
|---|---|---|---|---|
| GAN | Generator and discriminator | Generate high-resolution images | Training instability, mode collapse, vanishing gradient | Generating partial images |
| CGAN | Add additional information to the input layer of the generator and discriminator | Effectively constrain the overly free GAN | Requires labeled data, training imbalance, the quality of generated images is low | Generating specified target images |
| DCGAN | Combine GAN with CNN, use BN and other techniques to train the model | Rich variety of generated images | The quality of generated images is low and model training is unstable | Generating most images |
| WGAN | Use Wasserstein distance instead of JS divergence in traditional GAN | Prevents GAN training instability and mode collapse | Unreasonable parameter settings can easily lead to gradient dispersion | The GAN model does not converge and the mode collapses |
| LSGAN | Use the least squares loss function instead of a traditional cross entropy loss function | Generate high-quality samples | The gradient vanishes or explodes during training | Generating high-quality images |
| BigGAN | Expand the scale of the model, use truncation and orthogonal regularization to train | Model training is stable and can generate highly clear images | Large number of parameters and difficult to train | Suitable for generating highly clear images |
Figure 1Improved GAN model.
Figure 2A structural diagram of the generator network.
Figure 3Comparison of two deep residual network structures.
The attention mechanism algorithm.
|
|
|
|
|
|
|
|
Figure 4The discriminator network structure.
The DRSN-GAN algorithm.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
A comparison of PSNR, SSIM, and Params for various algorithms using the GoPro dataset.
| Method | PSNR | SSIM | Params (M) |
|---|---|---|---|
| Xu et al. [ | 25.10 | 0.890 | — |
| Sun et al. [ | 24.64 | 0.842 | — |
| Deep Deblur [ | 29.23 | 0.916 | 11.8 |
| SRN [ | 30.10 | 0.932 | 6.8 |
| DeblurGAN [ | 28.64 | 0.927 | 14.5 |
| DeblurGAN-v2 [ | 29.55 | 0.934 | 60.9 |
| Suin et al. [ | 31.85 | 0.948 | 23.1 |
| MPRNet [ | 32.66 | 0.959 | 20.1 |
| DRSN-GAN | 32.67 | 0.965 | 15.7 |
Figure 5Sample (a) clear and (b) blurred images from the GoPro dataset. Corresponding results are shown for the (c) MPRNet and (d) DRSN-GAN algorithms.
Figure 6Sample (a) clear and (b) blurred images used for YOLO detection. Corresponding results are shown for the (c) MPRNet and (d) DRSN-GAN algorithms.
A comparison of PSNR and SSIM values for various algorithms using the Köhler dataset.
| Sun et al. [ | Deep Deblur [ | SRN [ | DeblurGAN [ | DeblurGAN-v2 [ | MPRNet [ | DRSN-GAN | |
|---|---|---|---|---|---|---|---|
| PSNR | 25.22 | 26.48 | 26.75 | 26.10 | 26.72 | 26.80 |
|
| SSIM | 0.773 | 0.807 | 0.837 | 0.816 | 0.836 | 0.836 |
|
Figure 7Sample (a) clear and (b) blurred images from the Köhler dataset. Corresponding results are shown for the (c) MPRNet and (d) DRSN-GAN algorithms.
Figure 8Sample (a) clear and (b) blurred images from the Lai dataset. Corresponding results are shown for the (c) MPRNet and (d) DRSN-GAN algorithms.
Figure 9Sample (a) clear and (b) blurred and noisy images from the GoPro dataset. Corresponding results are shown for the (c) MPRNet and (d) DRSN-GAN algorithms.
Figure 10Sample (a) clear and (b) blurred and noisy images from the Köhler dataset. Corresponding results are shown for the (c) MPRNet and (d) DRSN-GAN algorithms.