From 015cd03590c6da4fc41d217102541d45e22e37ed Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Tue, 25 Jun 2024 16:32:43 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=BC=82=E5=B8=B8=E5=A4=84?=
 =?UTF-8?q?=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 photo_review/photo_review.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/photo_review/photo_review.py b/photo_review/photo_review.py
index a42bbe0..92ff420 100644
--- a/photo_review/photo_review.py
+++ b/photo_review/photo_review.py
@@ -98,9 +98,13 @@ def merge_result(result1, result2):
 
 # 获取图片旋转角度
 def get_image_rotation_angle(img):
+    angle = 0
     model = paddleclas.PaddleClas(model_name="text_image_orientation")
     result = model.predict(input_data=img)
-    angle = int(next(result)[0]["label_names"][0])
+    try:
+        angle = int(next(result)[0]["label_names"][0])
+    except Exception as e:
+        logging.error("获取图片旋转角度失败", e)
     return angle