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
558 stars 41 forks source link

how to train latent encoder and decoder #44

Closed milkieTigera closed 1 year ago

milkieTigera commented 1 year ago

how to train latent encoder and decoder，i have not found code about it, that mean only pretrained encoder-decoder can be used?

Algolzw commented 1 year ago

Hi, you can easily train the encoder and decoder in the unet-latent dir. The only thing you need to change is to modify the path in the train.yml file.

milkieTigera commented 1 year ago

thanks a lot> Hi, you can easily train the encoder and decoder in the unet-latent dir. The only thing you need to change is to modify the path in the train.yml file.