From 0e67e017233448fdb70373b874fe44292f70f88f Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Wed, 28 Aug 2024 14:04:11 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E8=80=97=E6=97=B6=E6=97=A5?=
 =?UTF-8?q?=E5=BF=97?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 docker-compose.yml                | 2 +-
 photo_review/auto_photo_review.py | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 61dc8ce..f0d4b66 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,6 +1,6 @@
 x-env:
   &template
-  image: fcb_photo_review:1.13.5
+  image: fcb_photo_review:1.13.6
   restart: always
 
 services:
diff --git a/photo_review/auto_photo_review.py b/photo_review/auto_photo_review.py
index a9b1317..f38d259 100644
--- a/photo_review/auto_photo_review.py
+++ b/photo_review/auto_photo_review.py
@@ -82,12 +82,16 @@ def information_extraction(ie, phrecs, identity):
 
         image = image_util.read(img_path)
         target_images = []
+        det_time = time.time()
         target_images += detector.get_book_areas(image)  # 识别文档区域并裁剪
+        logging.info(f"det耗时：{time.time() - det_time}")
         if not target_images:
             target_images.append(image)  # 识别失败
         angle_count = defaultdict(int, {"0": 0})  # 分割后图片的最优角度统计
         for target_image in target_images:
+            dewarp_time = time.time()
             dewarped_image = dewarp.dewarp_image(target_image)  # 去扭曲
+            logging.info(f"dewarp耗时：{time.time() - dewarp_time}")
             angles = image_util.parse_rotation_angles(dewarped_image)
             zx_ie_results = []
             split_results = image_util.split(dewarped_image)