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

算法img bug

This commit is contained in:
曹前明 2022-10-27 21:37:50 +08:00
parent 22412d0be1
commit 1dba032737
1 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
apps/ai/main.py
View File

@ -17,7 +17,7 @@ algo_dict = {
}
def ai_analyse(codes: list, global_img: str, face_img: str = None, is_dahua_pic: bool = True):
def ai_analyse(codes: list, global_img: str, face_img: str = '', is_dahua_pic: bool = True):
"""算法分析图片
Args:
@ -26,6 +26,8 @@ def ai_analyse(codes: list, global_img: str, face_img: str = None, is_dahua_pic:
face_img (str): 人脸图片url地址
"""
results = {} # dict key: 触发的事件标识字符 value: 多个矩形框坐标列表; 有两个图片key值
global_img_path = ''
face_img_path = ''
if is_dahua_pic: # 先保存到本地/主要是现在算法必须得用可访问的远程地址
from apps.ecm.service import save_dahua_pic
global_img_path = save_dahua_pic(global_img)
@ -55,7 +57,8 @@ def ai_analyse(codes: list, global_img: str, face_img: str = None, is_dahua_pic:
for x in res.FireinfoList:
if x.fire == 3 and 'qiping' in codes: # 气瓶倾倒
qiping_qd = True
rectangle_dict['qiping'].append([(x.coord.uleft.x, x.coord.uleft.y), (x.coord.lright.x, x.coord.lright.y)]) # 加入矩形框
rectangle_dict['qiping'].append(
[(x.coord.uleft.x, x.coord.uleft.y), (x.coord.lright.x, x.coord.lright.y)]) # 加入矩形框
if x.fire == 0:
has_fire = True
if x.fire == 2:
@ -66,14 +69,14 @@ def ai_analyse(codes: list, global_img: str, face_img: str = None, is_dahua_pic:
results.update({i: rectangle_dict.get(i, [])})
except Exception:
myLogger.error('算法处理错误', exc_info=True)
if is_dahua_pic:
if global_img_path:
os.remove(settings.BASE_DIR + global_img_path) # 删除临时图片
if face_img:
os.remove(settings.BASE_DIR + face_img_path)
if face_img_path:
os.remove(settings.BASE_DIR + face_img_path)
return results
def ai_analyse_2(codes: list, global_img: str, face_img: str = None, is_dahua_pic: bool = True):
def ai_analyse_2(codes: list, global_img: str, face_img: str = '', is_dahua_pic: bool = True):
"""算法分析图片后保存并返回相对地址
Args:
@ -82,14 +85,15 @@ def ai_analyse_2(codes: list, global_img: str, face_img: str = None, is_dahua_pi
face_img (str): 人脸图片url地址
"""
results = ai_analyse(codes, global_img, face_img, is_dahua_pic)
if results: # 如果触发事件先保存下来
if results: # 如果触发事件先保存下来
from apps.ecm.service import save_dahua_pic
global_img_path = save_dahua_pic(global_img)
global_img_local = settings.BASE_DIR + global_img_path
draw(global_img_local, results)
results.update({'global_img': global_img_path})
return results
def draw(path, results):
rects = []
for i in results:
@ -100,6 +104,5 @@ def draw(path, results):
for i in results:
for m in results[i]:
cv2.rectangle(img, results[i][m][0], results[i][m][1], (0, 255, 0), 4)
cv2.putText(img, i, results[i][m][0], font, 2, (0,0,255), 1)
cv2.putText(img, i, results[i][m][0], font, 2, (0, 0, 255), 1)
cv2.imwrite(path, img)