(не людей!), реализован на Python, OpenCV и YOLOv4. Дополнительно стоит задача по трекингу объектов, которые были за детектированы YOLO. Для трекинга используется алгоритм SORT (Simple Online Realtime Tracing), замечу, не deepSORT! Подскажите пожалуйста, возможно ли, помимо координат bounding boxes и ID объектов, получить ещё номер класса объекта, если я дополнительно его передам ?
заведите отдельный словарик параллельно, всего и делов ;)
поясни пожалуйста по подробнее
{track_key: class_detected} для первого детекшена, а потом проверяйте наличие
в кадре может находиться несколько объектов одного и того-же класса, и для того что-бы их различать нужен трекер
трекер выдаёт track_key, которому при детекции можно сопоставить класс
О каком track_key ты говоришь ? Вот пояснение к методу update, класса SORT: Params:dets - a numpy array of detections in the format [[x1,y1,x2,y2,score],[x1,y1,x2,y2,score],...] Requires: this method must be called once for each frame even with empty detections (use np.empty((0, 5)) for frames without detections). Returns the a similar array, where the last column is the object ID. NOTE: The number of objects returned may differ from the number of detections provided.
Вот перевод: Параметры: dets - числовой массив обнаружений в формате [[x1,y1,x2,y2,оценка],[x1,y1,x2,y2,оценка],...] Требования: этот метод должен вызываться один раз для каждого кадра даже с пустыми обнаружениями (используйте np.empty((0, 5)) для кадров без обнаружений). Возвращает аналогичный массив, где последним столбцом является идентификатор объекта. ПРИМЕЧАНИЕ: Количество возвращенных объектов может отличаться от количества предоставленных обнаружений.
Обсуждают сегодня