From f8722b4429e80f96be04b36e4efd84ce6583bfa1 Mon Sep 17 00:00:00 2001
From: Colin Wong <cwongmath@gmail.com>
Date: Wed, 13 Jul 2022 04:13:01 -0500
Subject: [PATCH] Raise error on suffix-less model path (#8561)

Raise error on invalid model
---
 models/common.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/models/common.py b/models/common.py
index 61e94296b..fb5ac3a6f 100644
--- a/models/common.py
+++ b/models/common.py
@@ -441,6 +441,8 @@ class DetectMultiBackend(nn.Module):
                 output_details = interpreter.get_output_details()  # outputs
             elif tfjs:
                 raise Exception('ERROR: YOLOv5 TF.js inference is not supported')
+            else:
+                raise Exception(f'ERROR: {w} is not a supported format')
         self.__dict__.update(locals())  # assign all variables to self
 
     def forward(self, im, augment=False, visualize=False, val=False):