onnx-metadata.json found

1123
ZhangGe6 4 years ago
parent 323c85fb1f
commit 7f2d094bfb

@ -1,4 +1,3 @@
import json
from flask import Flask, render_template, request
from onnx_modifier import onnxModifier
app = Flask(__name__)
@ -28,7 +27,6 @@ def modify_and_download_model():
onnx_modifier.modify_node_io_name(modify_info['node_renamed_io'])
onnx_modifier.check_and_save_model()
return 'OK', 200
if __name__ == '__main__':

@ -70,8 +70,7 @@ class onnxModifier:
# print('removing node {} ...'.format(node_name))
self.remove_node_by_name(node_name)
# remove node initializers (parameters)
# aka, keep and only keep the initializers of left nodes
# remove node initializers (parameters) aka, keep and only keep the initializers of left nodes
left_node_inputs = []
for left_node in self.graph.node:
left_node_inputs += left_node.input
@ -99,8 +98,7 @@ class onnxModifier:
def check_and_save_model(self, save_dir='./res_onnx'):
save_path = os.path.join(save_dir, 'modified_' + self.model_name)
onnx.checker.check_model(self.model_proto)
onnx.save(self.model_proto, save_path)
onnx.save(self.model_proto, save_path)
def inference(self):
# model_proto_bytes = onnx._serialize(model_proto_from_stream)

@ -1353,7 +1353,8 @@ onnx.Metadata = class {
if (onnx.Metadata._metadata) {
return Promise.resolve(onnx.Metadata._metadata);
}
return context.request('onnx-metadata.json', 'utf-8', null).then((data) => {
// return context.request('onnx-metadata.json', 'utf-8', null).then((data) => {
return context.request('../static/onnx-metadata.json', 'utf-8', null).then((data) => {
onnx.Metadata._metadata = new onnx.Metadata(data);
return onnx.Metadata._metadata;
}).catch(() => {

Loading…
Cancel
Save