Browse Source

onnx.py to onnx_ml_pb2.py

Lutz Roeder 8 years ago
parent
commit
4dd2137db3
4 changed files with 3 additions and 3 deletions
  1. 1 0
      setup.py
  2. 2 2
      src/netron.py
  3. 0 0
      src/onnx_ml_pb2.py
  4. 0 1
      tools/onnx-generate

+ 1 - 0
setup.py

@@ -19,6 +19,7 @@ package_data={
     'netron': [ 
         'netron',
         'netron.py',
+        'onnx_ml_pb2.py',
         'logo.svg',
         'onnx.js',
         'onnx-operator.json',

+ 2 - 2
src/netron.py

@@ -5,7 +5,7 @@ import os
 import platform
 import sys
 import webbrowser
-import onnx
+import onnx_ml_pb2
 
 if sys.version_info[0] > 2:
     from urllib.parse import urlparse
@@ -99,7 +99,7 @@ class OnnxModel:
         self.file = file
     def optimize(self):
         # Remove raw initializer data
-        model = onnx.ModelProto()
+        model = onnx_ml_pb2.ModelProto()
         model.ParseFromString(self.data)
         for initializer in model.graph.initializer:
             self.remove_tensor_data(initializer)

+ 0 - 0
src/onnx.py → src/onnx_ml_pb2.py


+ 0 - 1
tools/onnx-generate

@@ -21,4 +21,3 @@ python onnx-operator-json.py
 
 echo "Generate '../src/onnx.py'"
 protoc --python_out=../src --proto_path ../third_party/onnx/onnx onnx-ml.proto
-mv ../src/onnx_ml_pb2.py ../src/onnx.py