feat(pipeline): 添加逆行处理器
- 新增 DrawDirectionProcessor 用于绘制车辆行驶方向和轨迹 - 新增 DrawGraffitiProcessor 用于绘制热力图和网格方向箭头 - 新增 DrawObjectBoxProcessor 用于绘制检测框、ID、标签和置信度 - 新增 GraffitiVisualizer 用于涂鸦可视化处理 - 新增 GraffitiProcessor 用于计算车辆轨迹并更新车道涂鸦
This commit is contained in:
@@ -0,0 +1,39 @@
|
||||
"""
|
||||
测试更新后的系统,验证移除旧处理器后的功能
|
||||
"""
|
||||
from pipeline.handler import (
|
||||
AdvancedHeatmapProcessor,
|
||||
EnhancedRetrogradeProcessor,
|
||||
DrawIdBoxProcessor,
|
||||
DrawHeatMapBoxProcessor,
|
||||
DrawMotionVectorProcessor,
|
||||
ResultLogger,
|
||||
BoxFilter
|
||||
)
|
||||
|
||||
def test_updated_system():
|
||||
print("测试更新后的系统...")
|
||||
|
||||
# 测试所有处理器是否都能正确创建
|
||||
proc1 = AdvancedHeatmapProcessor()
|
||||
proc2 = EnhancedRetrogradeProcessor()
|
||||
proc3 = DrawIdBoxProcessor()
|
||||
proc4 = DrawHeatMapBoxProcessor()
|
||||
proc5 = DrawMotionVectorProcessor()
|
||||
proc6 = ResultLogger("TestLogger")
|
||||
proc7 = BoxFilter("TestFilter")
|
||||
|
||||
print('所有处理器创建成功')
|
||||
print(f'高级热力图处理器: {proc1.name}')
|
||||
print(f'增强型逆行检测处理器: {proc2.name}')
|
||||
print(f'ID框绘制处理器: {proc3.name}')
|
||||
print(f'热力图框绘制处理器: {proc4.name}')
|
||||
print(f'运动向量绘制处理器: {proc5.name}')
|
||||
print(f'日志处理器: {proc6.name}')
|
||||
print(f'过滤器: {proc7.name}')
|
||||
|
||||
print("系统更新测试完成!")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
test_updated_system()
|
||||
Reference in New Issue
Block a user