Référence pour ultralytics/hub/__init__.py
Note
Ce fichier est disponible à l'adresse https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/hub/init.py. Si tu repères un problème, aide à le corriger en contribuant à une Pull Request 🛠️. Merci 🙏 !
ultralytics.hub.login(api_key=None, save=True)
Connecte-toi à l'API Ultralytics HUB à l'aide de la clé API fournie.
La session n'est pas stockée ; une nouvelle session est créée au besoin en utilisant les SETTINGS enregistrés ou la variable d'environnement HUB_API_KEY si l'authentification est réussie.
Paramètres :
Nom | Type | Description | Défaut |
---|---|---|---|
api_key |
str
|
Clé API à utiliser pour l'authentification. Si elle n'est pas fournie, elle sera récupérée à partir de SETTINGS ou de la variable d'environnement HUB_API_KEY. |
None
|
save |
bool
|
Enregistre ou non la clé API dans les PARAMÈTRES si l'authentification est réussie. |
True
|
Retourne :
Type | Description |
---|---|
bool
|
Vrai si l'authentification est réussie, Faux sinon. |
Code source dans ultralytics/hub/__init__.py
ultralytics.hub.logout()
Déconnecte-toi de Ultralytics HUB en supprimant la clé API du fichier de configuration. Pour te reconnecter, utilise 'yolo hub login'.
Code source dans ultralytics/hub/__init__.py
ultralytics.hub.reset_model(model_id='')
Réinitialise un modèle entraîné à un état non entraîné.
Code source dans ultralytics/hub/__init__.py
ultralytics.hub.export_fmts_hub()
Renvoie une liste des formats d'exportation pris en charge par HUB.
ultralytics.hub.export_model(model_id='', format='torchscript')
Exporte un modèle vers tous les formats.
Code source dans ultralytics/hub/__init__.py
ultralytics.hub.get_export(model_id='', format='torchscript')
Obtenir un dictionnaire de modèles exporté avec l'URL de téléchargement.
Code source dans ultralytics/hub/__init__.py
ultralytics.hub.check_dataset(path, task)
Fonction permettant de vérifier les erreurs dans le fichier Zip d'un jeu de données HUB avant de le télécharger. Elle vérifie qu'il n'y a pas d'erreurs dans un jeu de données avant qu'il ne soit téléchargé sur le HUB. vers le HUB. Des exemples d'utilisation sont donnés ci-dessous.
Paramètres :
Nom | Type | Description | Défaut |
---|---|---|---|
path |
str
|
Chemin d'accès à data.zip (avec data.yaml dans data.zip). |
requis |
task |
str
|
Tâche de l'ensemble de données. Les options sont "détecter", "segmenter", "poser", "classer", "obb". |
requis |
Exemple
Télécharge les fichiers *.zip à partir de https://github.com/ultralytics/hub/tree/main/example_datasets c'est-à-dire https://github.com/ultralytics/hub/raw/main/example_datasets/coco8.zip pour 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