算法识别bug

2022-10-26 21:47:49 +08:00 · 2022-10-26 21:47:49 +08:00 · 5b32b838eb
parent 2549adc08a
commit 5b32b838eb
1 changed files with 6 additions and 6 deletions
--- a/apps/ai/main.py
+++ b/apps/ai/main.py
@ -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类算法就无需再识别，都在一个算法里处理了
-                    for x in res.FireinfoList:
                    has_fire = False  # 默认没有灭火器
                    has_jingjiedai = False  # 默认没有警戒带
                    qiping_qd = False  # 气瓶倾倒未发生
                    rectangle_dict .update({'qiping': []})  # 气瓶倾倒的坐标字典
+                    for x in res.FireinfoList:
                        if x.fire == 3 and 'qiping' in codes:  # 气瓶倾倒
                            qiping_qd = True
                            rectangle_dict['qiping'].append([x.coord.x, x.coord.y])  # 加入矩形框