zimoyin
23555e0cc9
- 新增 DrawDirectionProcessor 用于绘制车辆行驶方向和轨迹 - 新增 DrawGraffitiProcessor 用于绘制热力图和网格方向箭头 - 新增 DrawObjectBoxProcessor 用于绘制检测框、ID、标签和置信度 - 新增 GraffitiVisualizer 用于涂鸦可视化处理 - 新增 GraffitiProcessor 用于计算车辆轨迹并更新车道涂鸦
10 lines
439 B
Python
10 lines
439 B
Python
from ..base_processor import BaseProcessor
|
|
from ..pipeline_data import PipelineData
|
|
import numpy as np
|
|
|
|
class ResultLogger(BaseProcessor):
|
|
"""示例处理器：打印检测结果日志"""
|
|
def process(self, data: PipelineData) -> PipelineData:
|
|
print(f"\n【{self.name}】帧{data.frame_idx} - 检测到目标数: {len(data.current_result.boxes)}")
|
|
print(f"缓存帧数: {len(data.result_cache)}")
|
|
return data |