Update generate_masks_from_ImageJ.py

stefanherdy · Jun 26, 2023 · 877654a · 877654a
1 parent 49afe47
commit 877654a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/generate_masks_from_ImageJ.py b/generate_masks_from_ImageJ.py
@@ -19,7 +19,7 @@
 
 
 # Define the function to be executed on each image
-def save_mask(image_path):
+def extract_mask(image_path):
     mask = np.zeros((1080, 1920))
     for i ,roi in enumerate(roiread(image_path)):
         #print(roi)
@@ -42,7 +42,7 @@ def save_mask(image_path):
         # Construct the full path to the image
         image_path = os.path.join(folder_path, filename)
         # Execute the function on the image
-        mask = save_mask(image_path)
+        mask = extract_mask(image_path)
         cv2.imwrite('./mask/' + filename, mask)
         #cv2.imshow('test', mask)
         #cv2.waitKey(0)