İçeriğe geç

Geri aramalar

Geri aramalar

Ultralytics çerçevesi, train, val, export ve predict modlarının stratejik aşamalarında giriş noktaları olarak geri aramaları destekler. Her geri arama bir Trainer, Validatorveya Predictor işlem türüne bağlı olarak nesne. Bu nesnelerin tüm özellikleri dokümanların Referans bölümünde bulunabilir.



İzle: Mastering Ultralytics YOLOv8 : Geri aramalar

Örnekler

Tahmin ile ek bilgi döndürme

Bu örnekte, her sonuç nesnesiyle birlikte orijinal kareyi döndürmek istiyoruz. Bunu şu şekilde yapabiliriz

from ultralytics import YOLO


def on_predict_batch_end(predictor):
    """Handle prediction batch end by combining results with corresponding frames; modifies predictor results."""
    _, image, _, _ = predictor.batch

    # Ensure that image is a list
    image = image if isinstance(image, list) else [image]

    # Combine the prediction results with the corresponding frames
    predictor.results = zip(predictor.results, image)


# Create a YOLO model instance
model = YOLO(f'yolov8n.pt')

# Add the custom callback to the model
model.add_callback("on_predict_batch_end", on_predict_batch_end)

# Iterate through the results and frames
for (result, frame) in model.predict():  # or model.track()
    pass

Tüm geri aramalar

İşte desteklenen tüm geri aramalar. Ek ayrıntılar için callbacks kaynak koduna bakın.

Eğitmen Geri Aramaları

Geri arama Açıklama
on_pretrain_routine_start Antrenman öncesi rutinin başlangıcında tetiklenir
on_pretrain_routine_end Antrenman öncesi rutinin sonunda tetiklenir
on_train_start Eğitim başladığında tetiklenir
on_train_epoch_start Her eğitim epokunun başlangıcında tetiklenir
on_train_batch_start Her eğitim grubunun başlangıcında tetiklenir
optimizer_step Optimize edici adım sırasında tetiklenir
on_before_zero_grad Gradyanlar sıfırlanmadan önce tetiklenir
on_train_batch_end Her eğitim grubunun sonunda tetiklenir
on_train_epoch_end Her eğitim epokunun sonunda tetiklenir
on_fit_epoch_end Her fit epokunun sonunda tetiklenir
on_model_save Model kaydedildiğinde tetiklenir
on_train_end Eğitim süreci sona erdiğinde tetiklenir
on_params_update Model parametreleri güncellendiğinde tetiklenir
teardown Eğitim süreci temizlenirken tetiklenir

Doğrulayıcı Geri Çağırmaları

Geri arama Açıklama
on_val_start Doğrulama başladığında tetiklenir
on_val_batch_start Her doğrulama grubunun başlangıcında tetiklenir
on_val_batch_end Her doğrulama grubunun sonunda tetiklenir
on_val_end Doğrulama sona erdiğinde tetiklenir

Tahminci Geri Çağırmaları

Geri arama Açıklama
on_predict_start Tahmin süreci başladığında tetiklenir
on_predict_batch_start Her tahmin grubunun başlangıcında tetiklenir
on_predict_postprocess_end Tahmin son işlemenin sonunda tetiklenir
on_predict_batch_end Her tahmin grubunun sonunda tetiklenir
on_predict_end Tahmin süreci sona erdiğinde tetiklenir

İhracatçı Geri Çağırmaları

Geri arama Açıklama
on_export_start Dışa aktarma işlemi başladığında tetiklenir
on_export_end Dışa aktarma işlemi sona erdiğinde tetiklenir


Oluşturuldu 2023-11-12, Güncellendi 2024-05-03
Yazarlar: glenn-jocher (4), RizwanMunawar (1), Laughing-q (1)

Yorumlar