(self)
| 328 | class TrackerPostProcess(nn.Module): |
| 329 | """ This module converts the model's output into the format expected by the coco api""" |
| 330 | def __init__(self): |
| 331 | super().__init__() |
| 332 | |
| 333 | @torch.no_grad() |
| 334 | def forward(self, track_instances: Instances, target_size) -> Instances: |