(tlista, tlistb)
| 345 | return torch.tensor(bboxes), torch.tensor(labels), torch.tensor(ids) |
| 346 | |
| 347 | def joint_stracks(tlista, tlistb): |
| 348 | exists = {} |
| 349 | res = [] |
| 350 | for t in tlista: |
| 351 | exists[t.track_id] = 1 |
| 352 | res.append(t) |
| 353 | for t in tlistb: |
| 354 | tid = t.track_id |
| 355 | if not exists.get(tid, 0): |
| 356 | exists[tid] = 1 |
| 357 | res.append(t) |
| 358 | return res |
| 359 | |
| 360 | |
| 361 | def sub_stracks(tlista, tlistb): |