FiddlmZmZmZ erJeddusJddlZddlmZeedsJ dde ddfd Z de d e ddfd Z dd Z dd Z ddZddZddZer e e eeedZyiZy#eef$rdZY@wxYw))LOGGERSETTINGS TESTS_RUNNINGclearmlTN)Task __version__titlereturncddl}tjx}r|D]}|js|j d|j }|rt |jdnd}|jj||j j|jdt||yy)zLog files (images) as debug samples in the ClearML task. Args: files (list[Path]): A list of file paths in PosixPath format. title (str): A title that groups together images with the same values. rNz _batch(\d+))r series local_path iteration) rer current_taskexistssearchnameintgroups get_logger report_imagereplacegroupstr)filesr rtaskfitrs e/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/ultralytics/utils/callbacks/clearml.py_log_debug_samplesr!s  ""t" AxxzYY~qvv635C A/1 !..rxxz2(FSVWXSYen/  # plot_pathcddlm}ddlm}|j |}|j }|j gdddgg}|j|tjjj|d|dy) zLog an image as a plot in the plot section of ClearML. Args: title (str): The title of the plot. plot_path (str | Path): The path to the saved image file. rN)rrr%Fauto)frameonaspectxticksyticksr )r r figurereport_interactive) matplotlib.imageimagematplotlib.pyplotpyplotimreadr+add_axesimshowrrrreport_matplotlib_figure)r r#mpimgpltimgfigaxs r _log_plotr:$su%# ,,y !C **,C lE&TV WBIIcN""$==Bsu>r"c  tjx}r/ddlm}ddlm}|j d|j dn`tj|jjxsd|jjdgddddd }tjd |jt|jd d y#t$r"}tjd |Yd}~yd}~wwxYw)zHInitialize and connect ClearML task at the start of pretraining routine.r)PatchPyTorchModelIO)PatchedMatplotlibN UltralyticsTF)pytorch matplotlib) project_name task_nametags output_urireuse_last_task_idauto_connect_frameworkszClearML Initialized a new task. If you want to run remotely, please add clearml-init and connect your arguments before initializing YOLO.General)rignore_remote_overrideszGClearML installed but not initialized correctly, not logging this run. )rr'clearml.binding.frameworks.pytorch_bindr<clearml.binding.matplotlib_bindr=update_current_taskinitargsprojectrrwarningconnectvars Exception)trainerrr<r=es r on_pretrain_routine_startrU8sf$$& &4 & T I  3 3D 9  1 1$ 799$\\11B]!,,++#_#(49(O D NN_  T',,'iQU V f`ab`cdeefsC C C:C55C:c tjx}r|jdk(r.tt |j j dd|j|jdjD]2\}}|jjd|||j4|jjD]2\}}|jjd|||j4yy) zKLog debug samples for the first epoch and report current training progress.r%ztrain_batch*.jpgMosaictrain)prefixrlrN) rrepochr!sortedsave_dirgloblabel_loss_itemstlossitemsr report_scalarr[)rSrkvs r on_train_epoch_endrfUs  ""t" ==A  vg&6&6&;&;