From 6960593041e1d070dcb4c54d2723efa1b77524b8 Mon Sep 17 00:00:00 2001 From: caoqianming Date: Mon, 13 Feb 2023 17:16:32 +0800 Subject: [PATCH] =?UTF-8?q?=E7=BB=98=E5=88=B6=E7=9F=A9=E5=BD=A2=E6=A1=86bu?= =?UTF-8?q?g2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/ai/main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/apps/ai/main.py b/apps/ai/main.py index 0d11d03c..971c8ced 100644 --- a/apps/ai/main.py +++ b/apps/ai/main.py @@ -4,6 +4,7 @@ import os import cv2 from django.conf import settings import logging +import math myLogger = logging.getLogger('log') @@ -62,7 +63,7 @@ def ai_analyse(codes: list, global_img: str, face_img: str = '', is_dahua_pic: b if x.fire == 3: # 气瓶倾倒 qiping_qd = True rectangle_dict['qiping'].append( - [(x.coord.uleft.x, x.coord.uleft.y), (x.coord.lright.x, x.coord.lright.y)]) # 加入矩形框 + [(math.ceil(x.coord.uleft.x), math.ceil(x.coord.uleft.y)), (math.ceil(x.coord.lright.x), math.ceil(x.coord.lright.y))]) # 加入矩形框 if x.fire == 0: has_fire = True if x.fire == 2: