Fix mapping error from mmpose's regression codec to SDK's components (#1684)

This commit is contained in:
Chen Xin 2023-01-31 15:58:04 +08:00 committed by GitHub
parent 2ceabd0274
commit a19cf64765
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 0 deletions

View File

@ -346,6 +346,10 @@ class PoseDetection(BaseTask):
params['modulate_kernel'] = self.model_cfg.kernel_sizes[-1]
elif codec.type == 'SimCCLabel':
component = 'SimCCLabelDecode'
elif codec.type == 'RegressionLabel':
component = 'DeepposeRegressionHeadDecode'
elif codec.type == 'IntegralRegressionLabel':
component = 'DeepposeRegressionHeadDecode'
else:
raise RuntimeError(f'Unsupported codecs type: {codec.type}')
postprocess = dict(params=params, type=component)

View File

@ -103,6 +103,8 @@ class End2EndModel(BaseBackendModel):
if codec.type == 'SimCCLabel':
batch_pred_x, batch_pred_y = batch_outputs
preds = self.head.decode((batch_pred_x, batch_pred_y))
elif codec.type in ['RegressionLabel', 'IntegralRegressionLabel']:
preds = self.head.decode(batch_outputs)
else:
preds = self.head.decode(batch_outputs[0])
results = self.pack_result(preds, data_samples)