diff --git a/utils/general.py b/utils/general.py index c2fcf5e9c7f9..02bc741ca3ba 100755 --- a/utils/general.py +++ b/utils/general.py @@ -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')):