सामग्री पर जाएं

के लिए संदर्भ ultralytics/utils/callbacks/clearml.py

नोट

यह फ़ाइल यहाँ उपलब्ध है https://github.com/ultralytics/ultralytics/बूँद/मुख्य/ultralytics/उपयोगिता/कॉलबैक/clearml.py। यदि आप कोई समस्या देखते हैं तो कृपया पुल अनुरोध का योगदान करके इसे ठीक करने में मदद करें 🛠️। 🙏 धन्यवाद !



ultralytics.utils.callbacks.clearml._log_debug_samples(files, title='Debug Samples')

लॉग फ़ाइलें (छवियां) में डीबग नमूने के रूप में ClearML नियत कार्य।

पैरामीटर:

नाम प्रकार विवरण: __________ चूक
files list

PosixPath प्रारूप में फ़ाइल पथ की एक सूची।

आवश्यक
title str

एक शीर्षक जो समान मानों के साथ छवियों को एक साथ समूहित करता है।

'Debug Samples'
में स्रोत कोड ultralytics/utils/callbacks/clearml.py
19 बांग्लादेश 19 बांग्लादेश 20 21 22 23 24 25 262728 2930313233343536
def _log_debug_samples(files, title="Debug Samples") -> None:
    """
    Log files (images) as debug samples in the ClearML task.

    Args:
        files (list): A list of file paths in PosixPath format.
        title (str): A title that groups together images with the same values.
    """
    import re

    if task := Task.current_task():
        for f in files:
            if f.exists():
                it = re.search(r"_batch(\d+)", f.name)
                iteration = int(it.groups()[0]) if it else 0
                task.get_logger().report_image(
                    title=title, series=f.name.replace(it.group(), ""), local_path=str(f), iteration=iteration
                )



ultralytics.utils.callbacks.clearml._log_plot(title, plot_path)

के प्लॉट अनुभाग में एक प्लॉट के रूप में एक छवि लॉग करें ClearML.

पैरामीटर:

नाम प्रकार विवरण: __________ चूक
title str

प्लॉट का शीर्षक।

आवश्यक
plot_path str

सहेजी गई छवि फ़ाइल का पथ.

आवश्यक
में स्रोत कोड ultralytics/utils/callbacks/clearml.py
39 40 41 42 43 44 45 46 47 48495051525354555657
def _log_plot(title, plot_path) -> None:
    """
    Log an image as a plot in the plot section of ClearML.

    Args:
        title (str): The title of the plot.
        plot_path (str): The path to the saved image file.
    """
    import matplotlib.image as mpimg
    import matplotlib.pyplot as plt

    img = mpimg.imread(plot_path)
    fig = plt.figure()
    ax = fig.add_axes([0, 0, 1, 1], frameon=False, aspect="auto", xticks=[], yticks=[])  # no ticks
    ax.imshow(img)

    Task.current_task().get_logger().report_matplotlib_figure(
        title=title, series="", figure=fig, report_interactive=False
    )



ultralytics.utils.callbacks.clearml.on_pretrain_routine_start(trainer)

प्रीट्रेनिंग रूटीन की शुरुआत में चलता है; कार्य को इनिशियलाइज़ और कनेक्ट/लॉग करता है ClearML.

में स्रोत कोड ultralytics/utils/callbacks/clearml.py
60 61 62 63 64 65 66 67 68 69 70 71 72 73 747576777879 80 81 8283
def on_pretrain_routine_start(trainer):
    """Runs at start of pretraining routine; initializes and connects/ logs task to ClearML."""
    try:
        if task := Task.current_task():
            # Make sure the automatic pytorch and matplotlib bindings are disabled!
            # We are logging these plots and model files manually in the integration
            PatchPyTorchModelIO.update_current_task(None)
            PatchedMatplotlib.update_current_task(None)
        else:
            task = Task.init(
                project_name=trainer.args.project or "YOLOv8",
                task_name=trainer.args.name,
                tags=["YOLOv8"],
                output_uri=True,
                reuse_last_task_id=False,
                auto_connect_frameworks={"pytorch": False, "matplotlib": False},
            )
            LOGGER.warning(
                "ClearML Initialized a new task. If you want to run remotely, "
                "please add clearml-init and connect your arguments before initializing YOLO."
            )
        task.connect(vars(trainer.args), name="General")
    except Exception as e:
        LOGGER.warning(f"WARNING ⚠️ ClearML installed but not initialized correctly, not logging this run. {e}")



