Spaces:
Running
Running
image isue fix
Browse files
comic_panel_extractor/panel_extractor.py
CHANGED
|
@@ -250,6 +250,7 @@ class PanelExtractor:
|
|
| 250 |
|
| 251 |
def _save_panels(self, panels: List[Tuple[int, int, int, int]], original: np.ndarray, width: int, height: int) -> Tuple[List[np.ndarray], List[PanelData], List[str]]:
|
| 252 |
"""Save panel images and return panel data."""
|
|
|
|
| 253 |
visual_output = original.copy()
|
| 254 |
panel_images = []
|
| 255 |
panel_data = []
|
|
@@ -307,7 +308,7 @@ class PanelExtractor:
|
|
| 307 |
continue
|
| 308 |
|
| 309 |
# Save panel
|
| 310 |
-
panel_img =
|
| 311 |
panel_images.append(panel_img)
|
| 312 |
panel_info = PanelData.from_coordinates(x1, y1, x2, y2)
|
| 313 |
panel_data.append(panel_info)
|
|
@@ -324,7 +325,7 @@ class PanelExtractor:
|
|
| 324 |
# If no valid panels and full-page backup exists
|
| 325 |
if not panel_images and maybe_full_page_panel and panel_idx == 0:
|
| 326 |
idx, (x1, y1, x2, y2) = maybe_full_page_panel
|
| 327 |
-
panel_img =
|
| 328 |
panel_images.append(panel_img)
|
| 329 |
panel_info = PanelData.from_coordinates(x1, y1, x2, y2)
|
| 330 |
panel_data.append(panel_info)
|
|
|
|
| 250 |
|
| 251 |
def _save_panels(self, panels: List[Tuple[int, int, int, int]], original: np.ndarray, width: int, height: int) -> Tuple[List[np.ndarray], List[PanelData], List[str]]:
|
| 252 |
"""Save panel images and return panel data."""
|
| 253 |
+
original_image = cv2.imread(self.config.input_path)
|
| 254 |
visual_output = original.copy()
|
| 255 |
panel_images = []
|
| 256 |
panel_data = []
|
|
|
|
| 308 |
continue
|
| 309 |
|
| 310 |
# Save panel
|
| 311 |
+
panel_img = original_image[y1:y2, x1:x2]
|
| 312 |
panel_images.append(panel_img)
|
| 313 |
panel_info = PanelData.from_coordinates(x1, y1, x2, y2)
|
| 314 |
panel_data.append(panel_info)
|
|
|
|
| 325 |
# If no valid panels and full-page backup exists
|
| 326 |
if not panel_images and maybe_full_page_panel and panel_idx == 0:
|
| 327 |
idx, (x1, y1, x2, y2) = maybe_full_page_panel
|
| 328 |
+
panel_img = original_image[y1:y2, x1:x2]
|
| 329 |
panel_images.append(panel_img)
|
| 330 |
panel_info = PanelData.from_coordinates(x1, y1, x2, y2)
|
| 331 |
panel_data.append(panel_info)
|