| 352 | return list(stracks.values()) |
| 353 | |
| 354 | def remove_duplicate_stracks(stracksa, stracksb): |
| 355 | pdist = matching.iou_distance(stracksa, stracksb) |
| 356 | pairs = np.where(pdist<0.15) |
| 357 | dupa, dupb = list(), list() |
| 358 | for p,q in zip(*pairs): |
| 359 | timep = stracksa[p].frame_id - stracksa[p].start_frame |
| 360 | timeq = stracksb[q].frame_id - stracksb[q].start_frame |
| 361 | if timep > timeq: |
| 362 | dupb.append(q) |
| 363 | else: |
| 364 | dupa.append(p) |
| 365 | resa = [t for i,t in enumerate(stracksa) if not i in dupa] |
| 366 | resb = [t for i,t in enumerate(stracksb) if not i in dupb] |
| 367 | return resa, resb |
| 368 | |
| 369 | |