Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Raise error on suffix-less model path (#8561)

Browse Source 
Raise error on invalid model
This commit is contained in:
Colin Wong authored and GitHub committed 2022-07-13 11:13:01 +02:00
1 parent 574ceedfc5
commit f8722b4429
1 file changed
+2
Download Patch File Download Diff File Expand all files Collapse all files
models/common.py
+2
View File
@@ -441,6 +441,8 @@ class DetectMultiBackend(nn.Module):
output_details = interpreter.get_output_details() # outputs output_details = interpreter.get_output_details() # outputs
elif tfjs: elif tfjs:
raise Exception('ERROR: YOLOv5 TF.js inference is not supported') 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 self.__dict__.update(locals()) # assign all variables to self
def forward(self, im, augment=False, visualize=False, val=False): def forward(self, im, augment=False, visualize=False, val=False):