jebin2 commited on
Commit
f9b5ba3
·
1 Parent(s): 55be59c
comic_panel_extractor/llm_panel_extractor.py CHANGED
@@ -235,7 +235,6 @@ def extract_panel_via_llm(input_image_path, config=None, reset=True):
235
  current_processed_image_path, accumulated_detected_boxes = panel_extractor.check_for_remaining_similarity(current_processed_image_path, accumulated_detected_boxes)
236
  all_processed_boxes = panel_extractor.pre_all_processed_boxes(accumulated_detected_boxes, original_width, original_height)
237
 
238
- all_path = [file for file in os.listdir(extractor_config.output_folder) if "_panel_" in file]
239
  remain_boxes = utils.get_remaining_areas((original_width, original_height), all_processed_boxes)
240
  if remain_boxes:
241
  remain_boxes = utils.is_valid_panel((original_width, original_height), remain_boxes, extractor_config.min_width_ratio, extractor_config.min_height_ratio)
@@ -244,6 +243,7 @@ def extract_panel_via_llm(input_image_path, config=None, reset=True):
244
  all_processed_boxes.extend(remain_boxes)
245
  accumulated_detected_boxes.extend(remain_boxes)
246
 
 
247
  print(f"Processing complete. Final result saved to: {extractor_config.output_folder}")
248
  print(f"Total panels detected: {len(all_path)}")
249
  return all_path, accumulated_detected_boxes, all_processed_boxes
 
235
  current_processed_image_path, accumulated_detected_boxes = panel_extractor.check_for_remaining_similarity(current_processed_image_path, accumulated_detected_boxes)
236
  all_processed_boxes = panel_extractor.pre_all_processed_boxes(accumulated_detected_boxes, original_width, original_height)
237
 
 
238
  remain_boxes = utils.get_remaining_areas((original_width, original_height), all_processed_boxes)
239
  if remain_boxes:
240
  remain_boxes = utils.is_valid_panel((original_width, original_height), remain_boxes, extractor_config.min_width_ratio, extractor_config.min_height_ratio)
 
243
  all_processed_boxes.extend(remain_boxes)
244
  accumulated_detected_boxes.extend(remain_boxes)
245
 
246
+ all_path = [file for file in os.listdir(extractor_config.output_folder) if "_panel_" in file]
247
  print(f"Processing complete. Final result saved to: {extractor_config.output_folder}")
248
  print(f"Total panels detected: {len(all_path)}")
249
  return all_path, accumulated_detected_boxes, all_processed_boxes