constant

comic_panel_extractor/config.py

@@ -21,9 +21,18 @@ class Config:
 
 	# Read from TOML config
 	EPOCH: int = int(config_data.get("EPOCH", 200))
+	DEFAULT_IMAGE_SIZE: int = int(config_data.get("DEFAULT_IMAGE_SIZE", 640))
+	BATCH: int = int(config_data.get("BATCH", 10))
+	RESUME_TRAIN: bool = str(config_data.get("RESUME_TRAIN", "True")).lower() in ("1", "true", "yes")
 	YOLO_BASE_MODEL_NAME: str = config_data.get("YOLO_BASE_MODEL_NAME", "yolo11s-seg")
 	YOLO_MODEL_NAME: str = config_data.get("YOLO_MODEL_NAME", f"comic_panel_{YOLO_BASE_MODEL_NAME}")
-	IMAGE_SOURCE_PATH: str = config_data.get("IMAGE_SOURCE_PATH", "")
+	image_path_from_config = config_data.get("IMAGE_SOURCE_PATH", "")
+	# Ensure absolute path
+	IMAGE_SOURCE_PATH: str = (
+		image_path_from_config
+		if os.path.isabs(image_path_from_config)
+		else os.path.join(CURRENT_PATH, image_path_from_config)
+	)
 
 	# Derived paths
 	yolo_base_model_path: str = f"{current_path}/{YOLO_BASE_MODEL_NAME}.pt"
@@ -46,7 +55,6 @@ class Config:
 	panel_filename_pattern: str = r"panel_\d+_\((\d+), (\d+), (\d+), (\d+)\)\.jpg"
 
 	# Static constants
-	DEFAULT_IMAGE_SIZE: int = 640
 	SUPPORTED_EXTENSIONS: list = ('jpg', 'jpeg', 'png', 'JPG', 'JPEG', 'PNG')
 
 def get_text_cood_file_path(config: Config):

comic_panel_extractor/config.toml

@@ -1,4 +1,7 @@
 EPOCH=200
+DEFAULT_IMAGE_SIZE=640
+BATCH=10
+RESUME_TRAIN="true"
 YOLO_BASE_MODEL_NAME="yolo11s-seg"
 YOLO_MODEL_NAME="comic_panel_yolo11s-seg"
-IMAGE_SOURCE_PATH="/home/jebin/git/comic-panel-extractor/comic_panel_extractor/images"
+IMAGE_SOURCE_PATH="comic-panel-extractor/comic_panel_extractor/images"

comic_panel_extractor/static/annotator.html

@@ -683,9 +683,6 @@
                         📤 Drop or click to upload
                     </label>
                 </div>
-                <button class="btn btn-primary btn-sm trainBtn" id="trainBtn">
-                    Train
-                </button>
             </div>
 
             <!-- Progress -->
@@ -788,9 +785,18 @@
 
 
             <!-- Quick Actions -->
-            <!-- <div class="sidebar-section">
+            <div class="sidebar-section">
                 <div class="section-title">Actions</div>
-                <button class="btn btn-primary btn-block" id="reloadBtn">
+                <button class="btn btn-primary btn-sm trainBtn" id="trainBtn">
+                    Train
+                </button>
+                <button class="btn btn-primary btn-sm trainBtn" id="deployModalBtn">
+                    Deploy Model
+                </button>
+                <button class="btn btn-primary btn-sm trainBtn" id="resetModalBtn">
+                    Reset Model
+                </button>
+                <!-- <button class="btn btn-primary btn-block" id="reloadBtn">
                     🔄 Reload Annotations
                 </button>
                 <button class="btn btn-primary btn-block" id="detectBtn" style="display: none; margin-top: 8px;">
@@ -798,8 +804,8 @@
                 </button>
                 <button class="btn btn-secondary btn-block" id="downloadBtn" style="display: none; margin-top: 8px;">
                     📥 Download
-                </button>
-            </div>-->
+                </button> -->
+            </div>
 
             <!-- Quick Help -->
             <div class="sidebar-section">

comic_panel_extractor/yolo_manager.py

@@ -79,7 +79,7 @@ class YOLOManager:
               data_yaml_path: str,
               run_name: Optional[str] = None,
               device: int = 0,
-              resume: bool = True,
+              resume: bool = Config.RESUME_TRAIN,
               **kwargs) -> YOLO:
         """
         Train YOLO model with given parameters.
@@ -108,7 +108,7 @@ class YOLOManager:
             'data': data_yaml_path,
             'imgsz': Config.DEFAULT_IMAGE_SIZE,
             'epochs': Config.EPOCH,
-            'batch': 10,
+            'batch': Config.BATCH,
             'name': run_name,
             'device': device,
             'cache': True,