Raise error on invalid model

ultralytics · Jul 12, 2022 · a3375ac · a3375ac
1 parent 574ceed
commit a3375ac
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/models/common.py b/models/common.py
@@ -441,6 +441,8 @@ def wrap_frozen_graph(gd, inputs, outputs):
                 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):