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

CoreML NMS and half fixes (#143)

Browse source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Glenn Jocher 2023-01-05 00:20:54 +01:00 • committed by GitHub
parent 55bdca6768
commit 172cef2d20
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
18 changed files with 95 additions and 414 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/yolo/data/utils.py
View file

@ -201,7 +201,7 @@ def check_dataset_yaml(data, autodownload=True):
extract_dir, autodownload = data.parent, False
# Read yaml (optional)
if isinstance(data, (str, Path)):
data = yaml_load(data) # dictionary
data = yaml_load(data, append_filename=True) # dictionary
# Checks
for k in 'train', 'val', 'names':