a yf[ @sddlmZmZddlmZz6er&Jeddus6JddlZeedsLJiZWne e fyldZYn0dd d Z dddZ ddZ ddZddZddZddZereeeedniZdS))SETTINGS TESTS_RUNNING)model_info_for_loggerswandbTN __version__Precision Recall CurveRecall Precisionc CsRddl}||||dd}dddd}|||d } tjd tj|d || d S) a Create and log a custom metric visualization to wandb.plot.pr_curve. This function crafts a custom metric visualization that mimics the behavior of the default wandb precision-recall curve while allowing for enhanced customization. The visual metric is useful for monitoring model performance across different classes. Args: x (List): Values for the x-axis; expected to have length N. y (List): Corresponding values for the y-axis; also expected to have length N. classes (List): Labels identifying the class of each point; length N. title (str, optional): Title for the plot; defaults to 'Precision Recall Curve'. x_title (str, optional): Label for the x-axis; defaults to 'Recall'. y_title (str, optional): Label for the y-axis; defaults to 'Precision'. Returns: (wandb.Object): A wandb object suitable for logging, showcasing the crafted metric visualization. rN)classyxr r r )r r r )titlez x-axis-titlez y-axis-titlezwandb/area-under-curve/v0)Z dataframe)fields string_fields)pandasZ DataFrameroundwbZ plot_tableTable) r r classesrx_titley_titlerdfrrrZ/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/utils/callbacks/wb.py _custom_tables  rprecision-recalldFc  Csddl} |durg}| |d|d|d} | } | | || j|ddd} |rtjtt | | ||gd} tj |tj j | |||dinvd gt| }t|D]>\}}| | | | | |||||gt| qtj |t| | ||||id d dS) a Log a metric curve visualization. This function generates a metric curve based on input data and logs the visualization to wandb. The curve can represent aggregated data (mean) or individual class data, depending on the 'only_mean' flag. Args: x (np.ndarray): Data points for the x-axis with length N. y (np.ndarray): Corresponding data points for the y-axis with shape CxN, where C is the number of classes. names (list, optional): Names of the classes corresponding to the y-axis data; length C. Defaults to []. id (str, optional): Unique identifier for the logged data in wandb. Defaults to 'precision-recall'. title (str, optional): Title for the visualization plot. Defaults to 'Precision Recall Curve'. x_title (str, optional): Label for the x-axis. Defaults to 'Recall'. y_title (str, optional): Label for the y-axis. Defaults to 'Precision'. num_x (int, optional): Number of interpolated data points for visualization. Defaults to 100. only_mean (bool, optional): Flag to indicate if only the mean curve should be plotted. Defaults to True. Note: The function leverages the '_custom_table' function to generate the actual visualization. rN)Zaxisr )datacolumns)rmeanF)commit)numpyZlinspacertolistZinterpr"rrlistziprunlogZplotlinelen enumerateextendr)r r namesidrrrZnum_xZ only_meannpZx_newZx_logZy_logtableriyirrr _plot_curve/s"" r4cCsX|D]F\}}|d}t||kr tjj|jtt |i|d|t|<q dS)z_Logs plots from the input dictionary if they haven't been logged already at the specified step. timestampstepN) copyitems_processed_plotsgetrr(r)stemZImagestr)plotsr7nameparamsr5rrr _log_plotses  rAcCs,tjp&tj|jjpd|jjt|jddS)z0Initiate and start project if module is present.ZYOLOv8)projectr?configN)rr(initargsrBr?varstrainerrrron_pretrain_routine_startnsrIcCsjtjj|j|jddt|j|jddt|jj|jdd|jdkrftjjt||jdddS)zCLogs training metrics and model information at the end of an epoch.r6rN) rr(r)metricsepochrAr> validatorrrGrrron_fit_epoch_endss  rNcCs\tjj|j|jdd|jddtjj|j|jdd|jdkrXt|j|jdddS)z>Log metrics and save images at the end of each training epoch.train)prefixrJr6N) rr(r)Zlabel_loss_itemsZtlossrLlrrAr>rGrrron_train_epoch_end|s" rRc Cst|jj|jddt|j|jddtjddtjjdd}|j rl| |jtjj |dgdt |jj j|jj jD]>\}}|\}}}}t||t|jj jd ||||d qtjd S) z6Save the best model as an artifact at end of training.rJr6modelZrun__model)typer?best)aliaseszcurves/)r.r/rrrN)rArMr>rLrZArtifactr(r/rVexistsZadd_fileZ log_artifactr'rKZcurvesZcurves_resultsr4r&r.valuesfinish)rHartZ curve_nameZ curve_valuesr r rrrrr on_train_ends$    r\)rIrRrNr\)rrr )Nrrrr rF)Zultralytics.utilsrrZultralytics.utils.torch_utilsrrrhasattrr: ImportErrorAssertionErrorrr4rArIrNrRr\ callbacksrrrrs@    6