From a3fa1e502ea867f4f2de0255fa4d397eb08d22f5 Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Sun, 29 Sep 2024 13:55:37 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4=E5=9B=BE=E7=89=87=E5=90=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 services/paddle_services/det_book.py | 4 ++--
 services/paddle_services/dewarp.py   | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/services/paddle_services/det_book.py b/services/paddle_services/det_book.py
index efdc50f..6cab66c 100644
--- a/services/paddle_services/det_book.py
+++ b/services/paddle_services/det_book.py
@@ -17,10 +17,10 @@ def main():
     img_path = request.form.get('img_path')
     result = detector.get_book_areas(img_path)
 
-    dirname, img_name, ext = parse_img_path(img_path)
+    dirname, img_name, img_ext = parse_img_path(img_path)
     books_path = []
     for i in range(len(result)):
-        save_path = os.path.join(dirname, img_name + '_book_' + str(i) + '.' + ext)
+        save_path = os.path.join(dirname, f'{img_name}.book_{i}.{img_ext}')
         cv2.imwrite(save_path, result[i])
         books_path.append(save_path)
     return books_path
diff --git a/services/paddle_services/dewarp.py b/services/paddle_services/dewarp.py
index 53b5293..5979fe4 100644
--- a/services/paddle_services/dewarp.py
+++ b/services/paddle_services/dewarp.py
@@ -17,8 +17,8 @@ def main():
     img_path = request.form.get('img_path')
     img = cv2.imread(img_path)
     dewarped_img = dewarper.dewarp_image(img)
-    dirname, img_name, ext = parse_img_path(img_path)
-    save_path = os.path.join(dirname, img_name + '_dewarped.' + ext)
+    dirname, img_name, img_ext = parse_img_path(img_path)
+    save_path = os.path.join(dirname, f'{img_name}.dewarped.{img_ext}')
     cv2.imwrite(save_path, dewarped_img)
     return save_path