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

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

नोट

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



ultralytics.utils.callbacks.neptune._log_scalars(scalars, step=0)

NeptuneAI प्रयोग लकड़हारा करने के लिए scalars लॉग करें.

में स्रोत कोड ultralytics/utils/callbacks/neptune.py
def _log_scalars(scalars, step=0):
    """Log scalars to the NeptuneAI experiment logger."""
    if run:
        for k, v in scalars.items():
            run[k].append(value=v, step=step)



ultralytics.utils.callbacks.neptune._log_images(imgs_dict, group='')

NeptuneAI प्रयोग लकड़हारा करने के लिए scalars लॉग करें.

में स्रोत कोड ultralytics/utils/callbacks/neptune.py
def _log_images(imgs_dict, group=""):
    """Log scalars to the NeptuneAI experiment logger."""
    if run:
        for k, v in imgs_dict.items():
            run[f"{group}/{k}"].upload(File(v))



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

NeptuneAI प्रयोग लकड़हारा करने के लिए लॉग प्लॉट.

पैरामीटर:

नाम प्रकार या क़िस्‍म चूक
title str

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

आवश्यक
plot_path PosixPath | str

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

आवश्यक
में स्रोत कोड ultralytics/utils/callbacks/neptune.py
def _log_plot(title, plot_path):
    """
    Log plots to the NeptuneAI experiment logger.

    Args:
        title (str): Title of the plot.
        plot_path (PosixPath | str): 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)
    run[f"Plots/{title}"].upload(fig)



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

प्रशिक्षण दिनचर्या शुरू होने से पहले कॉलबैक फ़ंक्शन कहा जाता है।

में स्रोत कोड ultralytics/utils/callbacks/neptune.py
def on_pretrain_routine_start(trainer):
    """Callback function called before the training routine starts."""
    try:
        global run
        run = neptune.init_run(project=trainer.args.project or "YOLOv8", name=trainer.args.name, tags=["YOLOv8"])
        run["Configuration/Hyperparameters"] = {k: "" if v is None else v for k, v in vars(trainer.args).items()}
    except Exception as e:
        LOGGER.warning(f"WARNING ⚠️ NeptuneAI installed but not initialized correctly, not logging this run. {e}")



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

कॉलबैक फ़ंक्शन को प्रत्येक प्रशिक्षण युग के अंत में बुलाया जाता है।

में स्रोत कोड ultralytics/utils/callbacks/neptune.py
def on_train_epoch_end(trainer):
    """Callback function called at end of each training epoch."""
    _log_scalars(trainer.label_loss_items(trainer.tloss, prefix="train"), trainer.epoch + 1)
    _log_scalars(trainer.lr, trainer.epoch + 1)
    if trainer.epoch == 1:
        _log_images({f.stem: str(f) for f in trainer.save_dir.glob("train_batch*.jpg")}, "Mosaic")



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

कॉलबैक फ़ंक्शन को प्रत्येक फिट (ट्रेन + वैल) युग के अंत में बुलाया जाता है।

में स्रोत कोड ultralytics/utils/callbacks/neptune.py
def on_fit_epoch_end(trainer):
    """Callback function called at end of each fit (train+val) epoch."""
    if run and trainer.epoch == 0:
        from ultralytics.utils.torch_utils import model_info_for_loggers

        run["Configuration/Model"] = model_info_for_loggers(trainer)
    _log_scalars(trainer.metrics, trainer.epoch + 1)



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

कॉलबैक फ़ंक्शन को प्रत्येक सत्यापन के अंत में बुलाया जाता है।

में स्रोत कोड ultralytics/utils/callbacks/neptune.py
def on_val_end(validator):
    """Callback function called at end of each validation."""
    if run:
        # Log val_labels and val_pred
        _log_images({f.stem: str(f) for f in validator.save_dir.glob("val*.jpg")}, "Validation")



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

प्रशिक्षण के अंत में कॉलबैक फ़ंक्शन कहा जाता है।

में स्रोत कोड ultralytics/utils/callbacks/neptune.py
def on_train_end(trainer):
    """Callback function called at end of training."""
    if run:
        # 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)
        # Log the final model
        run[f"weights/{trainer.args.name or trainer.args.task}/{trainer.best.name}"].upload(File(str(trainer.best)))





बनाया गया 2023-11-12, अपडेट किया गया 2024-06-02
लेखक: ग्लेन-जोचर (5), बुरहान-क्यू (1), लाफिंग-क्यू (1)