DiffFPR: Diffusion Prior for Oversampled Fourier Phase Retrieval

Ji Li, Chao Wang.

This repository contains the code with the paper "DiffFPR: Diffusion Prior for Oversampled Fourier Phase Retrieval", which is accepted by ICML 2024.

This code is based on the OpenAI Guided Diffusion and DiffPIR.

---

Setting Up

Clone and Install

git clone https://github.com/Chilie/DiffFPR.git

cd DiffFPR

pip install -r requirements.txt

Model Download

From the link, download the checkpoint "ffhq_10m.pt" and paste it to ./model_zoo/

mkdir model_zoo
mv {DOWNLOAD_DIR}/ffqh_10m.pt ./model_zoo/

Do not forget to rename "ffhq_10m" to "diffusion_ffhq_10m" for code consistency.

Inference Code

python main_pr.py

Citation

If you find this repo helpful, please cite:

@inproceedings{li2024diff, % DiffFPR

title={DiffFPR: Diffusion Prior for Oversampled Fourier Phase Retrieval},

author={Ji Li and Chao Wang},

booktitle={Proceedings of the 41th international conference on machine learning},

year={2024},

}

Acknowledgments

This work was partly supported by the fund from Capital Normal University.