o V h@sddlZddlmZddlmZddlmZddlmZm Z m Z m Z ddl m Z mZmZdZd d ed ed efd d ZddZd!defddZddZd"dedefddZd"dedefddZdeded dfddZdS)#N)HUBDatasetStats)Auth)HUBTrainingSession) HUB_API_ROOT HUB_WEB_ROOTPREFIXevents)LOGGERSETTINGSchecks) rrrloginlogout reset_modelexport_fmts_hub export_model get_export check_datasetrTapi_keysavereturnc Cstdddlm}td}td}|p|}|r#|dkr#d|ind}||}|jrQ|r;|j|kr;t d|ji|j|ksB|sDdnd }t t |d St t d |d d S)aa Log in to the Ultralytics HUB API using the provided API key. The session is not stored; a new session is created when needed using the saved SETTINGS or the HUB_API_KEY environment variable if successfully authenticated. Args: api_key (str, optional): API key to use for authentication. If not provided, it will be retrieved from SETTINGS or HUB_API_KEY environment variable. save (bool, optional): Whether to save the API key to SETTINGS if authentication is successful. Returns: (bool): True if authentication is successful, False otherwise. zhub-sdk>=0.0.12r) HUBClientz/settings?tab=api+keysrNu!New authentication successful ✅uAuthenticated ✅TzGet API key from z" and then run 'yolo login API_KEY'F) r check_requirementshub_sdkrrr get authenticatedrupdater infor) rrr api_key_url saved_key active_key credentialsclient log_messager$L/var/www/vscode/kcb/lib/python3.10/site-packages/ultralytics/hub/__init__.pyr s    r cCsdtd<ttddS)z Log out of Ultralytics HUB by removing the API key from the settings file. To log in again, use 'yolo login'. Examples: >>> from ultralytics import hub >>> hub.logout() rru2logged out ✅. To log in again, use 'yolo login'.N)r r rrr$r$r$r%r Fsr rmodel_idcCsbtjtdd|idtjid}|jdkr ttddSt td|jd |j dS) z,Reset a trained model to an untrained state.z /model-resetmodelId x-api-keyjsonheaderszModel reset successfullyNzModel reset failure  ) requestspostrrr status_coder rrwarningreason)r&rr$r$r%rRs " "rcCs*ddlm}t|dddddgS)z/Returns a list of HUB-supported export formats.rexport_formatsArgumentNultralytics_tfliteultralytics_coreml)ultralytics.engine.exporterr5listr4r$r$r%r[s r torchscriptformatcCs|tvsJd|dttjtd|dd|idtjid}|jdks9Jt|d |jd |jt t|d d S) aC Export a model to a specified format for deployment via the Ultralytics HUB API. Args: model_id (str): The ID of the model to export. An empty string will use the default model. format (str): The format to export the model to. Must be one of the supported formats returned by export_fmts_hub(). Raises: AssertionError: If the specified format is not supported or if the export request fails. Examples: >>> from ultralytics import hub >>> hub.export_model(model_id="your_model_id", format="torchscript") Unsupported export format '', valid formats are z /v1/models/z/exportr=r(r)r,z export failure r-u export started ✅N) rr.r/rrrr0rr2r rr&r=r3r$r$r%rbs *rcCs||tvsJd|dttjtdtj||ddtjid}|jdks:Jt|d|jd |j| S) a$ Retrieve an exported model in the specified format from Ultralytics HUB using the model ID. Args: model_id (str): The ID of the model to retrieve from Ultralytics HUB. format (str): The export format to retrieve. Must be one of the supported formats returned by export_fmts_hub(). Raises: AssertionError: If the specified format is not supported or if the API request fails. Examples: >>> from ultralytics import hub >>> hub.get_export(model_id="your_model_id", format="torchscript") r>r?z /get-export)apiKeyr'r=r(r)r,z get_export failure r-) rr.r/rrrr0rr2r*r@r$r$r%rzs  *rpathtaskcCs&t||dtdtddS)a Check HUB dataset Zip file for errors before upload. Args: path (str): Path to data.zip (with data.yaml inside data.zip). task (str): Dataset task. Options are 'detect', 'segment', 'pose', 'classify', 'obb'. Examples: >>> 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 Note: Download *.zip files from https://github.com/ultralytics/hub/tree/main/example_datasets i.e. https://github.com/ultralytics/hub/raw/main/example_datasets/coco8.zip for coco8.zip. rBrCu7Checks completed correctly ✅. Upload this dataset to z /datasets/.N)rget_jsonr rrrDr$r$r%rsr)NT)r)rr<)r.ultralytics.data.utilsrultralytics.hub.authrultralytics.hub.sessionrultralytics.hub.utilsrrrrultralytics.utilsr r r __all__strboolr r rrrrrr$r$r$r%s   ,