Write predicted mask [debugging]

Author: dzenanz

multic/segmentationschool/Codes/IterativePredict_1X.py

@@ -196,6 +196,12 @@ def predict(args):
 
 
         slide.close()
+
+        if args.base_dir and os.path.exists(args.base_dir):
+            mask_filename = args.base_dir + "/" + dirs['fileID'] + ".png"
+            print(f"Writing mask to file: {mask_filename}")
+            cv2.imwrite(mask_filename, wsiMask)
+
         print('\n\nStarting XML construction: ')
 
         if extname=='.scn':