Spaces:
Running
Running
nc
Browse files
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
|