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:
Ayush Chaurasia
2023-01-10 17:03:28 +05:30
committed by GitHub
parent 0a7d86cd26
commit 341ce0ff3f
28 changed files with 383 additions and 6 deletions

View File

@ -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