Technology of Graphic & Image
|
617-622

Image restoration based on attention and convolution feature rearrangement

Wu Kaijun
Shan Hongquan
Mei Yuan
Xu Zehao
Wang Mengsi
School of Electronic & Information Engineering, Lanzhou Jiaotong University, Lanzhou 730030, China

Abstract

In recent years, the deep learning network model based on UNet and GAN(generative adversarial network) has shown unique advantages in the field of image restoration. But, there are still artifacts, blurring, degradation of texture details, difficult to repair large-area damage, and the repaired holes are incompatible with the background image. In order to solve the problems that the current existing models are not friendly to the repair of large-area damaged images and the degraded images after repair, this paper improved the Shift-UNet model by studying the existing methods. On the basis of U-Net and GAN, this paper added an improved attention mechanism Attention-UNet between each layer of encoder and decoder and integrated it into the Shift-UNet to form Attention-Shift-UNet. And through research, it changed the activation function of the original down-sampling part from Leaky_ ReLU to SiLU. The improved model not only achieved good results on the 64×64 central cover, but also realized the random cover, and the cover area increased from 20 % to 80 %. The experimental results show that the repair effect of the proposed model is better, especially for the repair of large-area damaged images. After testing on CelebA, Paris Architecture and Paris Stree View datasets, the evaluation indexes have been significantly improved, and SSIM has increased from 0.944 5 to 0.947 1. PSNR is increased from 27.992 7 to 28.553 6. L2 loss is decreased from 0.001 7 to 0.001 5.

Foundation Support

国家自然科学基金资助项目(61966022)
甘肃省自然科学基金资助项目(21JR7RA300)
甘肃省敦煌文物保护研究中心开放课题资助项目(GDW2021YB15)

Publish Information

DOI: 10.19734/j.issn.1001-3695.2022.06.0273
Publish at: Application Research of Computers Printed Article, Vol. 40, 2023 No. 2
Section: Technology of Graphic & Image
Pages: 617-622
Serial Number: 1001-3695(2023)02-051-0617-06

Publish History

[2022-08-11] Accepted Paper
[2023-02-05] Printed Article

Cite This Article

邬开俊, 单宏全, 梅源, 等. 基于注意力和卷积特征重排的图像修复 [J]. 计算机应用研究, 2023, 40 (2): 617-622. (Wu Kaijun, Shan Hongquan, Mei Yuan, et al. Image restoration based on attention and convolution feature rearrangement [J]. Application Research of Computers, 2023, 40 (2): 617-622. )

About the Journal

  • Application Research of Computers Monthly Journal
  • Journal ID ISSN 1001-3695
    CN  51-1196/TP

Application Research of Computers, founded in 1984, is an academic journal of computing technology sponsored by Sichuan Institute of Computer Sciences under the Science and Technology Department of Sichuan Province.

Aiming at the urgently needed cutting-edge technology in this discipline, Application Research of Computers reflects the mainstream technology, hot technology and the latest development trend of computer application research at home and abroad in a timely manner. The main contents of the journal include high-level academic papers in this discipline, the latest scientific research results and major application results. The contents of the columns involve new theories of computer discipline, basic computer theory, algorithm theory research, algorithm design and analysis, blockchain technology, system software and software engineering technology, pattern recognition and artificial intelligence, architecture, advanced computing, parallel processing, database technology, computer network and communication technology, information security technology, computer image graphics and its latest hot application technology.

Application Research of Computers has many high-level readers and authors, and its readers are mainly senior and middle-level researchers and engineers engaged in the field of computer science, as well as teachers and students majoring in computer science and related majors in colleges and universities. Over the years, the total citation frequency and Web download rate of Application Research of Computers have been ranked among the top of similar academic journals in this discipline, and the academic papers published are highly popular among the readers for their novelty, academics, foresight, orientation and practicality.


Indexed & Evaluation

  • The Second National Periodical Award 100 Key Journals
  • Double Effect Journal of China Journal Formation
  • the Core Journal of China (Peking University 2023 Edition)
  • the Core Journal for Science
  • Chinese Science Citation Database (CSCD) Source Journals
  • RCCSE Chinese Core Academic Journals
  • Journal of China Computer Federation
  • 2020-2022 The World Journal Clout Index (WJCI) Report of Scientific and Technological Periodicals
  • Full-text Source Journal of China Science and Technology Periodicals Database
  • Source Journal of China Academic Journals Comprehensive Evaluation Database
  • Source Journals of China Academic Journals (CD-ROM Version), China Journal Network
  • 2017-2019 China Outstanding Academic Journals with International Influence (Natural Science and Engineering Technology)
  • Source Journal of Top Academic Papers (F5000) Program of China's Excellent Science and Technology Journals
  • Source Journal of China Engineering Technology Electronic Information Network and Electronic Technology Literature Database
  • Source Journal of British Science Digest (INSPEC)
  • Japan Science and Technology Agency (JST) Source Journal
  • Russian Journal of Abstracts (AJ, VINITI) Source Journals
  • Full-text Journal of EBSCO, USA
  • Cambridge Scientific Abstracts (Natural Sciences) (CSA(NS)) core journals
  • Poland Copernicus Index (IC)
  • Ulrichsweb (USA)