Skip to content

Référence pour ultralytics/trackers/track.py

Note

Ce fichier est disponible à l'adresse https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/trackers/track .py. Si tu repères un problème, aide à le corriger en contribuant à une Pull Request 🛠️. Merci 🙏 !



ultralytics.trackers.track.on_predict_start(predictor, persist=False)

Initialise les traqueurs pour le suivi des objets pendant la prédiction.

Paramètres :

Nom Type Description DĂ©faut
predictor object

L'objet prédicteur pour lequel il faut initialiser les traceurs.

requis
persist bool

Indique s'il faut faire persister les traqueurs s'ils existent déjà. La valeur par défaut est False.

False

Augmente :

Type Description
AssertionError

Si le type de tracker n'est pas 'bytetrack' ou 'botsort'.

Code source dans ultralytics/trackers/track.py
def on_predict_start(predictor: object, persist: bool = False) -> None:
    """
    Initialize trackers for object tracking during prediction.

    Args:
        predictor (object): The predictor object to initialize trackers for.
        persist (bool, optional): Whether to persist the trackers if they already exist. Defaults to False.

    Raises:
        AssertionError: If the tracker_type is not 'bytetrack' or 'botsort'.
    """
    if hasattr(predictor, "trackers") and persist:
        return

    tracker = check_yaml(predictor.args.tracker)
    cfg = IterableSimpleNamespace(**yaml_load(tracker))

    if cfg.tracker_type not in ["bytetrack", "botsort"]:
        raise AssertionError(f"Only 'bytetrack' and 'botsort' are supported for now, but got '{cfg.tracker_type}'")

    trackers = []
    for _ in range(predictor.dataset.bs):
        tracker = TRACKER_MAP[cfg.tracker_type](args=cfg, frame_rate=30)
        trackers.append(tracker)
    predictor.trackers = trackers



ultralytics.trackers.track.on_predict_postprocess_end(predictor, persist=False)

Post-traite les boîtes détectées et mets à jour le suivi des objets.

Paramètres :

Nom Type Description DĂ©faut
predictor object

L'objet prédicteur contenant les prédictions.

requis
persist bool

Indique s'il faut faire persister les traqueurs s'ils existent déjà. La valeur par défaut est False.

False
Code source dans ultralytics/trackers/track.py
def on_predict_postprocess_end(predictor: object, persist: bool = False) -> None:
    """
    Postprocess detected boxes and update with object tracking.

    Args:
        predictor (object): The predictor object containing the predictions.
        persist (bool, optional): Whether to persist the trackers if they already exist. Defaults to False.
    """
    bs = predictor.dataset.bs
    path, im0s = predictor.batch[:2]

    is_obb = predictor.args.task == "obb"
    for i in range(bs):
        if not persist and predictor.vid_path[i] != str(predictor.save_dir / Path(path[i]).name):  # new video
            predictor.trackers[i].reset()

        det = (predictor.results[i].obb if is_obb else predictor.results[i].boxes).cpu().numpy()
        if len(det) == 0:
            continue
        tracks = predictor.trackers[i].update(det, im0s[i])
        if len(tracks) == 0:
            continue
        idx = tracks[:, -1].astype(int)
        predictor.results[i] = predictor.results[i][idx]

        update_args = dict()
        update_args["obb" if is_obb else "boxes"] = torch.as_tensor(tracks[:, :-1])
        predictor.results[i].update(**update_args)



ultralytics.trackers.track.register_tracker(model, persist)

Enregistre les rappels de suivi dans le modèle pour le suivi des objets pendant la prédiction.

Paramètres :

Nom Type Description DĂ©faut
model object

L'objet du modèle pour lequel enregistrer les rappels de suivi.

requis
persist bool

S'il faut faire persister les traceurs s'ils existent déjà.

requis
Code source dans ultralytics/trackers/track.py
def register_tracker(model: object, persist: bool) -> None:
    """
    Register tracking callbacks to the model for object tracking during prediction.

    Args:
        model (object): The model object to register tracking callbacks for.
        persist (bool): Whether to persist the trackers if they already exist.
    """
    model.add_callback("on_predict_start", partial(on_predict_start, persist=persist))
    model.add_callback("on_predict_postprocess_end", partial(on_predict_postprocess_end, persist=persist))





Créé le 2023-11-12, Mis à jour le 2023-11-25
Auteurs : glenn-jocher (3), Laughing-q (1)