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