Method
__init__
(self,
num_select=100,
nms_iou_threshold=-1,
num_body_points=17)
Source from the content-addressed store, hash-verified
| 297 | """This module converts the model's output into the format expected by the |
| 298 | coco api.""" |
| 299 | def __init__(self, |
| 300 | num_select=100, |
| 301 | nms_iou_threshold=-1, |
| 302 | num_body_points=17) -> None: |
| 303 | super().__init__() |
| 304 | self.num_select = num_select |
| 305 | self.nms_iou_threshold = nms_iou_threshold |
| 306 | self.num_body_points = num_body_points |
| 307 | |
| 308 | @torch.no_grad() |
| 309 | def forward(self, outputs, target_sizes, not_to_xyxy=False, test=False): |
Callers
nothing calls this directly
Tested by
no test coverage detected