Can you make .py file of the predicted masks for each image in the test set? 