Boundary-aware Decoupled Flow Networks for Realistic Extreme Rescaling

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Li, Jinmin, Dai, Tao, Zhang, Jingyun, Liu, Kang, Wang, Jun, Wang, Shaoming, Xia, Shu-Tao, Guo, Rizen
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910443874811904
author Li, Jinmin
Dai, Tao
Zhang, Jingyun
Liu, Kang
Wang, Jun
Wang, Shaoming
Xia, Shu-Tao
Guo, Rizen
author_facet Li, Jinmin
Dai, Tao
Zhang, Jingyun
Liu, Kang
Wang, Jun
Wang, Shaoming
Xia, Shu-Tao
Guo, Rizen
contents Recently developed generative methods, including invertible rescaling network (IRN) based and generative adversarial network (GAN) based methods, have demonstrated exceptional performance in image rescaling. However, IRN-based methods tend to produce over-smoothed results, while GAN-based methods easily generate fake details, which thus hinders their real applications. To address this issue, we propose Boundary-aware Decoupled Flow Networks (BDFlow) to generate realistic and visually pleasing results. Unlike previous methods that model high-frequency information as standard Gaussian distribution directly, our BDFlow first decouples the high-frequency information into \textit{semantic high-frequency} that adheres to a Boundary distribution and \textit{non-semantic high-frequency} counterpart that adheres to a Gaussian distribution. Specifically, to capture semantic high-frequency parts accurately, we use Boundary-aware Mask (BAM) to constrain the model to produce rich textures, while non-semantic high-frequency part is randomly sampled from a Gaussian distribution.Comprehensive experiments demonstrate that our BDFlow significantly outperforms other state-of-the-art methods while maintaining lower complexity. Notably, our BDFlow improves the PSNR by 4.4 dB and the SSIM by 0.1 on average over GRAIN, utilizing only 74% of the parameters and 20% of the computation. The code will be available at https://github.com/THU-Kingmin/BAFlow.
format Preprint
id arxiv_https___arxiv_org_abs_2405_02941
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Boundary-aware Decoupled Flow Networks for Realistic Extreme Rescaling
Li, Jinmin
Dai, Tao
Zhang, Jingyun
Liu, Kang
Wang, Jun
Wang, Shaoming
Xia, Shu-Tao
Guo, Rizen
Computer Vision and Pattern Recognition
Recently developed generative methods, including invertible rescaling network (IRN) based and generative adversarial network (GAN) based methods, have demonstrated exceptional performance in image rescaling. However, IRN-based methods tend to produce over-smoothed results, while GAN-based methods easily generate fake details, which thus hinders their real applications. To address this issue, we propose Boundary-aware Decoupled Flow Networks (BDFlow) to generate realistic and visually pleasing results. Unlike previous methods that model high-frequency information as standard Gaussian distribution directly, our BDFlow first decouples the high-frequency information into \textit{semantic high-frequency} that adheres to a Boundary distribution and \textit{non-semantic high-frequency} counterpart that adheres to a Gaussian distribution. Specifically, to capture semantic high-frequency parts accurately, we use Boundary-aware Mask (BAM) to constrain the model to produce rich textures, while non-semantic high-frequency part is randomly sampled from a Gaussian distribution.Comprehensive experiments demonstrate that our BDFlow significantly outperforms other state-of-the-art methods while maintaining lower complexity. Notably, our BDFlow improves the PSNR by 4.4 dB and the SSIM by 0.1 on average over GRAIN, utilizing only 74% of the parameters and 20% of the computation. The code will be available at https://github.com/THU-Kingmin/BAFlow.
title Boundary-aware Decoupled Flow Networks for Realistic Extreme Rescaling
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2405.02941