mmdet bug

vernitgarg · Jan 21, 2020 · ee4cd7e · ee4cd7e
1 parent fb45201
commit ee4cd7e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/mmskeleton/apis/estimation.py b/mmskeleton/apis/estimation.py
@@ -2,11 +2,13 @@
 import numpy as np
 import torch
 import mmcv
-import mmdet
+
 from mmskeleton.datasets.utils.video_demo import VideoDemo
 from mmskeleton.utils import get_mmskeleton_url
 from mmskeleton.processor.apis import init_twodimestimator, inference_twodimestimator
 
+import mmdet.apis
+
 
 def init_pose_estimator(detection_cfg, estimation_cfg, device=None):
 

diff --git a/mmskeleton/utils/third_party.py b/mmskeleton/utils/third_party.py
@@ -4,6 +4,7 @@
 COCO = lazy_import.lazy_module("pycocotools.COCO")
 COCOeval = lazy_import.lazy_module("pycocotools.COCOeval")
 mmdet = lazy_import.lazy_module("mmdet")
+lazy_import.lazy_module("mmdet.apis")
 
 
 def is_exist(module_name):