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

कॉलबैक

कॉलबैक

Ultralytics फ्रेमवर्क ट्रेन, वैल, निर्यात और भविष्यवाणी मोड के रणनीतिक चरणों में प्रवेश बिंदुओं के रूप में कॉलबैक का समर्थन करता है। प्रत्येक कॉलबैक एक स्वीकार करता है Trainer, Validatorनहीं तो Predictor ऑपरेशन प्रकार के आधार पर ऑब्जेक्ट। इन वस्तुओं के सभी गुण डॉक्स के संदर्भ अनुभाग में पाए जा सकते हैं।



सतर्कता: माहिर Ultralytics YOLOv8: कॉलबैक

उदाहरण

भविष्यवाणी के साथ अतिरिक्त जानकारी लौटाना

इस उदाहरण में, हम प्रत्येक परिणाम ऑब्जेक्ट के साथ मूल फ्रेम वापस करना चाहते हैं। यहां बताया गया है कि हम ऐसा कैसे कर सकते हैं

from ultralytics import YOLO


def on_predict_batch_end(predictor):
    # Retrieve the batch data
    _, 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

सभी कॉलबैक

यहां सभी समर्थित कॉलबैक हैं। अतिरिक्त विवरण के लिए कॉलबैक स्रोत कोड देखें।

ट्रेनर कॉलबैक

कॉलबैक विवरण: __________
on_pretrain_routine_start पूर्व-प्रशिक्षण दिनचर्या की शुरुआत में ट्रिगर किया गया
on_pretrain_routine_end पूर्व-प्रशिक्षण दिनचर्या के अंत में ट्रिगर किया गया
on_train_start प्रशिक्षण शुरू होने पर ट्रिगर
on_train_epoch_start प्रत्येक प्रशिक्षण युग की शुरुआत में ट्रिगर
on_train_batch_start प्रत्येक प्रशिक्षण बैच की शुरुआत में ट्रिगर किया गया
optimizer_step ऑप्टिमाइज़र चरण के दौरान ट्रिगर किया गया
on_before_zero_grad ग्रेडिएंट शून्य होने से पहले ट्रिगर किया गया
on_train_batch_end प्रत्येक प्रशिक्षण बैच के अंत में ट्रिगर किया गया
on_train_epoch_end प्रत्येक प्रशिक्षण युग के अंत में ट्रिगर किया गया
on_fit_epoch_end प्रत्येक फिट युग के अंत में ट्रिगर किया गया
on_model_save मॉडल सहेजे जाने पर ट्रिगर किया गया
on_train_end प्रशिक्षण प्रक्रिया समाप्त होने पर ट्रिगर किया गया
on_params_update मॉडल पैरामीटर अपडेट होने पर ट्रिगर किया जाता है
teardown ट्रिगर जब प्रशिक्षण प्रक्रिया को साफ किया जा रहा है

सत्यापनकर्ता कॉलबैक

कॉलबैक विवरण: __________
on_val_start सत्यापन शुरू होने पर ट्रिगर किया गया
on_val_batch_start प्रत्येक सत्यापन बैच की शुरुआत में ट्रिगर किया गया
on_val_batch_end प्रत्येक सत्यापन बैच के अंत में ट्रिगर किया गया
on_val_end सत्यापन समाप्त होने पर ट्रिगर किया गया

प्रेडिक्टर कॉलबैक

कॉलबैक विवरण: __________
on_predict_start भविष्यवाणी प्रक्रिया शुरू होने पर ट्रिगर किया गया
on_predict_batch_start प्रत्येक भविष्यवाणी बैच की शुरुआत में ट्रिगर किया गया
on_predict_postprocess_end भविष्यवाणी पोस्टप्रोसेसिंग के अंत में ट्रिगर किया गया
on_predict_batch_end प्रत्येक भविष्यवाणी बैच के अंत में ट्रिगर किया गया
on_predict_end भविष्यवाणी प्रक्रिया समाप्त होने पर ट्रिगर किया गया

निर्यातक कॉलबैक

कॉलबैक विवरण: __________
on_export_start निर्यात प्रक्रिया शुरू होने पर ट्रिगर किया गया
on_export_end निर्यात प्रक्रिया समाप्त होने पर ट्रिगर किया गया


2023-11-12 बनाया गया, अपडेट किया गया 2024-02-03
लेखक: ग्लेन-जोचर (3), chr043416@gmail.com (1), लाफिंग-क्यू (1)

टिप्पणियाँ