Fix: Fix wrong import of inference and export

Seeed-Studio · Jul 17, 2024 · be9162e · be9162e
1 parent 35a3862
commit be9162e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tools/export.py b/tools/export.py
@@ -483,11 +483,11 @@ def main():
     args, cfg = build_config(args)
 
     if 'runner_type' not in cfg:
-        from mmengine.runner import Runner
+        from sscma.engine import Runner
 
         runner = Runner.from_cfg(cfg)
     else:
-        from mmengine.registry import RUNNERS
+        from sscma.registry import RUNNERS
 
         runner = RUNNERS.build(cfg)
 

diff --git a/tools/inference.py b/tools/inference.py
@@ -280,11 +280,11 @@ def main():
     args, cfg = build_config(args)
 
     if 'runner_type' not in cfg:
-        from mmengine.runner import Runner
+        from sscma.engine import Runner
 
         runner = Runner.from_cfg(cfg)
     else:
-        from mmengine.registry import RUNNERS
+        from sscma.registry import RUNNERS
 
         runner = RUNNERS.build(cfg)