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

算法识别bug

This commit is contained in:
曹前明 2022-10-26 21:47:49 +08:00
parent 2549adc08a
commit 5b32b838eb
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/ai/main.py
View File

@ -38,11 +38,11 @@ def ai_analyse(codes: list, global_img: str, face_img: str = None):
else: else:
is_happend, res = f(ip=settings.AI_IP, pic_url=global_img) is_happend, res = f(ip=settings.AI_IP, pic_url=global_img)
if i in ['fire1', 'jingjiedai', 'qiping']: # 如果是这3类算法就无需再识别，都在一个算法里处理了 if i in ['fire1', 'jingjiedai', 'qiping']: # 如果是这3类算法就无需再识别，都在一个算法里处理了
for x in res.FireinfoList:
has_fire = False # 默认没有灭火器 has_fire = False # 默认没有灭火器
has_jingjiedai = False # 默认没有警戒带 has_jingjiedai = False # 默认没有警戒带
qiping_qd = False # 气瓶倾倒未发生 qiping_qd = False # 气瓶倾倒未发生
rectangle_dict .update({'qiping': []}) # 气瓶倾倒的坐标字典 rectangle_dict .update({'qiping': []}) # 气瓶倾倒的坐标字典
for x in res.FireinfoList:
if x.fire == 3 and 'qiping' in codes: # 气瓶倾倒 if x.fire == 3 and 'qiping' in codes: # 气瓶倾倒
qiping_qd = True qiping_qd = True
rectangle_dict['qiping'].append([x.coord.x, x.coord.y]) # 加入矩形框 rectangle_dict['qiping'].append([x.coord.x, x.coord.y]) # 加入矩形框