O rastreamento de objetos com Trackers e YOLO consiste principalmente nas seguintes etapas:
1. preparação de arquivos de vídeo
Prepare o vídeo de entrada (por exemplo, input.mp4) e coloque-o no diretório do projeto.
2. preparação do código de rastreamento
Um código de amostra completo é mostrado abaixo:
import supervision as sv
from trackers import SORTTracker
from ultralytics import YOLO
# 初始化组件
tracker = SORTTracker()
model = YOLO("yolo11m.pt")
annotator = sv.LabelAnnotator(text_position=sv.Position.CENTER)
# 定义帧处理函数
def callback(frame, _):
result = model(frame)[0]
detections = sv.Detections.from_ultralytics(result)
detections = tracker.update(detections)
return annotator.annotate(frame, detections, labels=detections.tracker_id)
# 处理视频
sv.process_video(
source_path="input.mp4",
target_path="output.mp4",
callback=callback
)
3. executar o código
Salve o código como track.py e execute-o diretamente:python track.py
A execução do código gera um vídeo de saída output.mp4 com anotação de ID de rastreamento.
Essa resposta foi extraída do artigoTrackers: biblioteca de ferramentas de código aberto para rastreamento de objetos de vídeoO
Português do Brasil 
简体中文 
English 
日本語 
Deutsch