vinthony / ghost-free-shadow-removal

[AAAI 2020] Towards Ghost-free Shadow Removal via Dual Hierarchical Aggregation Network and Shadow Matting GAN

https://arxiv.org/abs/1911.08718

How to save predict_mask to png?

qpaozixu opened this issue 3 years ago · comments

qpaozixu commented 3 years ago

HI, your work is well, I want to save the predict_mask to png, how could I do?
Thanks for your help!

Xiaodong Cun commented 3 years ago

run the colab and add some similar code as in line 18