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

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

Примечание

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



ultralytics.trackers.basetrack.TrackState

Класс перечисления, представляющий возможные состояния отслеживаемого объекта.

Атрибуты:

Имя Тип Описание
New int

Состояние, когда объект только что обнаружен.

Tracked int

Состояние, когда объект успешно отслеживается в последующих кадрах.

Lost int

Состояние, когда объект больше не отслеживается.

Removed int

Состояние, когда объект снимается с отслеживания.

Исходный код в ultralytics/trackers/basetrack.py
class TrackState:
    """
    Enumeration class representing the possible states of an object being tracked.

    Attributes:
        New (int): State when the object is newly detected.
        Tracked (int): State when the object is successfully tracked in subsequent frames.
        Lost (int): State when the object is no longer tracked.
        Removed (int): State when the object is removed from tracking.
    """

    New = 0
    Tracked = 1
    Lost = 2
    Removed = 3



ultralytics.trackers.basetrack.BaseTrack

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

Атрибуты:

Имя Тип Описание
_count int

Счетчик уровня класса для уникальных идентификаторов треков.

track_id int

Уникальный идентификатор для трека.

is_activated bool

Флаг, указывающий, активен ли трек в данный момент.

state TrackState

Текущее состояние трассы.

history OrderedDict

Упорядоченная история состояний трека.

features list

Список признаков, извлеченных из объекта для отслеживания.

curr_feature any

Текущая характеристика отслеживаемого объекта.

score float

Показатель уверенности при отслеживании.

start_frame int

Номер кадра, на котором началось отслеживание.

frame_id int

Идентификатор последнего кадра, обработанного треком.

time_since_update int

Кадры, прошедшие с момента последнего обновления.

location tuple

Расположение объекта в контексте многокамерного трекинга.

Методы:

Имя Описание
end_frame

Возвращает идентификатор последнего кадра, на котором отслеживался объект.

next_id

Увеличивает и возвращает следующий глобальный идентификатор трека.

activate

Абстрактный метод для активации трека.

predict

Абстрактный метод, позволяющий предсказать следующее состояние трека.

update

Абстрактный метод обновления трека новыми данными.

mark_lost

Пометьте трек как потерянный.

mark_removed

Пометь трек как удаленный.

reset_id

Сбрось глобальный счетчик идентификаторов треков.

Исходный код в ultralytics/trackers/basetrack.py
class BaseTrack:
    """
    Base class for object tracking, providing foundational attributes and methods.

    Attributes:
        _count (int): Class-level counter for unique track IDs.
        track_id (int): Unique identifier for the track.
        is_activated (bool): Flag indicating whether the track is currently active.
        state (TrackState): Current state of the track.
        history (OrderedDict): Ordered history of the track's states.
        features (list): List of features extracted from the object for tracking.
        curr_feature (any): The current feature of the object being tracked.
        score (float): The confidence score of the tracking.
        start_frame (int): The frame number where tracking started.
        frame_id (int): The most recent frame ID processed by the track.
        time_since_update (int): Frames passed since the last update.
        location (tuple): The location of the object in the context of multi-camera tracking.

    Methods:
        end_frame: Returns the ID of the last frame where the object was tracked.
        next_id: Increments and returns the next global track ID.
        activate: Abstract method to activate the track.
        predict: Abstract method to predict the next state of the track.
        update: Abstract method to update the track with new data.
        mark_lost: Marks the track as lost.
        mark_removed: Marks the track as removed.
        reset_id: Resets the global track ID counter.
    """

    _count = 0

    def __init__(self):
        """Initializes a new track with unique ID and foundational tracking attributes."""
        self.track_id = 0
        self.is_activated = False
        self.state = TrackState.New
        self.history = OrderedDict()
        self.features = []
        self.curr_feature = None
        self.score = 0
        self.start_frame = 0
        self.frame_id = 0
        self.time_since_update = 0
        self.location = (np.inf, np.inf)

    @property
    def end_frame(self):
        """Return the last frame ID of the track."""
        return self.frame_id

    @staticmethod
    def next_id():
        """Increment and return the global track ID counter."""
        BaseTrack._count += 1
        return BaseTrack._count

    def activate(self, *args):
        """Abstract method to activate the track with provided arguments."""
        raise NotImplementedError

    def predict(self):
        """Abstract method to predict the next state of the track."""
        raise NotImplementedError

    def update(self, *args, **kwargs):
        """Abstract method to update the track with new observations."""
        raise NotImplementedError

    def mark_lost(self):
        """Mark the track as lost."""
        self.state = TrackState.Lost

    def mark_removed(self):
        """Mark the track as removed."""
        self.state = TrackState.Removed

    @staticmethod
    def reset_id():
        """Reset the global track ID counter."""
        BaseTrack._count = 0

end_frame property

Верни идентификатор последнего кадра трека.

__init__()

Инициализирует новый трек с уникальным ID и основополагающими атрибутами отслеживания.

Исходный код в ultralytics/trackers/basetrack.py
def __init__(self):
    """Initializes a new track with unique ID and foundational tracking attributes."""
    self.track_id = 0
    self.is_activated = False
    self.state = TrackState.New
    self.history = OrderedDict()
    self.features = []
    self.curr_feature = None
    self.score = 0
    self.start_frame = 0
    self.frame_id = 0
    self.time_since_update = 0
    self.location = (np.inf, np.inf)

activate(*args)

Абстрактный метод для активации трека с предоставленными аргументами.

Исходный код в ultralytics/trackers/basetrack.py
def activate(self, *args):
    """Abstract method to activate the track with provided arguments."""
    raise NotImplementedError

mark_lost()

Пометьте трек как потерянный.

Исходный код в ultralytics/trackers/basetrack.py
def mark_lost(self):
    """Mark the track as lost."""
    self.state = TrackState.Lost

mark_removed()

Пометь трек как удаленный.

Исходный код в ultralytics/trackers/basetrack.py
def mark_removed(self):
    """Mark the track as removed."""
    self.state = TrackState.Removed

next_id() staticmethod

Инкрементируй и возвращай глобальный счетчик идентификаторов треков.

Исходный код в ultralytics/trackers/basetrack.py
@staticmethod
def next_id():
    """Increment and return the global track ID counter."""
    BaseTrack._count += 1
    return BaseTrack._count

predict()

Абстрактный метод, позволяющий предсказать следующее состояние трека.

Исходный код в ultralytics/trackers/basetrack.py
def predict(self):
    """Abstract method to predict the next state of the track."""
    raise NotImplementedError

reset_id() staticmethod

Сбрось глобальный счетчик идентификаторов треков.

Исходный код в ultralytics/trackers/basetrack.py
@staticmethod
def reset_id():
    """Reset the global track ID counter."""
    BaseTrack._count = 0

update(*args, **kwargs)

Абстрактный метод для обновления трека новыми наблюдениями.

Исходный код в ultralytics/trackers/basetrack.py
def update(self, *args, **kwargs):
    """Abstract method to update the track with new observations."""
    raise NotImplementedError





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