How to save predict_mask to png?

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

297 stars 59 forks source link

Closed qpaozixu closed 2 years ago

qpaozixu commented 2 years ago

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

vinthony commented 2 years ago

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