Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Update warning emojis (#8678)

Browse Source
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-07-22 15:23:22 +02:00
1 parent 38721de7ef
commit b17629e54f
1 file changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files
utils/general.py
+2 -2
View File
@@ -474,7 +474,7 @@ def check_dataset(data, autodownload=True):
for k in 'train', 'val', 'nc': for k in 'train', 'val', 'nc':
assert k in data, emojis(f"data.yaml '{k}:' field missing ❌") assert k in data, emojis(f"data.yaml '{k}:' field missing ❌")
if 'names' not in data: if 'names' not in data:
LOGGER.warning(emojis("data.yaml 'names:' field missing ⚠, assigning default names 'class0', 'class1', etc.")) LOGGER.warning(emojis("data.yaml 'names:' field missing ⚠️, assigning default names 'class0', 'class1', etc."))
data['names'] = [f'class{i}' for i in range(data['nc'])] # default names data['names'] = [f'class{i}' for i in range(data['nc'])] # default names
# Resolve paths # Resolve paths
@@ -490,7 +490,7 @@ def check_dataset(data, autodownload=True):
if val: if val:
val = [Path(x).resolve() for x in (val if isinstance(val, list) else [val])] # val path val = [Path(x).resolve() for x in (val if isinstance(val, list) else [val])] # val path
if not all(x.exists() for x in val): if not all(x.exists() for x in val):
LOGGER.info(emojis('\nDataset not found ⚠, missing paths %s' % [str(x) for x in val if not x.exists()])) LOGGER.info(emojis('\nDataset not found ⚠️, missing paths %s' % [str(x) for x in val if not x.exists()]))
if not s or not autodownload: if not s or not autodownload:
raise Exception(emojis('Dataset not found ❌')) raise Exception(emojis('Dataset not found ❌'))
t = time.time() t = time.time()