From d303e83c9166b0a9ee616114d4242deffb3a94a4 Mon Sep 17 00:00:00 2001 From: Yang Nie Date: Mon, 8 May 2023 15:11:00 +0800 Subject: [PATCH] fix infer bug --- ppdet/data/source/category.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ppdet/data/source/category.py b/ppdet/data/source/category.py index 4da25a2d2f5..1358df489d7 100644 --- 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':