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

Add YOLOv5 dataset yamls (#207)

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-10 18:33:44 +01:00 • committed by GitHub
parent e371e81aa0
commit c7629e93bd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
14 changed files with 2018 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/nn/autobackend.py
View file

@ -227,7 +227,7 @@ class AutoBackend(nn.Module):
if 'names' not in locals():
names = yaml_load(data)['names'] if data else {i: f'class{i}' for i in range(999)}
if names[0] == 'n01440764' and len(names) == 1000: # ImageNet
names = yaml_load(ROOT / 'data/ImageNet.yaml')['names'] # human-readable names
names = yaml_load(ROOT / 'yolo/data/datasets/ImageNet.yaml')['names'] # human-readable names
self.__dict__.update(locals()) # assign all variables to self