onnx 探索onnx檔案

2021-10-13 23:11:45 字數 1141 閱讀 2748

採用protobuf的方式。其中 onnx.proto 就是定義的序列化檔案

protoc --decode=onnx.modelproto onnx.proto < yourfile.onnx

列印出來的結果:

ir_version:

4producer_name:

"pytorch"

producer_version:

"1.1"

graph

attribute

attribute

attribute

attribute

} node

attribute

} node

node

attribute

attribute

attribute

attribute

} node

attribute

} node

name:

"torch-jit-export"

initializer

input

dim

dim

dim }}

}}input

dim

dim

dim }}

}}input }}

}}input }}

}}input }}

}}input }}

}}input

dim

dim

dim }}

}}input }}

}}input }}

}}input }}

}}input }}

}}input }}

}}output

dim

dim

dim }}

}}}opset_import

採用這種方式,推薦找乙個只有幾層網路的onnx模型,可以清晰看到onnx模型的內部組成,內部結構

採用netron視覺化

這個只需要安裝netron之後,就ok了。方便用於檢查:資料維度,數值,連線關係等

onnx 使用初體驗

需要使用 pytorch,caffe2,coreml,onnx.由於 caffe2 只支援 python2,使用 python2.7.14來作為虛擬環境來編譯 pytorch的時候,報錯 or later clang error invalid deployment target for stdli...

ONNX原始碼安裝

onnx是facebook提出的乙個open neural network exchange協議,能夠讓訓練好的模型在不同的框架間進行互動。onnx的安裝相對來說不是特別麻煩,麻煩的是其依賴庫的安裝。onnx依賴於pybind11。首先依賴庫的安裝 sudo pip install pytest p...

onnx模型轉tensorflow模型

onnx是開源神經網路交換平台,有了它基本上不用糾結用什麼深度學習框架的問題了。我現在記錄一下怎麼將onnx模型轉換成tensorflow模型。1 安裝tensorflow和onnx 我是通過anaconda安裝的。詳情 這個部落格記載了安裝anaconda和onnx的詳情,安裝好anaconda後...