|hddlZddlmZddlmZddlmZddlmZm Z m Z m Z ddl m Z mZmZdZdded ed efd Zd Zdd efdZdZdd edefdZdd edefdZdeded dfdZy)N)HUBDatasetStats)Auth)HUBTrainingSession) HUB_API_ROOT HUB_WEB_ROOTPREFIXevents)LOGGERSETTINGSchecks) rrrloginlogout reset_modelexport_fmts_hub export_model get_export check_datasetr api_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 W/var/www/html/test/engine/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&rrFs HY KK6(LMNr'model_idc0tjtdd|idtji}|j dk(rt jtdyt jtd|j d |jy) z,Reset a trained model to an untrained state.z /model-resetmodelId x-api-keyjsonheaderszModel reset successfullyNzModel reset failure  ) requestspostrrr status_coder rrwarningreason)r*rs r&rrLsy l39h:OZegkgmguguYvwA}} vh678 NNfX1!--!((LMr'cDddlm}t|dddddgzS)z.Return a list of HUB-supported export formats.rexport_formatsArgumentNultralytics_tfliteultralytics_coreml)ultralytics.engine.exporterr;listr:s r&rrUs+:  ,QR0 15IK_4` ``r'formatcb|tvsJd|dttjtd|dd|idt j i}|j dk(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") Unsupported export format '', valid formats are z /v1/models/z/exportrBr-r.r1z export failure r2u export started ✅N) rr3r4rrrr5rr7r rr*rBr8s r&rr\s _& &v*EfXMbcrctbu(vv & . H:W5Xv>> from ultralytics import hub >>> result = hub.get_export(model_id="your_model_id", format="torchscript") rDrEz /get-export)apiKeyr,rBr-r.r1z get_export failure r2) rr3r4rrrr5rr7r/rFs r&rrts& _& &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. rIrJu7Checks completed correctly ✅. Upload this dataset to z /datasets/.N)rget_jsonr rrrLs r&rrs.(D)224 KKI,Wbcdr')NT)r)r torchscript)r3ultralytics.data.utilsrultralytics.hub.authrultralytics.hub.sessionrultralytics.hub.utilsrrrr ultralytics.utilsr r r __all__strboolr rrrrrrr)r'r&rWs2%6LL66 )3)T)T)XO N#Na838S803:ee3e4er'