FiGddlmZddlZddlZddlZddlmZddlmZddl m Z ddl m Z m Z ddlmZddlmZmZmZdd lmZmZmZmZmZdd lmZerd ed nd ed ZGddZy)) annotationsN) HTTPStatus)Path)Any)parse_qsurlparse) __version__)HELP_MSG HUB_WEB_ROOTPREFIX)IS_COLABLOGGERSETTINGSTQDMchecks) HUBModelErrorzpython-z-colabz-localceZdZdZddZedddZddZddZe ddZ dZ d dd Z e dd Z dd Zd Z d dd Ze ddZe ddZy)HUBTrainingSessionaHUB training session for Ultralytics HUB YOLO models. This class encapsulates the functionality for interacting with Ultralytics HUB during model training, including model creation, metrics tracking, and checkpoint uploading. Attributes: model_id (str): Identifier for the YOLO model being trained. model_url (str): URL for the model in Ultralytics HUB. rate_limits (dict[str, int]): Rate limits for different API calls in seconds. timers (dict[str, Any]): Timers for rate limiting. metrics_queue (dict[str, Any]): Queue for the model's metrics. metrics_upload_failed_queue (dict[str, Any]): Queue for metrics that failed to upload. model (Any): Model data fetched from Ultralytics HUB. model_file (str): Path to the model file. train_args (dict[str, Any]): Arguments for training the model. client (Any): Client for interacting with Ultralytics HUB. filename (str): Filename of the model. Examples: Create a training session with a model URL >>> session = HUBTrainingSession("https://hub.ultralytics.com/models/example-model") >>> session.upload_metrics() c:ddlm}dddd|_i|_i|_i|_d|_d|_d|_d|_ |j|\}}|_ |xstjd}|rd|ind}|||_ |r|j|y|jj |_y#t $rQ|j#t$d r6|jj&st)j*t,d YyYyYywxYw) aInitialize the HUBTrainingSession with the provided model identifier. Args: identifier (str): Model identifier used to initialize the HUB training session. It can be a URL string or a model key with specific format. Raises: ValueError: If the provided model identifier is invalid. ConnectionError: If connecting with global API key is not supported. ModuleNotFoundError: If hub-sdk package is not installed. r) HUBClienti,)metricsckpt heartbeatNapi_key/models/zzPlease log in using 'yolo login API_KEY'. You can find your API Key at: https://hub.ultralytics.com/settings?tab=api+keys.)hub_sdkr rate_limits metrics_queuemetrics_upload_failed_queuetimersmodel model_url model_file train_args_parse_identifierfilenamergetclient load_model Exception startswithr authenticatedrwarningr )self identifierrrmodel_id active_key credentialss Y/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/ultralytics/hub/session.py__init__zHUBTrainingSession.__init__.s &'(#CH+-(  ,0+A+A*+M(4=7 Y 7 1;y*-  ,  )![[..0  $$ ~X%>? HaHahggIb? s C CADDNc ||}|rF|jtds.|j||jjsJd|S#t t tf$rYywxYw)aCreate an authenticated HUBTrainingSession or return None. Args: identifier (str): Model identifier used to initialize the HUB training session. args (dict[str, Any], optional): Arguments for creating a new model if identifier is not a HUB model URL. Returns: session (HUBTrainingSession | None): An authenticated session or None if creation fails. rzHUB model not loaded correctlyN)r-r create_modelr#idPermissionErrorModuleNotFoundErrorAssertionError)clsr1argssessions r5create_sessionz!HUBTrainingSession.create_session\sm *oGJ11\N(2KL$$T*}}''I)II'N!4nE  sAAA+*A+c|jj||_|jjstd|dtd|jj |_|jjrtjd|j d|jjd}tj|ttddz |jj z |_y |j!|jj#|j$d tjt&d |j dy ) zLoad an existing model from Ultralytics HUB using the provided model identifier. Args: model_id (str): The identifier of the model to load. Raises: ValueError: If the specified HUB model does not exist. u❌ Model not found: 'z"'. Verify the model ID is correct.rzLoading trained HUB model  🚀best weights_dirhub) download_dirNrView model at )r*r#datarr r9r$ is_trainedrinfoget_weights_urlr check_filerrr%_set_train_argsstart_heartbeatrr )r0r2urls r5r+zHUBTrainingSession.load_modelqs[[&&x0 zz"8 Bd ef f(>$**--A :: " KK4T^^4DEJ K**,,V4C$//$xP]G^B_bgBgjnjtjtjwjwBwxDO   ""4#3#3K#@A vhnT^^,)>ub)I)Q)QRY[])^ _T]]+ " == ! !% (37==GI x ( 0 (zz}}">t}}oMt uv v(>$**--A ""4#3#3K#@A vhnT^^,kwargsprogress_total request_funcretryr0stream_responsetimeoutrus r5 retry_requestz7HUBTrainingSession.request_queue..retry_requestsBH519%! !IIK"$/NNfX-J8*#UV'88#NNfX-TU]T^#_`JJq!t$!..~xH$))(3==H$8$8V:;V;VVzz),;=8#O6"77%QG&'!H:RH\H\G]]^'_`))(*>*>?NNfX-=hZr(J^J^I_#`a 1a4 C! !HFJJy$90077 98MNOrPT)targetdaemonN) threadingThreadstart) r0rrrthreadrurrr>rrs ```` ````` r5 request_queuez HUBTrainingSession.request_queues7:+ + Z    M$ ? E E G!? "rPchtjtjtjh}||vS)zGDetermine if a request should be retried based on the HTTP status code.)rREQUEST_TIMEOUT BAD_GATEWAYGATEWAY_TIMEOUT)r retry_codess r5rz HUBTrainingSession._should_retryDs3  & &  " "  & & k))rPc.|j|jr |r d|d|dSdS|jtjk(r!|j}d|dd|dd |d dS |j j d d S#t$rYy wxYw)aGGenerate a retry message based on the response status code. Args: response (requests.Response): The HTTP response object. retry (int): The number of retry attempts allowed. timeout (int): The maximum timeout duration. Returns: (str): The retry message. z Retrying zx for zs.rYzRate limit reached (zX-RateLimit-Remaining/zX-RateLimit-Limitz). Please retry after z Retry-AfterrzNo JSON message.zUnable to read JSON.)rrrTOO_MANY_REQUESTSheadersjsonr)AttributeError)r0rrrrs r5rz'HUBTrainingSession._get_failure_messageNs   h22 3;@YugVG9B7 Hb H  ! !Z%A%A A&&G&w/F'G&H'ReJfIgh&&-m&<%=RA   .}}**96HII! .- .s(B BBc|j|jj|jj dS)z(Upload model metrics to Ultralytics HUB.T)rr)rr#upload_metricsr copyr|s r5rz!HUBTrainingSession.upload_metricsgs4!!$**";";TEWEWE\E\E^gk!llrPc t|}|js|jd|j}|rC|jr3t j t dtj||n t j t d|dy|j|jj|t||||dd| |r|jjndd y) aUpload a model checkpoint to Ultralytics HUB. Args: epoch (int): The current training epoch. weights (str): Path to the model weights file. is_best (bool): Indicates if the current model is the best one so far. map (float): Mean average precision of the model. final (bool): Indicates if the model is the final model after training. rta! Model 'best.pt' not found, copying 'last.pt' to 'best.pt' and uploading. This often happens when resuming training in transient environments like Google Colab. For more reliable training, consider using Ultralytics HUB Cloud. Learn more at https://docs.ultralytics.com/hub/cloud-training.z# Model upload issue. Missing model .N iT) epochweightsis_bestmapfinalrrrrr)ris_file with_namesuffixrr/r shutilrrr# upload_modelrgstatst_size)r0rrrrrrts r5rzHUBTrainingSession.upload_modelks"w- $$tGNN+;%<=DhUU  D'*&)LWIUVWX  JJ # #L95:7<<>11   rPct|ddd5}|jdD]}|jt| dddy#1swYyxYw)zGDisplay a progress bar to track the upload progress of a file download.BT)totalunit unit_scale unit_divisor chunk_sizeN)r iter_contentrlen)content_lengthrpbarrHs r5rz(HUBTrainingSession._show_upload_progresssXSTPT U 'Y] ---> ' CI& ' ' ' 's 2A  Ac2|jdD]}y)z(Process the streamed HTTP response data.rrN)r)r_s r5rz#HUBTrainingSession._iterate_contents$&&$&7 A  rP)r1rg)N)r1rgr>zdict[str, Any] | None)r2rg)rjzdict[str, Any])rTTNN) rintrrrboolrurrz int | Nonerz bool | None)rrreturnr)rrrrrrg)FgF) rrrrgrrrfloatrrrNone)rrrr)rr)__name__ __module__ __qualname____doc__r6 classmethodr@r+r8 staticmethodr'rMrrrrrrrrPr5rrs70,\(D2(DT++>&D%)'+O#O# O#  O#  O##O#%O#b**.2m , , ,  ,  ,  ,  , \'' rPr) __future__rrrrhttprpathlibrtypingr urllib.parserr ultralyticsr ultralytics.hub.utilsr r r ultralytics.utilsr rrrrultralytics.utils.errorsr AGENT_NAMErrrPr5rsY#  +#@@FF2.6w{m6 *gk]RX