jddlmZmZmZ erJeddusJddlZddlmZeedsJn#eef$rdZYnwxYwdde d dfd Z de d e d dfd Z dd Z ddZ ddZddZddZere e eeedniZdS))LOGGERSETTINGS TESTS_RUNNINGclearmlTN)Task __version__ Debug Samplestitlereturncddl}tjx}r|D]}|r|d|j}|r't |dnd}| ||j | dt||dSdS)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 h/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/ultralytics/utils/callbacks/clearml.py_log_debug_samplesr"sIII """t  Axxzz YY~qv6635r#c  tjx}r7ddlm}ddlm}|d|dnHtj|jj pd|jj dgddddd }tj d | t|jd d dS#t$r"}tj d |Yd}~dSd}~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_startrV8sGf$&& &4  T S S S S S I I I I I I  3 3D 9 9 9  1 1$ 7 7 7 79$\1B]!,+#_#(49(O(O D N_    T',''iQU VVVVV fffdabddeeeeeeeeefsB>C C. C))C.ctjx}r|jdkr5tt |jdd||jd D]5\}}| d|||j6|j D]7\}}| d|||j6dSdS) zKLog debug samples for the first epoch and report current training progress.r&ztrain_batch*.jpgMosaictrain)prefixrlrN) rrepochr"sortedsave_dirgloblabel_loss_itemstlossitemsr report_scalarr\)rTrkvs r!on_train_epoch_endrgUs """tQ =A   vg&6&;&;rs>=========   I $ & & & &NNN 77M * *** **^$GGGSt&SST(ffff: Q Q Q Q<<<<"VVVV u u u u, %>0, $   s(5 AA