.h*ddlmZddlmZddlmZddlmZmZm Z ddl m Z m Z m Z dZdded ed efd Zd Zdd efdZdZdd edefdZdd edefdZdeded dfdZy))HUBDatasetStats)Auth)HUBTrainingSession) HUB_API_ROOT HUB_WEB_ROOTPREFIX)LOGGERSETTINGSchecks) rrrloginlogout reset_modelexport_fmts_hub export_model get_export check_datasetNapi_keysavereturnctjdddlm}td}t j d}|xs|}|r |dk7rd|ind}||}|jre|r0|j|k7r!t jd|ji|j|k(s|sdnd }tjt|y tjtd |d y )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_messages V/var/www/html/ai-service/venv/lib/python3.12/site-packages/ultralytics/hub/__init__.pyr r s /0!!N"89K Y'I%IJ-7J">W3LT_ /ex   vh{m,-  vh/ }<^_`cNdtd<tjtdy)zJLog out of Ultralytics HUB by removing the API key from the settings file.rru2logged out ✅. To log in again, use 'yolo login'.N)r r rrr&r%r r Cs HY KK6(LMNr&model_idc0ddl}|jtdd|idtji}|j dk(rt jtdyt jtd |j d |jy) z,Reset a trained model to an untrained state.rNz /model-resetmodelId x-api-keyjsonheaderszModel reset successfullyzModel reset failure  ) requestspostrrr status_coder rrwarningreason)r)r2rs r%rrIs~ l39h:OZegkgmguguYv wA}} vh678 NNfX1!--!((LMr&cDddlm}t|dddddgzS)z.Return a list of HUB-supported export formats.rexport_formatsArgumentNultralytics_tfliteultralytics_coreml)ultralytics.engine.exporterr:listr9s r%rrTs+:  ,QR0 15IK_4` ``r&formatcbddl}|tvsJd|dt|jtd|dd|idt j i }|j d k(s)Jt|d |j d |jtjt|d y)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") rNUnsupported export format '', valid formats are z /v1/models/z/exportrAr,r-r0z export failure r1u export started ✅) r2rr3rrrr4rr6r rr)rAr2r7s r%rr[s  _& &v*EfXMbcrctbu(vv & . H:W5Xv>> from ultralytics import hub >>> result = hub.get_export(model_id="your_model_id", format="torchscript") rNrCrDz /get-export)apiKeyr+rAr,r-r0z get_export failure r1) r2rr3rrrr4rr6r.rEs r%rrus& _& &v*EfXMbcrctbu(vv & . $8v Ndfnn-  A ==C bF8F83G VWXYX`X`Wa!bb  668Or&pathtaskctt||jtjdtdy)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 Notes: 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. rHrIu7Checks completed correctly ✅. Upload this dataset to z /datasets/.N)rget_jsonr rrrKs r%rrs.(D)224 KKI,Wbcdr&)NT)r)r torchscript)ultralytics.data.utilsrultralytics.hub.authrultralytics.hub.sessionrultralytics.hub.utilsrrrultralytics.utilsr r r __all__strboolr r rrrrrr(r&r%rVs3%6DD66 )3)T)T)XO N#Na838S843>ee3e4er&