search

Algolzw / image-restoration-sde

Image Restoration with Mean-Reverting Stochastic Differential Equations, ICML 2023. Winning solution of the NTIRE 2023 Image Shadow Removal Challenge.
https://algolzw.github.io/ir-sde/index.html
MIT License
521 stars 39 forks source link

refusion model #58

Open gayathrivenkat17 opened 8 months ago

gayathrivenkat17 commented 8 months ago

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

Algolzw commented 8 months ago

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. :)

gayathrivenkat17 commented 8 months ago

thank you