Pārlūkot izejas kodu

Fix `check_suffix()` (#4712)

Fix a bug when `file=''`
modifyDataloader
Glenn Jocher GitHub pirms 3 gadiem
vecāks
revīzija
f984cce52a
Šim parakstam datu bāzē netika atrasta zināma atslēga GPG atslēgas ID: 4AEE18F83AFDEB23
1 mainītis faili ar 2 papildinājumiem un 2 dzēšanām
  1. +2
    -2
      utils/general.py

+ 2
- 2
utils/general.py Parādīt failu

@@ -244,7 +244,7 @@ def check_imshow():

def check_suffix(file='yolov5s.pt', suffix=('.pt',), msg=''):
# Check file(s) for acceptable suffixes
if any(suffix):
if file and suffix:
if isinstance(suffix, str):
suffix = [suffix]
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=''):
# Search/download file (if necessary) and return path
check_suffix(file, suffix)
check_suffix(file, suffix) # optional
file = str(file) # convert to str()
if Path(file).is_file() or file == '': # exists
return file

Notiek ielāde…
Atcelt
Saglabāt