採用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後...