zimoyin
23555e0cc9
- 新增 DrawDirectionProcessor 用于绘制车辆行驶方向和轨迹 - 新增 DrawGraffitiProcessor 用于绘制热力图和网格方向箭头 - 新增 DrawObjectBoxProcessor 用于绘制检测框、ID、标签和置信度 - 新增 GraffitiVisualizer 用于涂鸦可视化处理 - 新增 GraffitiProcessor 用于计算车辆轨迹并更新车道涂鸦
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
"""
|
|
测试更新后的系统，验证移除旧处理器后的功能
|
|
"""
|
|
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() |