(tlista, tlistb)
| 301 | return torch.tensor(bboxes), torch.tensor(labels), torch.tensor(ids) |
| 302 | |
| 303 | def joint_stracks(tlista, tlistb): |
| 304 | exists = {} |
| 305 | res = [] |
| 306 | for t in tlista: |
| 307 | exists[t.track_id] = 1 |
| 308 | res.append(t) |
| 309 | for t in tlistb: |
| 310 | tid = t.track_id |
| 311 | if not exists.get(tid, 0): |
| 312 | exists[tid] = 1 |
| 313 | res.append(t) |
| 314 | return res |
| 315 | |
| 316 | |
| 317 | def sub_stracks(tlista, tlistb): |