Перейти к содержимому

Ссылка для ultralytics/trackers/track.py

Примечание

Этот файл доступен по адресу https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/trackers/track .py. Если ты заметил проблему, пожалуйста, помоги исправить ее, отправив Pull Request 🛠️. Спасибо 🙏!



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

Инициализируй трекеры для отслеживания объектов во время предсказания.

Параметры:

Имя Тип Описание По умолчанию
predictor object

Объект предсказателя, для которого нужно инициализировать трекеры.

требуется
persist bool

Нужно ли сохранять трекеры, если они уже существуют. По умолчанию - False.

False

Поднимает:

Тип Описание
AssertionError

Если тип_трекера не 'bytetrack' или 'botsort'.

Исходный код в 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)

Постпроцессинг обнаруженных коробок и обновление отслеживания объектов.

Параметры:

Имя Тип Описание По умолчанию
predictor object

Объект предсказателя, содержащий предсказания.

требуется
persist bool

Нужно ли сохранять трекеры, если они уже существуют. По умолчанию - False.

False
Исходный код в 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)

Зарегистрируй в модели обратные вызовы слежения для отслеживания объектов во время предсказания.

Параметры:

Имя Тип Описание По умолчанию
model object

Объект модели, для которого нужно зарегистрировать обратные вызовы отслеживания.

требуется
persist bool

Нужно ли сохранять трекеры, если они уже существуют.

требуется
Исходный код в 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))





Создано 2023-11-12, Обновлено 2023-11-25
Авторы: glenn-jocher (3), Laughing-q (1)