23555e0cc9
- 新增 DrawDirectionProcessor 用于绘制车辆行驶方向和轨迹 - 新增 DrawGraffitiProcessor 用于绘制热力图和网格方向箭头 - 新增 DrawObjectBoxProcessor 用于绘制检测框、ID、标签和置信度 - 新增 GraffitiVisualizer 用于涂鸦可视化处理 - 新增 GraffitiProcessor 用于计算车辆轨迹并更新车道涂鸦
23 lines
772 B
Python
23 lines
772 B
Python
"""
|
|
测试新创建的高级处理器
|
|
"""
|
|
from pipeline.handler.advanced_heatmap_processor import AdvancedHeatmapProcessor
|
|
from pipeline.handler.enhanced_reverse_direction_processor import EnhancedRetrogradeProcessor
|
|
|
|
|
|
def test_advanced_processors():
|
|
print("测试高级处理器...")
|
|
|
|
# 测试AdvancedHeatmapProcessor
|
|
advanced_heatmap_proc = AdvancedHeatmapProcessor()
|
|
print(f"AdvancedHeatmapProcessor创建成功: {advanced_heatmap_proc.name}")
|
|
|
|
# 测试EnhancedRetrogradeProcessor
|
|
enhanced_retrograde_proc = EnhancedRetrogradeProcessor()
|
|
print(f"EnhancedRetrogradeProcessor创建成功: {enhanced_retrograde_proc.name}")
|
|
|
|
print("所有高级处理器测试完成!")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_advanced_processors() |