Add YOLOv3 and YOLOv5 model *.yaml files (#169)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
@ -217,8 +217,8 @@ def check_file(file, suffix=''):
|
||||
return file
|
||||
else: # search
|
||||
files = []
|
||||
for d in 'data', 'v8', 'utils': # search directories
|
||||
files.extend(glob.glob(str(ROOT / "yolo" / d / '**' / file), recursive=True)) # find file
|
||||
for d in 'models', 'yolo/data': # search directories
|
||||
files.extend(glob.glob(str(ROOT / d / '**' / file), recursive=True)) # find file
|
||||
assert len(files), f'File not found: {file}' # assert file was found
|
||||
assert len(files) == 1, f"Multiple files match '{file}', specify exact path: {files}" # assert unique
|
||||
return files[0] # return file
|
||||
|
Reference in New Issue
Block a user