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
576 stars 42 forks source link

building and training a custom dataset #114

Open fn6767 opened 1 week ago

fn6767 commented 1 week ago

Excuse me, if I want to train on my own dataset for image restoration and completion tasks, is the img size limited to 256? What size can I set? Can I continue training on your provided pre-trained models with my own dataset?

Algolzw commented 1 week ago

Hi! Sure, you can finetune the model on your dataset. The size is also not limited. We usually crop images with a fixed size during training but it won't affect the inference.

fn6767 commented 1 week ago

Thanks!