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

12
apps/ai/main.py
View File

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