diff --git a/extras/easypose/pose.py b/extras/easypose/pose.py index 10d4086..1d37888 100644 --- a/extras/easypose/pose.py +++ b/extras/easypose/pose.py @@ -50,14 +50,8 @@ class SimCC(BaseModel): return tensor def postprocess(self, tensor: List[np.ndarray]): - simcc_x, simcc_y = tensor - simcc_x = np.squeeze(simcc_x, axis=0) - simcc_y = np.squeeze(simcc_y, axis=0) - keypoints = simcc_decoder(simcc_x, - simcc_y, - self.input_shape[2:], - self.dx, - self.dy, - self.scale) + kpts = tensor[0][0] + scores = np.expand_dims(tensor[1][0], axis=-1) + keypoints = np.concatenate([kpts, scores], axis=-1) return keypoints diff --git a/extras/mmdeploy/configs/pose-detection_simcc_onnxruntime_static-384x288.py b/extras/mmdeploy/configs/pose-detection_simcc_onnxruntime_static-384x288.py index 9d41100..bfa43b4 100644 --- a/extras/mmdeploy/configs/pose-detection_simcc_onnxruntime_static-384x288.py +++ b/extras/mmdeploy/configs/pose-detection_simcc_onnxruntime_static-384x288.py @@ -2,7 +2,7 @@ _base_ = ["./pose-detection_static.py", "../_base_/backends/onnxruntime.py"] onnx_config = dict( input_shape=[288, 384], - output_names=["simcc_x", "simcc_y"], + output_names=["kpts", "scores"], ) -codebase_config = dict(export_postprocess=False) # do not export get_simcc_maximum +codebase_config = dict(export_postprocess=True) # export get_simcc_maximum diff --git a/extras/mmdeploy/configs/pose-detection_simcc_onnxruntime_static-384x288_fp16.py b/extras/mmdeploy/configs/pose-detection_simcc_onnxruntime_static-384x288_fp16.py index 1fee327..6263dac 100644 --- a/extras/mmdeploy/configs/pose-detection_simcc_onnxruntime_static-384x288_fp16.py +++ b/extras/mmdeploy/configs/pose-detection_simcc_onnxruntime_static-384x288_fp16.py @@ -2,7 +2,7 @@ _base_ = ["./pose-detection_static.py", "../_base_/backends/onnxruntime-fp16.py" onnx_config = dict( input_shape=[288, 384], - output_names=["simcc_x", "simcc_y"], + output_names=["kpts", "scores"], ) -codebase_config = dict(export_postprocess=False) # do not export get_simcc_maximum +codebase_config = dict(export_postprocess=True) # export get_simcc_maximum