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