Algolzw / image-restoration-sde

Image Restoration with Mean-Reverting Stochastic Differential Equations, ICML 2023. Winning solution of the NTIRE 2023 Image Shadow Removal Challenge.

Home Page:https://algolzw.github.io/ir-sde/index.html

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

refusion model

gayathrivenkat17 opened this issue · comments

Hi how can i train the refusion model for dehazing task?
in the dehazing folder if we train what we get is latent_dehazing.pth.. I want to get the other latent_refusion_dehazing.pth.. what should i do to train it from scratch if i have to use this for different datasets

Hi if you want to retrain the dehazing model on other datasets then you need to retrain the unet-latent model and use it in training the latent-dehazing model. BTW, the training of unet-latent is very fast and easy. :)

thank you