mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
Fix mapping error from mmpose's regression codec to SDK's components (#1684)
This commit is contained in:
parent
2ceabd0274
commit
a19cf64765
@ -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)
|
||||
|
@ -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)
|
||||
|
Loading…
x
Reference in New Issue
Block a user