优化代码

Object/SageMakerAiObject.py

@@ -16,7 +16,7 @@ from io import BytesIO
 import boto3
 import cv2
 import numpy as np
-from PIL import Image
+from PIL import Image, UnidentifiedImageError
 
 from AnsjerPush.Config.aiConfig import AI_IDENTIFICATION_TAGS_DICT
 from AnsjerPush.config import AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, CONFIG_US, CONFIG_EUR
@@ -78,10 +78,14 @@ class SageMakerAiObject:
                 return False
 
             return results
+        except UnidentifiedImageError as e:
+            LOGGER.info('***sagemakerUnidentifiedImageError***uid={},errLine={errLine}, errMsg={errMsg}'
+                        .format(uid, errLine=e.__traceback__.tb_lineno, errMsg=repr(e)))
+            return 'imageError'
         except Exception as e:
-            LOGGER.info('***sage_maker_ai_server***uid={},errLine={errLine}, errMsg={errMsg}'
+            LOGGER.info('***sagemakerException***uid={},errLine={errLine}, errMsg={errMsg}'
                         .format(uid, errLine=e.__traceback__.tb_lineno, errMsg=repr(e)))
-            return False
+            return 'imageError'
 
     @staticmethod
     def get_table_name(uid, ai_result, detect_group):