调整图片名

2024-09-29 13:55:37 +08:00
parent 7a4cb5263a
commit a3fa1e502e
2 changed files with 4 additions and 4 deletions
--- 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