ultralytics.utils.callbacks.clearml.on_train_epoch_end(trainer)

लॉग डिबग नमूने के पहले युग के लिए YOLO प्रशिक्षण और वर्तमान प्रशिक्षण प्रगति की रिपोर्ट करें।

में स्रोत कोड ultralytics/utils/callbacks/clearml.py
def on_train_epoch_end(trainer):
    """Logs debug samples for the first epoch of YOLO training and report current training progress."""
    if task := Task.current_task():
        # Log debug samples
        if trainer.epoch == 1:
            _log_debug_samples(sorted(trainer.save_dir.glob("train_batch*.jpg")), "Mosaic")
        # Report the current training progress
        for k, v in trainer.label_loss_items(trainer.tloss, prefix="train").items():
            task.get_logger().report_scalar("train", k, v, iteration=trainer.epoch)
        for k, v in trainer.lr.items():
            task.get_logger().report_scalar("lr", k, v, iteration=trainer.epoch)



ultralytics.utils.callbacks.clearml.on_fit_epoch_end(trainer)

एक युग के अंत में लकड़हारे को मॉडल जानकारी की रिपोर्ट करता है।

में स्रोत कोड ultralytics/utils/callbacks/clearml.py
 99 100 101 102 103 104 105 106 107 108109110 111112
def on_fit_epoch_end(trainer):
    """Reports model information to logger at the end of an epoch."""
    if task := Task.current_task():
        # You should have access to the validation bboxes under jdict
        task.get_logger().report_scalar(
            title="Epoch Time", series="Epoch Time", value=trainer.epoch_time, iteration=trainer.epoch
        )
        for k, v in trainer.metrics.items():
            task.get_logger().report_scalar("val", k, v, iteration=trainer.epoch)
        if trainer.epoch == 0:
            from ultralytics.utils.torch_utils import model_info_for_loggers

            for k, v in model_info_for_loggers(trainer).items():
                task.get_logger().report_single_value(k, v)



ultralytics.utils.callbacks.clearml.on_val_end(validator)

लेबल और भविष्यवाणियों सहित लॉग सत्यापन परिणाम।

में स्रोत कोड ultralytics/utils/callbacks/clearml.py
115 116 117 118 119
def on_val_end(validator):
    """Logs validation results including labels and predictions."""
    if Task.current_task():
        # Log val_labels and val_pred
        _log_debug_samples(sorted(validator.save_dir.glob("val*.jpg")), "Validation")



ultralytics.utils.callbacks.clearml.on_train_end(trainer)

अंतिम मॉडल और प्रशिक्षण पूरा होने पर उसका नाम लॉग करता है।

में स्रोत कोड ultralytics/utils/callbacks/clearml.py
122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139
def on_train_end(trainer):
    """Logs final model and its name on training completion."""
    if task := Task.current_task():
        # Log final results, CM matrix + PR plots
        files = [
            "results.png",
            "confusion_matrix.png",
            "confusion_matrix_normalized.png",
            *(f"{x}_curve.png" for x in ("F1", "PR", "P", "R")),
        ]
        files = [(trainer.save_dir / f) for f in files if (trainer.save_dir / f).exists()]  # filter
        for f in files:
            _log_plot(title=f.stem, plot_path=f)
        # Report final metrics
        for k, v in trainer.validator.metrics.results_dict.items():
            task.get_logger().report_single_value(k, v)
        # Log the final model
        task.update_output_model(model_path=str(trainer.best), model_name=trainer.args.name, auto_delete_file=False)





2023-11-12 बनाया गया, अपडेट किया गया 2023-11-25
लेखक: ग्लेन-जोचर (3), लाफिंग-क्यू (1)