增加异常处理
This commit is contained in:
@@ -98,9 +98,13 @@ def merge_result(result1, result2):
|
|||||||
|
|
||||||
# 获取图片旋转角度
|
# 获取图片旋转角度
|
||||||
def get_image_rotation_angle(img):
|
def get_image_rotation_angle(img):
|
||||||
|
angle = 0
|
||||||
model = paddleclas.PaddleClas(model_name="text_image_orientation")
|
model = paddleclas.PaddleClas(model_name="text_image_orientation")
|
||||||
result = model.predict(input_data=img)
|
result = model.predict(input_data=img)
|
||||||
|
try:
|
||||||
angle = int(next(result)[0]["label_names"][0])
|
angle = int(next(result)[0]["label_names"][0])
|
||||||
|
except Exception as e:
|
||||||
|
logging.error("获取图片旋转角度失败", e)
|
||||||
return angle
|
return angle
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user