地平线部分参考算法的模型结构比较复杂,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模型可视化图在附件提供


