tools

libin-star · Aug 16, 2022 · 4283fc6 · 4283fc6
1 parent 77bf2c0
commit 4283fc6
Show file tree

Hide file tree

Showing 9 changed files with 3 additions and 161 deletions.
diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md
diff --git a/Dockerfile b/Dockerfile
diff --git a/models/Models/research.py b/models/Models/research.py
@@ -302,7 +302,7 @@ def warmup(self, imgsz=(1, 3, 640, 640)):
     @staticmethod
     def model_type(p='path/to/model.pt'):
         # Return model type from model path, i.e. path='path/to/model.onnx' -> type=onnx
-        from export import export_formats
+        from tools.export import export_formats
         suffixes = list(export_formats().Suffix) + ['.xml']  # export suffixes
         check_suffix(p, suffixes)  # checks
         p = Path(p).name  # eliminate trailing separators

diff --git a/models/common.py b/models/common.py
@@ -485,7 +485,7 @@ def warmup(self, imgsz=(1, 3, 640, 640), half=False):
     @staticmethod
     def model_type(p='path/to/model.pt'):
         # Return model type from model path, i.e. path='path/to/model.onnx' -> type=onnx
-        from export import export_formats
+        from tools.export import export_formats
         suffixes = list(export_formats().Suffix) + ['.xml']  # export suffixes
         check_suffix(p, suffixes)  # checks
         p = Path(p).name  # eliminate trailing separators

diff --git a/export.py → tools/export.py b/export.py → tools/export.py
diff --git a/hubconf.py → tools/hubconf.py b/hubconf.py → tools/hubconf.py
diff --git a/setup.cfg → tools/setup.cfg b/setup.cfg → tools/setup.cfg
diff --git a/tutorial.ipynb → tools/tutorial.ipynb b/tutorial.ipynb → tools/tutorial.ipynb
diff --git a/utils/benchmarks.py b/utils/benchmarks.py
@@ -37,7 +37,7 @@
     sys.path.append(str(ROOT))  # add ROOT to PATH
 # ROOT = ROOT.relative_to(Path.cwd())  # relative
 
-import export
+import tools.export as export
 import val
 from utils import notebook_init
 from utils.general import LOGGER, print_args