search

GuoLanqing / ShadowDiffusion

ShadowDiffusion (CVPR2023), Pytorch implementation
133 stars 6 forks source link

About Predict #22

Closed finnn00 closed 2 months ago

finnn00 commented 8 months ago

If now I only have a shadow image, how can the model predict and generate the shadow-free image ? Does the model include a mask detector?

GuoLanqing commented 8 months ago

This repo does not include a mask detector. But you can try to initialize the shadow mask as all-one matrix and re-train (or finetune) the model.