a yf @sddlmZmZmZmZz`er"Jeddus2JddlZejddddsNJddlZddlZddl m Z da iZ d a WneeefydZYn0dd d Zdd dZddZddZddZddZddZddZddZereeeeeedniZdS) )LOGGERSETTINGS TESTS_RUNNINGchecksZdvcTNdvclivez2.11.0)verbose)PathFcCs`tr\|j}td|}rF|d}tdd|j}t|||j}t t j |||dS)zDLogs images at specified path with an optional prefix using DVCLive.z _batch(\d+)Z_batchN) livenameresearchsubstemr with_suffixsuffixZ log_imageospathjoin)rprefixr mniZnew_stemr[/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/utils/callbacks/dvc.py _log_imagessrcCs>|D]0\}}|d}t||krt|||t|<qdS)zRLogs plot images for training progress if they have not been previously processed. timestampN)items_processed_plotsgetr)plotsrr paramsrrrr _log_plots*s  r"c Csg}g}|jj}t|j}|jjdkr4|dg7}t|jt D]B\}}t|D]0\}}| ||g|| ||g|qTqDt j d||ddddS)z@Logs the confusion matrix for the given validator using DVCLive.detect backgroundconfusion_matrixzcf.jsonT)r normalizedN) r%matrixlistnamesvaluestask enumerateTZastypeintextendr Zlog_sklearn_plot) validatortargetspredsr'r)tipredpinumrrr_log_confusion_matrix3s  r7c CsVztjdddatdWn4tyP}ztd|WYd}~n d}~00dS)zMInitializes DVCLive logger for training metadata during pre-training routine.T)Z save_dvc_expZ cache_imagesz[DVCLive is detected and auto logging is enabled (run 'yolo settings dvc=False' to disable).uVWARNING ⚠️ DVCLive installed but not initialized correctly, not logging this run. N)rLiver rinfo Exceptionwarning)trainererrron_pretrain_routine_startDs r>cCst|jddS)zQLogs plots related to the training process at the end of the pretraining routine.trainN)r"r r<rrron_pretrain_routine_endNsrAcCstrt|jdS)z:Logs the training parameters if DVCLive logging is active.N)r Z log_paramsargsr@rrron_train_startSsrCcCsdadS)z[Sets the global variable _training_epoch value to True at the start of training each epoch.TN)_training_epochr@rrron_train_epoch_startYsrEcCstrtri|j|jdd|j|j}|D]\}}t||q0|jdkrddl m }||D]\}}tj||ddqht |j dt |j j dtdadS) z]Logs training metrics and model info, and advances to next step on the end of each fit epoch.r?rr)model_info_for_loggersFZplotvalN)r rDlabel_loss_itemstlossmetricslrr log_metricepochZultralytics.utils.torch_utilsrGr"r r0Z next_step)r< all_metricsmetricvaluerGrrron_fit_epoch_end_s    rScCstri|j|jdd|j|j}|D]\}}tj||ddq,t|jdt|j jdt |j |j rtj |j dddtd S) z_Logs the best metrics, plots, and confusion matrix at the end of training if DVCLive is active.r?rFFrHrITmodel)copytypeN)r rJrKrLrMrrNr"r r0r7bestexistsZ log_artifactend)r<rPrQrRrrr on_train_endts    rZ)r>rArCrErSrZ)r )r )Zultralytics.utilsrrrrrZ check_versionrr pathlibrr rrD ImportErrorAssertionError TypeErrorrr"r7r>rArCrErSrZ callbacksrrrrsB