Terugbellen

Terugbellen

Ultralytics Het framework ondersteunt callbacks als ingangspunten in strategische stadia van de train-, val-, export- en predictmodi. Elke callback accepteert een Trainer, Validatorof Predictor object afhankelijk van het type operatie. Alle eigenschappen van deze objecten kunnen worden gevonden in de Referentie sectie van de docs.

Kijken: Ultralytics YOLOv8 onder de knie krijgen: Callbacks

Voorbeelden

Extra informatie teruggeven met voorspelling

In dit voorbeeld willen we bij elk resultaatobject het originele frame teruggeven. Hier is hoe we dat kunnen doen

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

Alle callbacks

Hier zijn alle ondersteunde callbacks. Zie de broncode van callbacks voor meer details.

Trainer terugbellen

Validator callbacks

Voorspellende Callbacks

Exporteur Callbacks

Reacties