import time

import cv2

from doc_dewarp import dewarp
from object_detection import doc_detect
from util import image_util

if __name__ == '__main__':
    image = image_util.read("PH20240813000294_3_081442_1.jpg")
    target_images = []
    target_images += doc_detect.capture_target_area(image)  # 识别文档区域并裁剪
    dewarped_image = dewarp.dewarp_image(target_images[0])  # 去扭曲
    cv2.imwrite("dewarped_image.jpg", dewarped_image)
    time.sleep(3600)