专栏算法工具链如何可视化地平线的参考算法模型结构

如何可视化地平线的参考算法模型结构

chenyueyue2023-08-16
170
0
地平线部分参考算法的模型结构比较复杂,OE包中又没有提供专门的可视化脚本,给一些想要可视化模型结构的开发者造成了困扰。地平线的QAT Plugin中提供了模型可视化接口visualize_model,本文将基于编译脚本tool/compile_perf.py, 实现参考算法模型结构的可视化。
以下是地平线工具链手册中对visualize_model接口做的简短介绍,

本文将以bev_mt_lss参考算法为例,介绍如何可视化模式结构:

compile_perf.py脚本修改

在OE包horizon_model_train_sample/scripts/tools/compile_perf.py脚本中增加visualize_model接口,如下所示:

compile_perf.py脚本运行

注:1. 使用可视化接口前要from horizon_plugin_pytorch.quantization import visualize_model;
2.可以通过修改save_path配置可视化图存储路径;

可视化结果

完整的lss模型可视化图在附件提供

算法工具链
评论0
0/1000