Ссылка для ultralytics/hub/__init__.py
Примечание
Этот файл доступен по адресу https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/hub/init.py. Если ты обнаружил проблему, пожалуйста, помоги исправить ее, создав Pull Request 🛠️. Спасибо 🙏!
ultralytics.hub.login(api_key=None, save=True)
Войди в API Ultralytics HUB, используя предоставленный ключ API.
Сессия не сохраняется; новая сессия создается при необходимости с использованием сохраненных SETTINGS или переменной окружения HUB_API_KEY переменной окружения при успешной аутентификации.
Параметры:
Имя | Тип | Описание | По умолчанию |
---|---|---|---|
api_key |
str
|
Ключ API, который будет использоваться для аутентификации. Если он не указан, то будет получен из переменной окружения SETTINGS или HUB_API_KEY. |
None
|
save |
bool
|
Сохранять ли API-ключ в SETTINGS, если аутентификация прошла успешно. |
True
|
Возвращается:
Тип | Описание |
---|---|
bool
|
True, если аутентификация прошла успешно, False в противном случае. |
Исходный код в ultralytics/hub/__init__.py
ultralytics.hub.logout()
Выйди из Ultralytics HUB, удалив API-ключ из файла настроек. Чтобы войти снова, используй 'yolo hub login'.
Исходный код в 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)
Функция для проверки Zip-файла набора данных HUB на наличие ошибок перед загрузкой. Она проверяет набор данных на наличие ошибок перед его загрузкой в HUB. Примеры использования приведены ниже.
Параметры:
Имя | Тип | Описание | По умолчанию |
---|---|---|---|
path |
str
|
Путь к data.zip (с data.yaml внутри data.zip). |
требуется |
task |
str
|
Задание для набора данных. Варианты: "обнаружить", "сегментировать", "позировать", "классифицировать", "obb". |
требуется |
Пример
Загрузи файлы *.zip с сайта https://github.com/ultralytics/hub/tree/main/example_datasets. Например, 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