zcdsj
/
factory
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

algo import bug

This commit is contained in:
曹前明 2022-08-26 16:53:03 +08:00
parent 47355e1e3a
commit 5fca56ccba
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/ai/helmet/algo.py
View File

@ -4,8 +4,8 @@ import base64
from io import BytesIO from io import BytesIO
import requests import requests
import grpc import grpc
import BeiHangGrpc_pb2_grpc from . import BeiHangGrpc_pb2_grpc
import BeiHangGrpc_pb2 from . import BeiHangGrpc_pb2
# from apps.utils.tools import convert_to_base64 # from apps.utils.tools import convert_to_base64

2
apps/ecm/tasks.py
View File

@ -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) # 算法处理返回的事件结果 ec_codes = ai_analyse(algo_codes, global_img=global_img_o) # 算法处理返回的事件结果
if ec_codes: if ec_codes:
# 获取本次所有发生事件种类 # 获取本次所有发生事件种类
ecs = EventCate.objects.filter(code__in=ec_codes) ecs = EventCate.objects.filter(code__in=ec_codes.keys())
obj_cate = 'other' obj_cate = 'other'
ep = None ep = None
if 'helmet' in ec_codes: if 'helmet' in ec_codes: