search

Fayeben / GenerativeDiffusionPrior

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

关于训练 #7

Closed jjb202 closed 1 year ago

jjb202 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!