dump_percent implemented under ImageDumper.create_composite_image

Approved for merge
VIDA-NYU · Nov 29, 2023 · f68fd90 · f68fd90
1 parent b0dcda8
commit f68fd90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/tile2net/tileseg/utils/misc.py b/src/tile2net/tileseg/utils/misc.py
@@ -295,6 +295,7 @@ def save_prob_and_err_mask(self, dump_dict, img_name, idx, prediction):
             return True, err_pil
         return False, err_pil
 
+    dump_percent = 100  # first one always dumps if args.dump_percent != 0
     def create_composite_image(self, input_image, prediction_pil, img_name):
         if not self.args.dump_percent:
             return
@@ -333,7 +334,6 @@ def get_dump_assets(self, dump_dict, img_name, idx, colorize_mask_fn, to_tensorb
                 mask_pil.save(mask_fn)
                 to_tensorboard.append(self.visualize(mask_pil))
 
-    dump_percent = 100  # first one always dumps
 
     def dump(self, dump_dict, val_idx, testing=None, grid=None):