So when I run the code I get a mask that seems to be made of 32x32 large square on a whole 512x512 pixels image. Is there a way to increase the resolution of the prediction and thus the mask generated? Thanks a lot in advance.