From 115a454637969e29b13ea0aea553022b439d93f2 Mon Sep 17 00:00:00 2001
From: Drago <DragoHG@users.noreply.github.com>
Date: Tue, 6 Sep 2022 15:37:44 -0300
Subject: [PATCH] Fix import issue with YoloV5 export models (#30)

---
 models/common.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/common.py b/models/common.py
index 7ac3a4a..9ac7e0f 100644
--- a/models/common.py
+++ b/models/common.py
@@ -539,7 +539,7 @@ class DetectMultiBackend(nn.Module):
     @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 yolov5.export import export_formats
         suffixes = list(export_formats().Suffix) + ['.xml']  # export suffixes
         check_suffix(p, suffixes)  # checks
         p = Path(p).name  # eliminate trailing separators