wangjin0928
/
AIlib2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

优化人群计数人数显示及边界设置

This commit is contained in:
th 2025-07-26 10:06:46 +08:00
parent 150e948fc2
commit 49f38b09ab
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
AI.py
View File

@ -764,10 +764,10 @@ def AI_process_Crowd(im0s,model,device,postPar):
if P == 0:
time_info = 'No Valid object find'
return [im0s[0], im0s[0], dets, 0], time_info
x1 = detP[:, 0] - 5
y1 = detP[:, 1] - 5
x2 = detP[:, 0] + 5
y2 = detP[:, 1] + 5
x1 = detP[:, 0] - p_postPar['expend']
y1 = detP[:, 1] - p_postPar['expend']
x2 = detP[:, 0] + p_postPar['expend']
y2 = detP[:, 1] + p_postPar['expend']
detP = np.column_stack((x1, y1, x2, y2))
detM = detM[:,:4]
a_x1, a_y1, a_x2, a_y2 = detM[:, 0], detM[:, 1], detM[:, 2], detM[:, 3]
@ -800,12 +800,11 @@ def AI_process_Crowd(im0s,model,device,postPar):
if len(p_indices) > 0:
detP = detP[p_indices]
detP = detP[:, :2]
detP[:, 0] = detP[:, 0] + 5
detP[:, 1] = detP[:, 1] + 5
dets= [[detM.tolist(), detP.tolist()]]
detP[:, 0] = detP[:, 0] + p_postPar['expend']
detP[:, 1] = detP[:, 1] + p_postPar['expend']
dets= [[detM.tolist(), detP.tolist(),p_postPar['psize']]]
else:
dets = [[detP.tolist()]]
dets = [[detP.tolist(),p_postPar['psize']]]
# for b in detM:
# b_label = '该建筑物下行人及数量：%d'%(int(b[4]))