Raise error on suffix-less model path (#8561)
Raise error on invalid model
This commit is contained in:
1 parent
574ceedfc5
commit
f8722b4429
1 file changed
+2
@@ -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):
|
||||
|
||||
Reference in new issue
Block a user