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
, Validator
veya 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)