Files
fcb_photo_review/dewarp_test.py
2024-08-22 11:25:02 +08:00

13 lines
440 B
Python

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)