about the mask maps

[zjuirene]

Hi, thank you for your excellent releasing code!
I have a little question, how can I visualize the predicted mask maps when validation.
I attempted to use the save_image() API from torchvision.utils, but the results are not consistent with the provided mask maps from the paper.

Looking forward to your reply! Thank you!