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