From 9c173439ebae548f99ee32d2e3efef22b26b4e35 Mon Sep 17 00:00:00 2001 From: Benyamin Jafari Date: Fri, 25 Nov 2022 23:30:31 +0330 Subject: [PATCH] Update coco.py --- mmskeleton/datasets/coco.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/mmskeleton/datasets/coco.py b/mmskeleton/datasets/coco.py index 147b4a6eb..c82fb5986 100644 --- a/mmskeleton/datasets/coco.py +++ b/mmskeleton/datasets/coco.py @@ -18,7 +18,8 @@ from .estimation import EstiamtionDataset from ..ops.nms.nms import oks_nms from ..ops.nms.nms import soft_oks_nms -from pycocotools import COCO, COCOeval +from pycocotools.coco import COCO +from pycocotools.cocoeval import COCOeval logger = logging.getLogger(__name__) @@ -420,4 +421,4 @@ def _do_python_keypoint_eval(self, res_file, res_folder): for ind, name in enumerate(stats_names): info_str.append((name, coco_eval.stats[ind])) - return info_str \ No newline at end of file + return info_str