jebin2 commited on
Commit
d28110d
·
1 Parent(s): e3c032a
comic_panel_extractor/panel_segmentation.py CHANGED
@@ -174,7 +174,7 @@ def create_image_with_panels_removed(
174
  segmentation_mask: np.ndarray,
175
  output_folder: str,
176
  output_path: str,
177
- save_debug: False
178
  ) -> None:
179
  """
180
  Create a version of the original image with detected panels blacked out.
@@ -219,7 +219,7 @@ def main(output_folder, input_image_path, original_image_path):
219
  # Load the input image
220
  image = imageio.imread(input_image_path)
221
  original_image = imageio.imread(original_image_path)
222
- save_debug = True
223
  # Create segmentation mask
224
  segmentation_mask = create_segmentation_mask(image, save_debug=save_debug)
225
 
 
174
  segmentation_mask: np.ndarray,
175
  output_folder: str,
176
  output_path: str,
177
+ save_debug: True
178
  ) -> None:
179
  """
180
  Create a version of the original image with detected panels blacked out.
 
219
  # Load the input image
220
  image = imageio.imread(input_image_path)
221
  original_image = imageio.imread(original_image_path)
222
+ save_debug = False
223
  # Create segmentation mask
224
  segmentation_mask = create_segmentation_mask(image, save_debug=save_debug)
225