diff --git a/segutils/trafficUtils.py b/segutils/trafficUtils.py index 4495431..fa5c648 100644 --- a/segutils/trafficUtils.py +++ b/segutils/trafficUtils.py @@ -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 )