diff --git a/dewarp_test.py b/dewarp_test.py
index 6cbd478..cd5cba5 100644
--- a/dewarp_test.py
+++ b/dewarp_test.py
@@ -1,9 +1,12 @@
-import time
+import cv2
+
+from doc_dewarp import dewarp
+from object_detection import doc_detect
+from util import image_util
 
 if __name__ == '__main__':
-    time.sleep(3600)
-    # 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)
+    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)