17 lines
478 B
Python
17 lines
478 B
Python
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)
|