From fd07b26c2e8f0e8e7aa986ada3629286478e96cb Mon Sep 17 00:00:00 2001 From: Qingren <45782082+Qingrenn@users.noreply.github.com> Date: Mon, 19 Sep 2022 14:13:55 +0800 Subject: [PATCH] [Feature] Add mmpose configs for compatibility with Hourglass model in ncnn backend (#1064) * [FEATURE] add mmpose configs for compatibility with Hourglass model [Motivation] - The added configs are aim to support Hourglass pose-detection model in ncnn backend. - Hourglass model needs explicit input shape with 256x256 or 384x384. - Hourglass model can be quantitized to int8 successfully in ncnn backend. [Complementary] The mmpose config of Hourglass mode is in: - https://github.com/open-mmlab/mmpose/blob/master/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hourglass52_coco_256x256.py - https://github.com/open-mmlab/mmpose/blob/master/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hourglass52_coco_384x384.py Quantitization results are shown in https://github.com/Qingrenn/mmdeploy-summer-camp/blob/main/week4/result.md * Keep only one config (pose-detection_ncnn-int8_static-256x256.py) and remove the others. --- configs/mmpose/pose-detection_ncnn-int8_static-256x256.py | 3 +++ 1 file changed, 3 insertions(+) create mode 100644 configs/mmpose/pose-detection_ncnn-int8_static-256x256.py diff --git a/configs/mmpose/pose-detection_ncnn-int8_static-256x256.py b/configs/mmpose/pose-detection_ncnn-int8_static-256x256.py new file mode 100644 index 000000000..9c5f5aab3 --- /dev/null +++ b/configs/mmpose/pose-detection_ncnn-int8_static-256x256.py @@ -0,0 +1,3 @@ +_base_ = ['./pose-detection_static.py', '../_base_/backends/ncnn-int8.py'] + +onnx_config = dict(input_shape=[256, 256])