FPhe|ddlZddlmZddlmZddlmZmZmZddl m Z m Z d dZ dZ d dZd Zdd Zdd Zdd Zy)N)HUBDatasetStatsAuth) HUB_API_ROOT HUB_WEB_ROOTPREFIX)LOGGERSETTINGSct|dy)a& Log in to the Ultralytics HUB API using the provided API key. Args: api_key (str, optional): May be an API key or a combination API key and model ID, i.e. key_id Example: ```python from ultralytics import hub hub.login('API_KEY') ``` T)verboseNr)api_keys cC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\ultralytics/hub/__init__.pyloginr s $cvdtd<tjtjtdy)z Log out of Ultralytics HUB by removing the API key from the settings file. To log in again, use 'yolo hub login'. Example: ```python from ultralytics import hub hub.logout() ``` r u6logged out ✅. To log in again, use 'yolo hub login'.N)r saver inforrrlogoutrs+HY MMO KK6(PQRrc,tjtdtj|d}|j dk(rt jtdyt jtd|j d|jy) z,Reset a trained model to an untrained state.z /model-reset)apiKeymodelIdjsonzModel reset successfullyNzModel reset failure  ) requestspostrrr status_coder rrwarningreason)model_idrs r reset_modelr%,sn l3TV^^`h:ijA}} vh678 NNfX1!--!((LMrcDddlm}t|dddddgzS)z/Returns a list of HUB-supported export formats.rexport_formatsArgumentNultralytics_tfliteultralytics_coreml)ultralytics.engine.exporterr(listr's rexport_fmts_hubr/5s+:  ,QR0 15IK_4` ``rcb|tvsJd|dttjtd|dd|idt j i}|j dk(s)Jt|d |j d |jtjt|d y ) zExport a model to all formats.Unsupported export format '', valid formats are z /v1/models/z/exportformatz x-api-key)rheadersrz export failure ru export started ✅N) r/rrrrr r rr"r rr#r3r$s r export_modelr6;s _& &v*EfXMbcrctbu(vv & k(7C$f-*DFNN; =A ==C ^F8F83CAMM?RSTUT\T\S]!^^  KK6(6("567rc<|tvsJd|dttjtdt j ||d}|j dk(s)Jt|d|j d|j|jS) z3Get an exported model dictionary with download URL.r1r2z /get-export)rrr3rrz get_export failure r) r/rrrrr r rr"rr5s r get_exportr8Es _& &v*EfXMbcrctbu(vv & k2$(FNN%-$*, -A ==C bF8F83G VWXYX`X`Wa!bb  668Orctt||jtjdtdy)a Function for error-checking HUB dataset Zip file before upload. It checks a dataset for errors before it is uploaded to the HUB. Usage examples are given below. Args: path (str, optional): Path to data.zip (with data.yaml inside data.zip). Defaults to ''. task (str, optional): Dataset task. Options are 'detect', 'segment', 'pose', 'classify'. Defaults to 'detect'. Example: ```python 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 ``` pathtasku7Checks completed correctly ✅. Upload this dataset to z /datasets/.N)rget_jsonr rrr:s r check_datasetr>Qs.$D)224 KKI,Wbcdr)r)r torchscript)rdetect)rultralytics.data.utilsrultralytics.hub.authrultralytics.hub.utilsrrrultralytics.utilsr r rrr%r/r6r8r>rrrrEs@2%DD. " S Na 8 er