Spaces:
Running
Running
debug f
Browse files
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:
|
| 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 =
|
| 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 |
|