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..cd5cba5 --- /dev/null +++ b/dewarp_test.py @@ -0,0 +1,12 @@ +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) diff --git a/docker-compose-dewarp.yml b/docker-compose-dewarp.yml new file mode 100644 index 0000000..283d88e --- /dev/null +++ b/docker-compose-dewarp.yml @@ -0,0 +1,23 @@ +x-env: + &template + image: fcb_photo_review:2.0.0 + restart: always + +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