onnx-trt

1. onnx是基于protobuf来做数据存储和传输,*.proto后缀文件, 其定义是protobuf语法，类似json

2. netron可用来查看onnx网络模型结构

3. onnx可以直接编辑修改，从而省去从源头（pytorch，tensorflow等修改在转化的步骤）

 proto文件记录了onnx的结构

 graph的node存储了网络结构conv，stride，pad等等信息，initializer存储了weight，bias等，input，output则存储对应的书输入输出。

 

pytorch生成onnx

pytorch中使用torch.onnx.export生成onnx

torch.onnx.export(
    model, //需要实例化

    # 这里的args，是指输入给model的参数，类型为tuple或者tensor
    (dummy,), 

    # 储存的文件路径
    "demo.onnx", 

    # 打印详细信息
    verbose=True, 

    # 为输入和输出节点指定名称，方便后面查看或者操作
    input_names=["image"], 
    output_names=["output"], 

    # 这里的opset，指，各类算子以何种方式导出，对应于symbolic_opset11
    opset_version=11, 

    # 表示他有batch、height、width3个维度是动态的，在onnx中给其赋值为-1
    # 通常，我们只设置batch为动态，其他的避免动态
    dynamic_axes={
        "image": {0: "batch", 2: "height", 3: "width"},
        "output": {0: "batch", 2: "height", 3: "width"},
    }