Thank you for your great work! I am curious about testing with my own dataset. I just want to get a mask image from my data. Thank you very much.