参考资料 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 密钥保存到设置中。 |
True
|
返回:
类型 | 说明 |
---|---|
bool
|
如果验证成功则为 True,否则为 False。 |
源代码 ultralytics/hub/__init__.py
ultralytics.hub.logout()
从设置文件中删除 API 密钥,退出Ultralytics HUB。要重新登录,请使用 "yolo hub login"。
源代码 ultralytics/hub/__init__.py
ultralytics.hub.reset_model(model_id='')
将训练好的模型重置为未训练状态。
源代码 ultralytics/hub/__init__.py
ultralytics.hub.export_fmts_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
|
数据集任务。选项包括 "检测"、"分割"、"摆放"、"分类 "和 "obb"。 |
所需 |
示例
从 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