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

ultralytics 8.0.166 expand HUBDatasetStats to Classify tasks (#4635)

Browse source
This commit is contained in:
Glenn Jocher 2023-08-29 16:17:59 +02:00 • committed by GitHub
parent 896da0c0a0
commit 52afadcb97
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 67 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/test_python.py
View file

@ -277,7 +277,7 @@ def test_data_utils():
# from ultralytics.utils.files import WorkingDirectory
# with WorkingDirectory(ROOT.parent / 'tests'):
for task in 'detect', 'segment', 'pose':
for task in 'detect', 'segment', 'pose', 'classify':
file = Path(TASK2DATA[task]).with_suffix('.zip') # i.e. coco8.zip
download(f'https://github.com/ultralytics/hub/raw/main/example_datasets/{file}', unzip=False, dir=TMP)
stats = HUBDatasetStats(TMP / file, task=task)