使用 TVMC 编译和优化模型

处理流程

graph TB; A["tvmc compile: 输入 ONNX model。输出 TVM runtime tar 文件"] B["python preprocess.py: 输入 jpg 格式文件。输出:Numpy npz格式文件"] C["tvmc run: 输入:TVM runtime tar 文件, 转换后的Numpy格式。输出:Numpy格式tensor文件"] D["输出文件解读:python postprocess.py:输入:推理输出的tensor文件。输出:识别准确率log信息"] A-->B-->C-->D

实现

参考

posted @ 2022-12-10 23:20  山岚2013  阅读(27)  评论(0编辑  收藏  举报