| 316 | |
| 317 | |
| 318 | def remove_duplicate_stracks(stracksa, stracksb): |
| 319 | pdist = matching.iou_distance(stracksa, stracksb) |
| 320 | pairs = np.where(pdist < 0.15) |
| 321 | dupa, dupb = list(), list() |
| 322 | for p, q in zip(*pairs): |
| 323 | timep = stracksa[p].frame_id - stracksa[p].start_frame |
| 324 | timeq = stracksb[q].frame_id - stracksb[q].start_frame |
| 325 | if timep > timeq: |
| 326 | dupb.append(q) |
| 327 | else: |
| 328 | dupa.append(p) |
| 329 | resa = [t for i, t in enumerate(stracksa) if not i in dupa] |
| 330 | resb = [t for i, t in enumerate(stracksb) if not i in dupb] |
| 331 | return resa, resb |
| 332 | |
| 333 | |
| 334 | def remove_fp_stracks(stracksa, n_frame=10): |