feat(pipeline): 添加逆行处理器
- 新增 DrawDirectionProcessor 用于绘制车辆行驶方向和轨迹 - 新增 DrawGraffitiProcessor 用于绘制热力图和网格方向箭头 - 新增 DrawObjectBoxProcessor 用于绘制检测框、ID、标签和置信度 - 新增 GraffitiVisualizer 用于涂鸦可视化处理 - 新增 GraffitiProcessor 用于计算车辆轨迹并更新车道涂鸦
This commit is contained in:
@@ -0,0 +1,31 @@
|
||||
"""
|
||||
测试处理器兼容性
|
||||
"""
|
||||
from pipeline.handler.advanced_heatmap_processor import AdvancedHeatmapProcessor
|
||||
from pipeline.handler.draw_heatmap_box_processor import DrawHeatMapBoxProcessor
|
||||
from pipeline.pipeline_data import PipelineData
|
||||
import numpy as np
|
||||
import cv2
|
||||
|
||||
|
||||
def test_compatibility():
|
||||
print("测试处理器兼容性...")
|
||||
|
||||
# 创建处理器
|
||||
heatmap_proc = AdvancedHeatmapProcessor()
|
||||
draw_proc = DrawHeatMapBoxProcessor()
|
||||
|
||||
print(f"处理器创建成功: {heatmap_proc.name}, {draw_proc.name}")
|
||||
|
||||
# 创建模拟数据
|
||||
data = PipelineData()
|
||||
# 创建一个模拟帧
|
||||
data.frame = np.zeros((480, 640, 3), dtype=np.uint8)
|
||||
data.frame_idx = 1
|
||||
|
||||
# 由于我们无法创建真实的YOLO结果,我们跳过处理步骤,主要验证类定义和依赖
|
||||
print("处理器兼容性测试完成!")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
test_compatibility()
|
||||
Reference in New Issue
Block a user