jebin2 commited on
Commit
e30f731
·
1 Parent(s): c59b15b

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 = visual_output[y1:y2, x1:x2]
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 = visual_output[y1:y2, x1:x2]
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)