search

Fayeben / GenerativeDiffusionPrior

Generative Diffusion Prior for Unified Image Restoration and Enhancement (CVPR2023)
Apache License 2.0
277 stars 30 forks source link

Extension for other types of inpainting tasks #24

Closed jiliya1991 closed 1 year ago

jiliya1991 commented 1 year ago

Great work! If I want to try the box-based inpainting tasks instead of 25%, how to extend the code? Thanks

Fayeben commented 1 year ago

Yes, our inpainting script can be easily applied on box-based inpainting. You can change the degradation model in this line: https://github.com/Fayeben/GenerativeDiffusionPrior/blob/b34494cd53e344a5c726e502f552cd7fb888aad0/scripts/sample_x0_inp.py#L102C22-L102C22