HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Add YOLOv3 and YOLOv5 model *.yaml files (#169)

Browse source 
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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
28 changed files with 383 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ultralytics/yolo/utils/checks.py
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