search

Fayeben / GenerativeDiffusionPrior

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

Output issue when the batch_size is increased #9

Closed asdjia closed 1 year ago

asdjia commented 1 year ago

Hi,

Thanks for your code. It is a nice job. I met some issues when I run your code. For low light image enhancement task, when I set a batch_size greater than 4. There is an output issue. Please see the images below: batch_size=1 batch_size=8

when the batch_size <=4, it works fine. But when the batch_size > 4. the output is almost completely black. May I ask the reason and how to solve this issue?

I appreciate your help!

Fayeben commented 1 year ago

Thanks for your question! Because we initialize the light mask and light factor for a single image, therefore, the increment of batch_size may lead to some problems. We will revise it in the future version.