zimoyin
23555e0cc9
- 新增 DrawDirectionProcessor 用于绘制车辆行驶方向和轨迹 - 新增 DrawGraffitiProcessor 用于绘制热力图和网格方向箭头 - 新增 DrawObjectBoxProcessor 用于绘制检测框、ID、标签和置信度 - 新增 GraffitiVisualizer 用于涂鸦可视化处理 - 新增 GraffitiProcessor 用于计算车辆轨迹并更新车道涂鸦
31 lines
913 B
Python
31 lines
913 B
Python
"""
|
|
测试处理器兼容性
|
|
"""
|
|
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() |