def counting(trk, start, end, cnt, index, results, svm, frame):
flag = 0
if trk.is_counted:
return trk, cnt, results, flag
cur_center = trk.cur_center
pre_center = trk.pre_center
#cls = trk.cls
if pre_center == []:
return trk, cnt, results, flag
if Geometry.is_segment_cross(start, end, pre_center, cur_center):
cnt[index] += 1
flag = 1
if not trk.is_crossed_first_line:
trk.crossed_line[0] = index
trk.is_crossed_first_line = True
elif index != trk.crossed_line[0]:
trk.crossed_line[1] = index
trk.is_crossed_second_line = True
print(trk.cur_box)
top, bottom = min(trk.cur_box[1],
trk.cur_box[3]), max(trk.cur_box[1],
trk.cur_box[3])
left, right = min(trk.cur_box[0],
trk.cur_box[2]), max(trk.cur_box[0],
trk.cur_box[2])
img = frame[top:bottom, left:right]
cls = int(svm_vehicle.get_label(svm, img))
trk.cls = cls
print(cls)
results[cls - 1][trk.crossed_line[0]][trk.crossed_line[1]] += 1
trk.is_counted = True
return trk, cnt, results, flag
def counting(trk, start, end, cnt, index, results):
flag = 0
if trk.is_counted:
return trk, cnt, results, flag
cur_center = trk.cur_center
pre_center = trk.pre_center
cls = trk.cls
if pre_center == []:
return trk, cnt, results, flag
if Geometry.is_segment_cross(start, end, pre_center, cur_center):
flag = 1
cnt[index] += 1
if not trk.is_crossed_first_line:
trk.crossed_line[0] = index
trk.is_crossed_first_line = True
elif index != trk.crossed_line[0]:
trk.crossed_line[1] = index
trk.is_crossed_second_line = True
results[cls - 1][trk.crossed_line[0]][trk.crossed_line[1]] += 1
trk.is_counted = True
return trk, cnt, results, flag