import time

import cv2

from doc_dewarp import dewarp
from paddle_detection import detector
from util import image_util

if __name__ == '__main__':
    image = image_util.read("PH20240813000294_3_081442_1.jpg")
    images = detector.get_book_areas(image)
    for i in range(len(images)):
        dewarped_image = dewarp.dewarp_image(images[i])  # 去扭曲
        cv2.imwrite(f"book_{i}.jpg", dewarped_image)
    print(f"{len(images)}张图片已保存")
    time.sleep(3600)