a yf @sddlZddlmZddlmZddlmZddlmZm Z m Z m Z ddl m Z mZmZdZdeed d d Zd d ZdddZddZdddZdddZeeddddZdS)N)HUBDatasetStats)Auth)HUBTrainingSession) HUB_API_ROOT HUB_WEB_ROOTPREFIXevents)LOGGERSETTINGSchecks) rrrloginlogout reset_modelexport_fmts_hub export_model get_export check_datasetrT)api_keyreturnc Cstdddlm}td}td}|p0|}|rF|dkrFd|ind}||}|jr|rv|j|krvt d|ji|j|ks|sdnd }t t |d St t d |d d SdS)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 hub login API_KEY'F) r check_requirementsZhub_sdkrrr get authenticatedrupdater infor) rsaverZ api_key_urlZ saved_keyZ active_key credentialsclient log_messager T/var/www/html/django/DPS/env/lib/python3.9/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 hub login'. Example: ```python from ultralytics import hub hub.logout() ``` rru6logged out ✅. To log in again, use 'yolo hub login'.N)r r rrr r r r!r Fs r rcCsbtjtdd|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)model_idrr r r!rUs " rcCs*ddlm}t|dddddgS)z/Returns a list of HUB-supported export formats.rexport_formatsZArgumentNZultralytics_tfliteZultralytics_coreml)Zultralytics.engine.exporterr1listr0r r r!r^s r torchscriptcCs|tvs Jd|dttjtd|dd|idtjid}|jdksrJt|d |jd |jt t|d d S) zExport a model to all formats.Unsupported export format '', valid formats are z /v1/models/z/exportformatr#r$r'z export failure r(u export started ✅N) rr)r*rrrr+rr-r rr.r7r/r r r!res  *rcCs||tvs Jd|dttjtdtj||ddtjid}|jdkstJt|d|jd |j| S) z3Get an exported model dictionary with download URL.r5r6z /get-export)ZapiKeyr"r7r#r$r'z get_export failure r() rr)r*rrrr+rr-r%r8r r r!ros  *r)pathtaskrcCs&t||dtdtddS)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): Path to data.zip (with data.yaml inside data.zip). task (str): Dataset task. Options are 'detect', 'segment', 'pose', 'classify', 'obb'. Example: 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. ```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 check_dataset("path/to/dota8.zip", task="obb") # OBB dataset check_dataset("path/to/imagenet10.zip", task="classify") # classification dataset ``` r9r:u7Checks completed correctly ✅. Upload this dataset to z /datasets/.N)rZget_jsonr rrr;r r r!r{sr)NT)r)rr4)rr4)r)Zultralytics.data.utilsrZultralytics.hub.authrZultralytics.hub.sessionrZultralytics.hub.utilsrrrrZultralytics.utilsr r r __all__strboolr r rrrrrr r r r!s   ,