image-restoration-sde icon indicating copy to clipboard operation
image-restoration-sde copied to clipboard

Published 20 hours ago • verified publisher icon Algolzw

Shadow removing model

Open heorhiikalaichev opened this issue 1 year ago • 3 comments

Hi!

I can't find a pretrained model for the shadow removing task in the list you provided.

https://drive.google.com/drive/folders/1RxkJ9u-vR3UlJ5PYy1D7jienEZd5d6my

heorhiikalaichev avatar Jul 27 '23 13:07 heorhiikalaichev

Hi here is the link for our Refusion model’s weights and results: https://drive.google.com/drive/folders/1Xy7z9CH0weck775ixePnjD6fZIjAbre7?usp=share_link

Algolzw avatar Jul 27 '23 14:07 Algolzw

Thank you so much!

Is it possible to provide a tile processing script for deshadow task? I have out of memory when trying to process 3072*4608 image.

heorhiikalaichev avatar Aug 03 '23 15:08 heorhiikalaichev

Hi, I think the current shadow removal code can't process such a large-size image. One possible way is to downsample the image with a smaller size (e.g., [w/2, h/2]) and then upsample it after deshadowing. It might lose some accuracy but is the simplest way to process large images.

And another choice is to re-train a latent-refusion model to process large-size images (please refer to the code of unet-latent and latent-dehazing).

Algolzw avatar Aug 03 '23 19:08 Algolzw