(
self,
num_select=100,
nms_iou_threshold=-1,
num_body_points=17,
body_model= dict(
type='smplx',
keypoint_src='smplx',
num_expression_coeffs=10,
keypoint_dst='smplx_137',
model_path='data/body_models/smplx',
use_pca=False,
use_face_contour=True)
)
| 382 | class PostProcess_SMPLX(nn.Module): |
| 383 | """ This module converts the model's output into the format expected by the coco api""" |
| 384 | def __init__( |
| 385 | self, |
| 386 | num_select=100, |
| 387 | nms_iou_threshold=-1, |
| 388 | num_body_points=17, |
| 389 | body_model= dict( |
| 390 | type='smplx', |
| 391 | keypoint_src='smplx', |
| 392 | num_expression_coeffs=10, |
| 393 | keypoint_dst='smplx_137', |
| 394 | model_path='data/body_models/smplx', |
| 395 | use_pca=False, |
| 396 | use_face_contour=True) |
| 397 | ) -> None: |
| 398 | super().__init__() |
| 399 | self.num_select = num_select |
| 400 | self.nms_iou_threshold = nms_iou_threshold |
| 401 | self.num_body_points=num_body_points |
| 402 | self.body_model = build_body_model(body_model) |
| 403 | |
| 404 | @torch.no_grad() |
| 405 | def forward(self, outputs, target_sizes, targets, data_batch_nc, not_to_xyxy=False, test=False): |
nothing calls this directly
no test coverage detected