nms

oliviermoliner · Jan 20, 2020 · 2d9f439 · 2d9f439
1 parent 0e2ec43
commit 2d9f439
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mmskeleton/datasets/coco.py b/mmskeleton/datasets/coco.py
@@ -20,8 +20,10 @@
 import numpy as np
 
 from .estimation import EstiamtionDataset
-oks_nms = lazy_import.lazy_module("mmskeleton.ops.nms.nms.oks_nms")
-soft_oks_nms = lazy_import.lazy_module("mmskeleton.ops.nms.nms.soft_oks_nms")
+# oks_nms = lazy_import.lazy_module("mmskeleton.ops.nms.nms.oks_nms")
+# soft_oks_nms = lazy_import.lazy_module("mmskeleton.ops.nms.nms.soft_oks_nms")
+from ..ops.nms.nms import oks_nms
+from ..ops.nms.nms import soft_oks_nms
 
 logger = logging.getLogger(__name__)