From e844abe21b7ee267a6a90356f3765321c1b9e712 Mon Sep 17 00:00:00 2001
From: kstheme <342200758@qq.com>
Date: Thu, 14 Mar 2024 18:25:19 +0800
Subject: [PATCH] Resolve the error issue when dp_type == PoseDataPreprocessor
 in td-hm_hrnet-w48_udp-8xb32-210e_coco-256x192.py.

---
 mmdeploy/codebase/mmpose/deploy/pose_detection_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mmdeploy/codebase/mmpose/deploy/pose_detection_model.py b/mmdeploy/codebase/mmpose/deploy/pose_detection_model.py
index a2ec9a21ad..98c6c79207 100644
--- a/mmdeploy/codebase/mmpose/deploy/pose_detection_model.py
+++ b/mmdeploy/codebase/mmpose/deploy/pose_detection_model.py
@@ -304,7 +304,7 @@ def build_pose_detection_model(
             from mmdet.models.data_preprocessors import DetDataPreprocessor
             data_preprocessor = DetDataPreprocessor(**dp)
         else:
-            assert dp_type == 'PoseDataPreprocessor'
+            assert dp_type == 'PoseDataPreprocessor' or dp_type == PoseDataPreprocessor
             data_preprocessor = PoseDataPreprocessor(**dp)
 
     backend_pose_model = __BACKEND_MODEL.build(