Skip to content

Latest commit

 

History

History
 
 

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 
 
 

readme.md

知识点

  1. 目标:本小节学习如何从源码编译onnx解析器 -文件结构: avatar
  2. nvonnxparser已经被nvidia给开源了,可以使用官方自带的解析器,也可以使用源代码进行编译
  • 详情
    • 为了使用onnx导出网络有两种方式
  1. 如果需要在其他地方使用解析器,请把onnx下的pb文件,使用protoc进行编译后得到头文件,替换src/onnx下的文件
  2. onnx本质上是一个protobuf序列化后储存的文件,因此他依赖了protobuf做反序列化解析
  3. 使用源代码进行编译的好处是,可以对其进行定制化,例如插件部分的时候,或者不支持的新算子的时候
  4. 1.4 与 1.5 的在main.cpp中的唯一区别是:
    • 1.4 指定的是自带的解析器
    • 1.5 是从源码编译解析器 avatar

执行方式

  1. 生成onnx文件
    python gen-onnx.py
    • 要求安装有pytorch
    • 产生的文件会储存在workspace下
    • 产生的demo.onnx可以使用netron进行打开查看
  2. 执行编译trt模型
    make run
    

编译proto

  1. bash onnx/make_pb.sh执行编译,它会自动复制文件到src/onnx下