更新 segutils/trafficUtils.py

This commit is contained in:
zhoushuliang 2025-07-04 13:42:52 +08:00
parent bb4a79ff90
commit 3932b44d61
1 changed files with 1 additions and 1 deletions

View File

@ -466,7 +466,7 @@ def tracfficAccidentMixFunction(preds,seg_pred_mulcls,pars):
#b_0 = box[1:5];b_0.insert(0,box[0]);b_0.append(box[5] )
b_0 = box[0:4];b_0.insert(0,box[5]);b_0.append(box[4])
det_coords_original.append( box )
if int(box[5]) != pars['CarId'] or int(box[5]) != pars['CthcId']: continue
if int(box[5]) != pars['CarId'] and int(box[5]) != pars['CthcId']: continue
det_coords.append(b_0)
#print('##line957:',det_coords_original )