Home

Awesome

Boosting Flow-based Generative Super-Resolution Models via Learned Prior [CVPR 2024]

This is the official repository of "Boosting Flow-based Generative Super-Resolution Models via Learned Prior". <br><br> arXiv

Li-Yuan Tsao, Yi-Chen Lo, Chia-Che Chang, Hao-Wei Chen, Roy Tseng, Chien Feng, Chun-Yi Lee

Overview

In this work, we identify several challenges in flow-based SR methods, including grid artifacts, exploding inverses, and suboptimal results due to a fixed sampling temperature. To tackle these issues, we introduce a learned prior, which is predicted by the proposed latent module, to the inference phase of flow-based SR models. <br><br> <img width="65%" height="65%" alt="image" src="https://github.com/liyuantsao/FlowSR-LP/assets/73187544/e0900fdf-b11b-44d0-912d-ca502c2628b3"> <br>

This framework not only addresses the inherent issues in flow-based SR models but also enhances the quality of synthesized images without modifying the original design or pre-trained weights of these models. Our proposed framework is effective, flexible in design, and able to generalize to both fixed-scale and arbitrary-scale SR frameworks without requiring customized components. <br><br> <img width="1050" alt="image" src="https://github.com/liyuantsao/FlowSR-LP/assets/73187544/647a7011-cf67-461b-b170-079b151432b2">

Launch your experiments

This repository includes the training/evaluation code for LINF-LP, along with the evaluation code for SRFlow-LP (the training code will be released soon), which are the implementations after integrating the proposed latent module with LINF and SRFlow.

To run your experiments on LINF-LP and SRFlow-LP, please refer to LINF-LP and SRFlow-LP, respectively.

Results

Arbitrary-scale SR results

Generative SR results

<img width="500" alt="image" src="https://github.com/liyuantsao/FlowSR-LP/assets/73187544/01206333-3fb5-4a8a-aeba-76825ebd8536">

Qualitative Results

<img width="60%" height="60%" alt="image" src="https://github.com/liyuantsao/FlowSR-LP/assets/73187544/229ff47e-86fa-4490-a573-1533a38f0f94">
<img alt="image" src="https://github.com/liyuantsao/FlowSR-LP/assets/73187544/9d876f03-3ecc-4b0d-bda0-eee8b55e94f9"> <br> <img width="50%" height="50%" alt="image" src="https://github.com/liyuantsao/FlowSR-LP/assets/73187544/47810536-4ecc-431e-8911-bac983fd62b0">
<img width="70%" height="70%" alt="image" src="https://github.com/liyuantsao/FlowSR-LP/assets/73187544/4bee463a-ca1c-494e-9e7b-af5b0167b8f0">

Citation

If you find our work helpful for your research, we would greatly appreciate your assistance in sharing it with the community and citing it using the following BibTex. Thank you for supporting our research.

@InProceedings{Tsao_2024_CVPR,
    author    = {Tsao, Li-Yuan and Lo, Yi-Chen and Chang, Chia-Che and Chen, Hao-Wei and Tseng, Roy and Feng, Chien and Lee, Chun-Yi},
    title     = {Boosting Flow-based Generative Super-Resolution Models via Learned Prior},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    month     = {June},
    year      = {2024},
    pages     = {26005-26015}
}

Acknowledgements

Our codes are built on LINF (Paper, code) and SRFlow (Paper, Code), we appriciate their amazing works that advance this community.