From caba06c328fd29567a09c4ebd66d610750284422 Mon Sep 17 00:00:00 2001 From: irexyc Date: Thu, 21 Sep 2023 15:20:42 +0800 Subject: [PATCH] collect bbox --- mmdeploy/codebase/mmpose/deploy/pose_detection.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mmdeploy/codebase/mmpose/deploy/pose_detection.py b/mmdeploy/codebase/mmpose/deploy/pose_detection.py index 8cb211ba79..f8cddc7d2f 100644 --- a/mmdeploy/codebase/mmpose/deploy/pose_detection.py +++ b/mmdeploy/codebase/mmpose/deploy/pose_detection.py @@ -78,7 +78,7 @@ def process_model_config( keys=['img'], meta_keys=[ 'ori_shape', 'img_shape', 'pad_shape', 'scale_factor', - 'flip_indices' + 'flip_indices', 'bbox' ])) else: test_pipeline.append( @@ -94,7 +94,7 @@ def process_model_config( keys=['img'], meta_keys=[ 'img_shape', 'pad_shape', 'ori_shape', 'img_norm_cfg', - 'scale_factor', 'bbox_score', 'center', 'scale' + 'scale_factor', 'bbox_score', 'center', 'scale', 'bbox' ])) cfg.test_dataloader.dataset.pipeline = test_pipeline