|
|
@@ -293,12 +293,14 @@ def check_imshow(): |
|
|
|
|
|
|
|
|
|
|
|
def check_suffix(file='yolov5s.pt', suffix=('.pt',), msg=''): |
|
|
|
# Check file(s) for acceptable suffixes |
|
|
|
# Check file(s) for acceptable suffix |
|
|
|
if file and suffix: |
|
|
|
if isinstance(suffix, str): |
|
|
|
suffix = [suffix] |
|
|
|
for f in file if isinstance(file, (list, tuple)) else [file]: |
|
|
|
assert Path(f).suffix.lower() in suffix, f"{msg}{f} acceptable suffix is {suffix}" |
|
|
|
s = Path(f).suffix.lower() # file suffix |
|
|
|
if len(s): |
|
|
|
assert s in suffix, f"{msg}{f} acceptable suffix is {suffix}" |
|
|
|
|
|
|
|
|
|
|
|
def check_yaml(file, suffix=('.yaml', '.yml')): |