Fayeben / GenerativeDiffusionPrior

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

关于训练 #7

Closed szpxmu closed 1 year ago

szpxmu commented 1 year ago

你好,我想换个数据集需要训练吗?还是只要用预训练模型来采样就行?

chensming commented 1 year ago

你好,我想换个数据集需要训练吗?还是只要用预训练模型来采样就行?

兄弟你好,我也在想这个问题,个人感觉你另一个数据集分布不能和imagenet差太多,毕竟那个diffusion是在imagenet训练的(但我不确定)

Fayeben commented 1 year ago

在imagenet上预训练的模型换个数据集不需要重新训练。分布差异的问题确实存在,但是恢复出图片的内容还是会尽可能一致,只是差异体现在图片整体色调上,用imagenet预训练模型恢复出来的会倾向于imagenet的整体色调,而不那么接近于GT。可以参考Low light enhancement和HDR任务,他们的色调和GT其实是有差异的,但是内容是尽可能保持一致的。

Fayeben commented 1 year ago

Since there are no more questions, I'll mark it closed. Thanks!