ONNX のモデルを JSON に変換する

漢なら ONNX モデルを JSON に変換したいですね!

ONNX は中身は protobuf(+ numpy バイナリデータ?)なので,

tensorflow の GraphDef を JSON 形式でシリアライズ/デシリアライズする
https://qiita.com/syoyo/items/3a1a5b84fb7c37c9adf5

と似たようなやりかたで, google.protobuf.json_format で JSON に変換できました.

import onnx
from google.protobuf.json_format import MessageToJson

model = onnx.load('mnist/model.onnx')

j = MessageToJson(model)
print(j)

TODO

• デシリアライズ時に onnx の protobuf 定義で JSON のチェックを行う
• ONNX JSON schema ほしい