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