diff --git a/app.py b/app.py index eb07487..1e9ac1c 100644 --- a/app.py +++ b/app.py @@ -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__': diff --git a/onnx_modifier.py b/onnx_modifier.py index 6fe4f6d..ebef62f 100644 --- a/onnx_modifier.py +++ b/onnx_modifier.py @@ -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) diff --git a/static/onnx.js b/static/onnx.js index a6cea7a..dc489c0 100644 --- a/static/onnx.js +++ b/static/onnx.js @@ -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(() => {