|
|
import cv2 |
|
|
import numpy as np |
|
|
import os |
|
|
import time |
|
|
from datetime import datetime |
|
|
|
|
|
|
|
|
def log_execution(func): |
|
|
def wrapper(*args, **kwargs): |
|
|
start_time = time.time() |
|
|
start_str = datetime.fromtimestamp(start_time).strftime('%Y-%m-%d %H:%M:%S') |
|
|
|
|
|
result = func(*args, **kwargs) |
|
|
|
|
|
end_time = time.time() |
|
|
end_str = datetime.fromtimestamp(end_time).strftime('%Y-%m-%d %H:%M:%S') |
|
|
duration = end_time - start_time |
|
|
|
|
|
return result |
|
|
return wrapper |
|
|
|
|
|
@log_execution |
|
|
def split_comic_panels(image_path, output_dir): |
|
|
img = cv2.imread(image_path) |
|
|
if img is None: |
|
|
print(f"Error: Could not read image from {image_path}") |
|
|
return |
|
|
|
|
|
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) |
|
|
|
|
|
th = cv2.adaptiveThreshold(gray, 255, |
|
|
cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY_INV, |
|
|
blockSize=51, C=10) |
|
|
kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (15,15)) |
|
|
clean = cv2.morphologyEx(th, cv2.MORPH_CLOSE, kernel) |
|
|
|
|
|
cnts, _ = cv2.findContours(clean, cv2.RETR_EXTERNAL, |
|
|
cv2.CHAIN_APPROX_SIMPLE) |
|
|
panels = [] |
|
|
for c in cnts: |
|
|
area = cv2.contourArea(c) |
|
|
if area < 10000: |
|
|
continue |
|
|
peri = cv2.arcLength(c, True) |
|
|
approx = cv2.approxPolyDP(c, 0.02 * peri, True) |
|
|
panels.append((cv2.boundingRect(approx), c)) |
|
|
@log_execution |
|
|
def panel_key(item): |
|
|
(x,y,w,h), _ = item |
|
|
return (y//50, x) |
|
|
|
|
|
panels_sorted = sorted(panels, key=panel_key) |
|
|
|
|
|
if not os.path.exists(output_dir): |
|
|
os.makedirs(output_dir) |
|
|
|
|
|
for i, ((x,y,w,h), contour) in enumerate(panels_sorted, 1): |
|
|
crop = img[y:y+h, x:x+w] |
|
|
cv2.imwrite(os.path.join(output_dir, f"panel_{i:02d}.png"), crop) |
|
|
|
|
|
if __name__ == "__main__": |
|
|
if not os.path.exists("comic_page.jpg"): |
|
|
dummy_image = np.zeros((600, 800, 3), dtype=np.uint8) |
|
|
cv2.putText(dummy_image, "Comic Page", (300, 300), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2, cv2.LINE_AA) |
|
|
cv2.rectangle(dummy_image, (50, 50), (350, 250), (255,255,255), 5) |
|
|
cv2.rectangle(dummy_image, (400, 50), (750, 250), (255,255,255), 5) |
|
|
cv2.rectangle(dummy_image, (50, 300), (750, 550), (255,255,255), 5) |
|
|
cv2.imwrite("comic_page.jpg", dummy_image) |
|
|
|
|
|
if not os.path.exists("output_panels"): |
|
|
os.makedirs("output_panels") |
|
|
split_comic_panels("comic_page.jpg", "output_panels") |
|
|
print("Comic panels split and saved to output_panels directory.") |
