white label

comic_panel_extractor/config.py

@@ -61,7 +61,7 @@ class Config:
 
         # Derive YOLO_MODEL_NAME if empty
         if not self.YOLO_MODEL_NAME:
-            self.YOLO_MODEL_NAME = f"comic_panel_{self.YOLO_BASE_MODEL_NAME}"
+            self.YOLO_MODEL_NAME = f"final_model_{self.YOLO_BASE_MODEL_NAME}"
 
         # Derived paths
         self.yolo_base_model_path = os.path.join(self.current_path, f"{self.YOLO_BASE_MODEL_NAME}.pt")

comic_panel_extractor/config.toml

@@ -4,6 +4,6 @@ BATCH = 10
 RESUME_TRAIN = true
 RECREATE_DATASET = true
 YOLO_BASE_MODEL_NAME = "yolo11s-seg"
-YOLO_MODEL_NAME = "comic_panel_yolo11s-seg"
+YOLO_MODEL_NAME = "final_model_yolo11s-seg"
 IMAGE_SOURCE_PATH = "images"
 YOLO_MODEL_REMOTE_URL = "https://huggingface.co/mosesb/best-comic-panel-detection/resolve/main/best.pt"

comic_panel_extractor/config.toml.bak

@@ -3,5 +3,5 @@ DEFAULT_IMAGE_SIZE=640
 BATCH=10
 RESUME_TRAIN="true"
 YOLO_BASE_MODEL_NAME="yolo11s-seg"
-YOLO_MODEL_NAME="comic_panel_yolo11s-seg"
+YOLO_MODEL_NAME="final_model_yolo11s-seg"
 IMAGE_SOURCE_PATH="images"

comic_panel_extractor/main.py

@@ -43,40 +43,7 @@ class ComicPanelExtractor:
             return None, None, all_path
         except Exception as e:
             print(f'{str(e)} {traceback.format_exc()}')
-
-        processed_image_path = self.image_processor.group_colors(self.config.input_path)
-
-        processed_image_path = BorderPanelExtractor(self.config).main(processed_image_path)
-
-        self.config.black_overlay_input_path = processed_image_path
-
-        _, _, processed_image_path = self.image_processor.preprocess_image(processed_image_path)
-
-        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.thick_black(processed_image_path)
-
-        processed_image_path = self.image_processor.remove_small_regions(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.detect_small_objects_and_set_white(processed_image_path)
-
-        processed_image_path = self.image_processor.thin_image_borders(processed_image_path)
-
-        panel_images, panel_data, all_panel_path = self.panel_extractor.extract_panels(
-            processed_image_path
-        )
-        
-        return panel_images, panel_data, all_panel_path
+            raise
     
     def cleanup(self):
         """Clean up temporary files if needed."""