优化AI代码

Object/SageMakerAiObject.py

@@ -47,8 +47,7 @@ class SageMakerAiObject:
                 LOGGER.info('SageMake fail')
                 return False
 
-            LOGGER.info(f'uid={uid} base64_list={base64_list}')
-            for i in range(base64_list):
+            for i in range(len(base64_list)):
                 base64_list[i] = base64_list[i].replace(' ', '+')
 
             img_list = list(map(base64.b64decode, base64_list))
@@ -59,7 +58,7 @@ class SageMakerAiObject:
 
             nms_threshold = 0.45
             confidence = 0.5
-            client_timeout = 30
+            client_timeout = 3
 
             start = time.time()
             results = ai.yolo_infer(img_list, nms_threshold, confidence, client_timeout)

Object/enums/MessageTypeEnum.py

@@ -12,8 +12,8 @@ from enum import IntEnum, unique
 @unique
 class MessageTypeEnum(IntEnum):
     AI_HUMAN_SHAPE = 1  # ai人形
-    AI_CAT = 2  # ai车型
-    AI_PET = 3  # ai宠物
+    AI_PET = 2  # ai宠物
+    AI_VEHICLE = 3  # ai车型
     AI_PARCEL = 4  # ai包裹
     MOTION_DETECTION = 51  # 移动侦测
     SENSOR_ALARM = 52  # 传感器报警