diff --git a/apps/ai/helmet/algo.py b/apps/ai/helmet/algo.py
index f9417428..a7a8c8eb 100644
--- a/apps/ai/helmet/algo.py
+++ b/apps/ai/helmet/algo.py
@@ -4,8 +4,8 @@ import base64
 from io import BytesIO
 import requests
 import grpc
-import BeiHangGrpc_pb2_grpc
-import BeiHangGrpc_pb2
+from . import BeiHangGrpc_pb2_grpc
+from . import BeiHangGrpc_pb2
 # from apps.utils.tools import convert_to_base64
 
 
diff --git a/apps/ecm/tasks.py b/apps/ecm/tasks.py
index d90c8468..4347adff 100644
--- a/apps/ecm/tasks.py
+++ b/apps/ecm/tasks.py
@@ -84,7 +84,7 @@ def snap_and_analyse(code: str, algo_codes: list, opl_id: str):
     ec_codes = ai_analyse(algo_codes, global_img=global_img_o)  # 算法处理返回的事件结果
     if ec_codes:
         # 获取本次所有发生事件种类
-        ecs = EventCate.objects.filter(code__in=ec_codes)
+        ecs = EventCate.objects.filter(code__in=ec_codes.keys())
         obj_cate = 'other'
         ep = None
         if 'helmet' in ec_codes: