Referência para ultralytics/hub/__init__.py
Nota
Este ficheiro está disponível em https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/hub/init.py. Se detectares um problema, por favor ajuda a corrigi-lo contribuindo com um Pull Request 🛠️. Obrigado 🙏!
ultralytics.hub.login(api_key=None, save=True)
Inicia sessão na API do Ultralytics HUB utilizando a chave de API fornecida.
A sessão não é armazenada; uma nova sessão é criada quando necessário utilizando as SETTINGS guardadas ou a variável de ambiente HUB_API_KEY se a autenticação for bem-sucedida.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
api_key |
str
|
Chave da API a utilizar para autenticação. Se não for fornecida, será obtida a partir da variável de ambiente SETTINGS ou HUB_API_KEY. |
None
|
save |
bool
|
Se queres guardar a chave da API em SETTINGS se a autenticação for bem sucedida. |
True
|
Devolve:
Tipo | Descrição |
---|---|
bool
|
Verdadeiro se a autenticação for bem sucedida, Falso caso contrário. |
Código fonte em ultralytics/hub/__init__.py
ultralytics.hub.logout()
Termina a sessão no Ultralytics HUB, removendo a chave API do ficheiro de definições. Para iniciar sessão novamente, utiliza 'yolo hub login'.
Código fonte em ultralytics/hub/__init__.py
ultralytics.hub.reset_model(model_id='')
Repõe um modelo treinado para um estado não treinado.
Código fonte em ultralytics/hub/__init__.py
ultralytics.hub.export_fmts_hub()
Devolve uma lista de formatos de exportação suportados pelo HUB.
ultralytics.hub.export_model(model_id='', format='torchscript')
Exporta um modelo para todos os formatos.
Código fonte em ultralytics/hub/__init__.py
ultralytics.hub.get_export(model_id='', format='torchscript')
Obtém um dicionário de modelos exportado com o URL de transferência.
Código fonte em ultralytics/hub/__init__.py
ultralytics.hub.check_dataset(path, task)
Função para verificar erros no ficheiro Zip do conjunto de dados do HUB antes do carregamento. Verifica se há erros em um conjunto de dados antes de ser carregado para o HUB. Dá exemplos de utilização a seguir.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
path |
str
|
Caminho para data.zip (com data.yaml dentro de data.zip). |
necessário |
task |
str
|
Tarefa do conjunto de dados. As opções são 'detetar', 'segmentar', 'posar', 'classificar', 'obb'. |
necessário |
Exemplo
Transfere ficheiros *.zip de https://github.com/ultralytics/hub/tree/main/example_datasets Ou seja, https://github.com/ultralytics/hub/raw/main/example_datasets/coco8.zip para 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