mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix `check_suffix()` (#4712) 

Fix a bug when `file=''`
pull/4713/head
Glenn Jocher 2021-09-08 15:06:31 +02:00 committed by GitHub
parent a2b3c71636
commit f984cce52a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/general.py
View File

@ -244,7 +244,7 @@ def check_imshow():
def check_suffix(file='yolov5s.pt', suffix=('.pt',), msg=''): def check_suffix(file='yolov5s.pt', suffix=('.pt',), msg=''):
# Check file(s) for acceptable suffixes # Check file(s) for acceptable suffixes
if any(suffix): if file and suffix:
if isinstance(suffix, str): if isinstance(suffix, str):
suffix = [suffix] suffix = [suffix]
for f in file if isinstance(file, (list, tuple)) else [file]: for f in file if isinstance(file, (list, tuple)) else [file]:
@ -258,7 +258,7 @@ def check_yaml(file, suffix=('.yaml', '.yml')):
def check_file(file, suffix=''): def check_file(file, suffix=''):
# Search/download file (if necessary) and return path # Search/download file (if necessary) and return path
check_suffix(file, suffix) check_suffix(file, suffix) # optional
file = str(file) # convert to str() file = str(file) # convert to str()
if Path(file).is_file() or file == '': # exists if Path(file).is_file() or file == '': # exists
return file return file