fix infer bug

PaddlePaddle · May 8, 2023 · d303e83 · d303e83
1 parent f8c2a8b
commit d303e83
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ppdet/data/source/category.py b/ppdet/data/source/category.py
@@ -114,8 +114,9 @@ def get_categories(metric_type, anno_file=None, arch=None):
     elif metric_type.lower() == 'widerface':
         return _widerface_category()
 
-    elif metric_type.lower() == 'keypointtopdowncocoeval' or metric_type.lower(
-    ) == 'keypointtopdownmpiieval':
+    elif metric_type.lower() in ['keypointtopdowncocoeval',
+                                 'keypointtopdownmpiieval',
+                                 'keypointtopdowncocowholebadyhandeval']:
         return (None, {'id': 'keypoint'})
 
     elif metric_type.lower() == 'pose3deval':