| 363 | |
| 364 | |
| 365 | def remove_duplicate_stracks(stracksa, stracksb): |
| 366 | pdist = matching.iou_distance(stracksa, stracksb) |
| 367 | pairs = np.where(pdist < 0.15) |
| 368 | dupa, dupb = list(), list() |
| 369 | for p, q in zip(*pairs): |
| 370 | timep = stracksa[p].frame_id - stracksa[p].start_frame |
| 371 | timeq = stracksb[q].frame_id - stracksb[q].start_frame |
| 372 | if timep > timeq: |
| 373 | dupb.append(q) |
| 374 | else: |
| 375 | dupa.append(p) |
| 376 | resa = [t for i, t in enumerate(stracksa) if not i in dupa] |
| 377 | resb = [t for i, t in enumerate(stracksb) if not i in dupb] |
| 378 | return resa, resb |
| 379 | |
| 380 | def vis_feature(frame_id,seq_num,img,track_features, det_features, cost_matrix, cost_matrix_det, cost_matrix_track,max_num=5, out_path='/home/XX/'): |
| 381 | num_zero = ["0000","000","00","0"] |