diff --git a/PH20240813000294_3_081442_1.jpg b/PH20240813000294_3_081442_1.jpg new file mode 100644 index 0000000..2e20a7e Binary files /dev/null and b/PH20240813000294_3_081442_1.jpg differ diff --git a/dewarp_test.py b/dewarp_test.py new file mode 100644 index 0000000..13b8d14 --- /dev/null +++ b/dewarp_test.py @@ -0,0 +1,16 @@ +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) diff --git a/docker-compose-dewarp.yml b/docker-compose-dewarp.yml new file mode 100644 index 0000000..8b6850b --- /dev/null +++ b/docker-compose-dewarp.yml @@ -0,0 +1,22 @@ +x-env: + &template + image: fcb_photo_review:2.0.0 + +services: + photo_dewarp: + <<: *template + build: + context: . + container_name: photo_dewarp + hostname: photo_dewarp + volumes: + - ./log:/app/log + - ./model:/app/model + command: [ "dewarp_test.py" ] + deploy: + resources: + reservations: + devices: + - device_ids: [ "1" ] + capabilities: [ "gpu" ] + driver: "nvidia" \ No newline at end of file