new flow

comic_panel_extractor/image_processor.py

@@ -752,7 +752,7 @@ class ImageProcessor:
         cv2.imwrite(output_path, output)
         return output_path
 
-    def detect_objects_and_draw_boxess_and_set_white(self, image_path, file_name="all_objects_detected.jpg", output_folder=None):
+    def detect_small_objects_and_set_white(self, image_path, file_name="detect_small_objects_and_set_white.jpg", output_folder=None):
         output_folder = output_folder or self.config.output_folder
 
         # Load image

comic_panel_extractor/main.py

@@ -28,7 +28,7 @@ class ComicPanelExtractor:
         """Complete pipeline to extract panels from a comic image."""
         print(f"Starting panel extraction for: {self.config.input_path}")
 
-        processed_image_path = self.config.input_path
+        processed_image_path = self.image_processor.group_colors(self.config.input_path)
 
         processed_image_path = BorderPanelExtractor(self.config).main(processed_image_path)
 
@@ -38,21 +38,21 @@ class ComicPanelExtractor:
 
         processed_image_path = self.image_processor.thin_image_borders(processed_image_path)
 
+        processed_image_path = self.image_processor.remove_diagonal_lines_and_set_white(processed_image_path)
+
         processed_image_path = self.image_processor.remove_dangling_lines(processed_image_path)
 
         processed_image_path = self.image_processor.remove_diagonal_only_cells(processed_image_path)
 
-        processed_image_path = self.image_processor.remove_small_continuity_components(processed_image_path)
-
         processed_image_path = self.image_processor.thick_black(processed_image_path)
 
         processed_image_path = self.image_processor.remove_small_regions(processed_image_path)
 
-        processed_image_path = self.image_processor.remove_diagonal_lines(processed_image_path)
-
         processed_image_path = self.image_processor.remove_small_regions(processed_image_path)
 
-        processed_image_path = self.image_processor.connect_horizontal_vertical_gaps(processed_image_path)
+        # processed_image_path = self.image_processor.connect_horizontal_vertical_gaps(processed_image_path)
+
+        processed_image_path = self.image_processor.detect_small_objects_and_set_white(processed_image_path)
 
         processed_image_path = self.image_processor.thin_image_borders(processed_image_path)
 

comic_panel_extractor/panel_segmentation.py

@@ -70,7 +70,6 @@ def extract_fully_white_panels(
 
         # 1. Threshold check
         if (
-            area < min_area_ratio * image_area or
             w < min_width_ratio * img_w or
             h < min_height_ratio * img_h
         ):