增加异常处理
This commit is contained in:
@@ -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)
|
||||
try:
|
||||
angle = int(next(result)[0]["label_names"][0])
|
||||
except Exception as e:
|
||||
logging.error("获取图片旋转角度失败", e)
|
||||
return angle
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user