์ฐธ์กฐ ultralytics/hub/__init__.py
์ฐธ๊ณ
์ด ํ์ผ์ https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/hub/init.py์์ ํ์ธํ ์ ์์ต๋๋ค. ๋ฌธ์ ๋ฅผ ๋ฐ๊ฒฌํ๋ฉด ํ ๋ฆฌํ์คํธ (๐ ๏ธ) ๋ฅผ ํตํด ๋ฌธ์ ๋ฅผ ํด๊ฒฐํ๋๋ก ๋์์ฃผ์ธ์. ๊ฐ์ฌํฉ๋๋ค ๐!
ultralytics.hub.login(api_key=None, save=True)
์ ๊ณต๋ API ํค๋ฅผ ์ฌ์ฉํ์ฌ Ultralytics HUB API์ ๋ก๊ทธ์ธํฉ๋๋ค.
์ธ์ ์ ์ ์ฅ๋์ง ์์ผ๋ฉฐ, ํ์ํ ๋ ์ ์ฅ๋ ์ค์ ๋๋ ์ธ์ฆ์ ์ฑ๊ณตํ๋ฉด HUB_API_KEY ํ๊ฒฝ ๋ณ์๋ฅผ ์ฌ์ฉํ์ฌ ์ ์ธ์ ์ด ์์ฑ๋ฉ๋๋ค.
๋งค๊ฐ๋ณ์:
์ด๋ฆ | ์ ํ | ์ค๋ช | ๊ธฐ๋ณธ๊ฐ |
---|---|---|---|
api_key |
str
|
์ธ์ฆ์ ์ฌ์ฉํ API ํค์ ๋๋ค. ์ ๊ณตํ์ง ์์ผ๋ฉด SETTINGS ๋๋ HUB_API_KEY ํ๊ฒฝ ๋ณ์์์ ๊ฒ์๋ฉ๋๋ค. |
None
|
save |
bool
|
์ธ์ฆ์ ์ฑ๊ณตํ ๊ฒฝ์ฐ API ํค๋ฅผ ์ค์ ์ ์ ์ฅํ ์ง ์ฌ๋ถ์ ๋๋ค. |
True
|
๋ฐํํฉ๋๋ค:
์ ํ | ์ค๋ช |
---|---|
bool
|
์ธ์ฆ์ ์ฑ๊ณตํ๋ฉด ์ฐธ์ด๊ณ , ๊ทธ๋ ์ง ์์ผ๋ฉด ๊ฑฐ์ง์ ๋๋ค. |
์ ์์ค ์ฝ๋ ultralytics/hub/__init__.py
ultralytics.hub.logout()
์ค์ ํ์ผ์์ API ํค๋ฅผ ์ ๊ฑฐํ์ฌ Ultralytics ํ๋ธ์์ ๋ก๊ทธ์์ํฉ๋๋ค. ๋ค์ ๋ก๊ทธ์ธํ๋ ค๋ฉด 'yolo ํ๋ธ ๋ก๊ทธ์ธ'์ ์ฌ์ฉํฉ๋๋ค.
์ ์์ค ์ฝ๋ ultralytics/hub/__init__.py
ultralytics.hub.reset_model(model_id='')
ํ์ต๋ ๋ชจ๋ธ์ ํ์ต๋์ง ์์ ์ํ๋ก ์ด๊ธฐํํฉ๋๋ค.
์ ์์ค ์ฝ๋ ultralytics/hub/__init__.py
ultralytics.hub.export_fmts_hub()
HUB์์ ์ง์ํ๋ ๋ด๋ณด๋ด๊ธฐ ํ์์ ๋ชฉ๋ก์ ๋ฐํํฉ๋๋ค.
ultralytics.hub.export_model(model_id='', format='torchscript')
๋ชจ๋ ํ์์ผ๋ก ๋ชจ๋ธ์ ๋ด๋ณด๋ ๋๋ค.
์ ์์ค ์ฝ๋ ultralytics/hub/__init__.py
ultralytics.hub.get_export(model_id='', format='torchscript')
๋ด๋ณด๋ธ ๋ชจ๋ธ ์ฌ์ ์ ๋ค์ด๋ก๋ URL๊ณผ ํจ๊ป ๊ฐ์ ธ์ต๋๋ค.
์ ์์ค ์ฝ๋ ultralytics/hub/__init__.py
ultralytics.hub.check_dataset(path, task)
์ ๋ก๋ ์ HUB ๋ฐ์ดํฐ์ธํธ Zip ํ์ผ์ ์ค๋ฅ๋ฅผ ๊ฒ์ฌํ๋ ๊ธฐ๋ฅ์ ๋๋ค. ๋ฐ์ดํฐ ์ธํธ๊ฐ HUB์ ์ ๋ก๋๋๊ธฐ ์ ์ ์ค๋ฅ๋ฅผ ํ์ธํฉ๋๋ค. ์ค๋ฅ๋ฅผ ํ์ธํฉ๋๋ค. ์ฌ์ฉ ์๋ ์๋์ ๊ฐ์ต๋๋ค.
๋งค๊ฐ๋ณ์:
์ด๋ฆ | ์ ํ | ์ค๋ช | ๊ธฐ๋ณธ๊ฐ |
---|---|---|---|
path |
str
|
data.zip ๊ฒฝ๋ก(data.zip ์์ data.yaml ํฌํจ). |
ํ์ |
task |
str
|
๋ฐ์ดํฐ ์ธํธ ์์ . ์ต์ ์ '๊ฐ์ง', '์ธ๊ทธ๋จผํธ', 'ํฌ์ฆ', '๋ถ๋ฅ', '์ค๋ธ์ ํธ'์ ๋๋ค. |
ํ์ |
์
https://github.com/ultralytics/hub/tree/main/example_datasets์์ *.zip ํ์ผ์ ๋ค์ด๋ก๋ํ์ธ์. ์: https://github.com/ultralytics/hub/raw/main/example_datasets/coco8.zip coco8.zip.
from ultralytics.hub import check_dataset
check_dataset('path/to/coco8.zip', task='detect') # detect dataset
check_dataset('path/to/coco8-seg.zip', task='segment') # segment dataset
check_dataset('path/to/coco8-pose.zip', task='pose') # pose dataset
check_dataset('path/to/dota8.zip', task='obb') # OBB dataset
check_dataset('path/to/imagenet10.zip', task='classify') # classification dataset