diff --git a/readme.md b/readme.md
index a299223..5058ace 100644
--- a/readme.md
+++ b/readme.md
@@ -56,9 +56,9 @@ Click `Open Model...` to upload the ONNX model to edit. The model will be parsed
Graph-level-operation elements are placed on the left-top of the page. Currently, there are three buttons: `Preview`，`Reset` and `Download`. They can do:
-- `Preview`：Preview the result model graph with all current modifications applied ；
-- `Reset`：Reset the model graph to its initial state；
-- `Download`：Save the modified model into disk.
+- `Preview`: Preview the result model graph with all current modifications applied;
+- `Reset`: Reset the model graph to its initial state;
+- `Download`: Save the modified model into disk.
Node-level-operation elements are all in the sidebar, which can be invoked by clicking a specific node. Let's take a closer look.
@@ -95,8 +95,10 @@ For example, Now we want remove the preprocess operators (`Sub->Mul->Sub->Trans
`onnx-modifier` is under active development :hammer_and_wrench:. Welcome to use, create issues and pull requests! 🥰
# Credits and referred materials
+- [Netron](https://github.com/lutzroeder/netron)
+- [flask](https://github.com/pallets/flask)
- ONNX Python API [Official doc](https://github.com/onnx/onnx/blob/main/docs/PythonAPIOverview.md), [Leimao's Blog](https://leimao.github.io/blog/ONNX-Python-API/)
- ONNX IO Stream [Leimao's Blog](https://leimao.github.io/blog/ONNX-IO-Stream/)
-- [Netron](https://github.com/lutzroeder/netron)
- [onnx-utils](https://github.com/saurabh-shandilya/onnx-utils)
-- [flask](https://github.com/pallets/flask)
+- [sweetalert](https://github.com/t4t5/sweetalert)
+
diff --git a/readme_zh-CN.md.md b/readme_zh-CN.md.md
index 17e4864..e45d2f3 100644
--- a/readme_zh-CN.md.md
+++ b/readme_zh-CN.md.md
@@ -95,9 +95,9 @@ math: true
`onnx-modifer`正在活跃地更新中:hammer_and_wrench:。 欢迎使用，提issue，如果有帮助的话，感谢给个:star:~
# 参考资料
-
+- [Netron](https://github.com/lutzroeder/netron)
+- [flask](https://github.com/pallets/flask)
- ONNX Python API [Official doc](https://github.com/onnx/onnx/blob/main/docs/PythonAPIOverview.md), [Leimao's Blog](https://leimao.github.io/blog/ONNX-Python-API/)
- ONNX IO Stream [Leimao's Blog](https://leimao.github.io/blog/ONNX-IO-Stream/)
-- [Netron](https://github.com/lutzroeder/netron)
- [onnx-utils](https://github.com/saurabh-shandilya/onnx-utils)
-- [flask](https://github.com/pallets/flask)
+- [sweetalert](https://github.com/t4t5/sweetalert)
\ No newline at end of file
diff --git a/static/favicon.ico b/static/favicon.ico
deleted file mode 100644
index c30ac5a..0000000
Binary files a/static/favicon.ico and /dev/null differ
diff --git a/static/icon.png b/static/icon.png
deleted file mode 100644
index 75a1a0a..0000000
Binary files a/static/icon.png and /dev/null differ
diff --git a/static/index.js b/static/index.js
index 6f3480e..5f655bd 100644
--- a/static/index.js
+++ b/static/index.js
@@ -215,8 +215,6 @@ host.BrowserHost = class {
const downloadButton = this.document.getElementById('download-graph');
downloadButton.addEventListener('click', () => {
console.log(this)
- // console.log(this._view._graph._modelNodeName2State)
- // console.log(this._view._graph._renameMap)
// https://healeycodes.com/talking-between-languages
fetch('/download', {
// Declare what type of data we're sending
@@ -225,10 +223,7 @@ host.BrowserHost = class {
},
// Specify the method
method: 'POST',
- // https://blog.csdn.net/Crazy_SunShine/article/details/80624366
body: JSON.stringify({
- // 'node_states' : this._mapToJson(this._view._graph._modelNodeName2State),
- // 'node_renamed_io' : this._twoLevelMapToJson(this._view._graph._renameMap),
'node_states' : this.mapToObjectRec(this._view._graph._modelNodeName2State),
'node_renamed_io' : this.mapToObjectRec(this._view._graph._renameMap),
}
@@ -246,8 +241,8 @@ host.BrowserHost = class {
swal("Success!", "Modified model has been successfuly saved in ./modified_onnx/", "success");
}
else {
- swal("Error happens!", "You can find it out or create an issue on", "error");
- // alert('Error happens, you can find it out or create an issue on ')
+ swal("Error happens!", "You can find it out or create an issue on https://github.com/ZhangGe6/onnx-modifier", "error");
+ // alert('Error happens, you can find it out or create an issue on https://github.com/ZhangGe6/onnx-modifier')
}
});
});
@@ -286,19 +281,14 @@ host.BrowserHost = class {
openFileDialog.click();
});
openFileDialog.addEventListener('change', (e) => {
- // console.log(e)
- // console.log(e.target.value)
if (e.target && e.target.files && e.target.files.length > 0) {
const files = Array.from(e.target.files);
const file = files.find((file) => this._view.accept(file.name));
// console.log(file)
this.upload_filename = file.name;
var form = new FormData();
- // console.log(file)
form.append('file', file);
- // console.log(form)
- // console.log(form.get('file'))
-
+
// https://stackoverflow.com/questions/66039996/javascript-fetch-upload-files-to-python-flask-restful
fetch('/return_file', {
method: 'POST',
@@ -361,14 +351,10 @@ host.BrowserHost = class {
require(id) {
const url = this._url('../static/' + id + '.js');
this.window.__modules__ = this.window.__modules__ || {};
- // console.log(this.window.__modules__)
- // console.log(url) // file:///C:/Users/ZhangGe/Desktop/netron/source/./onnx.js
- // console.log(this.window.__modules__[url]) // undefined
if (this.window.__modules__[url]) {
return Promise.resolve(this.window.__exports__[url]);
}
return new Promise((resolve, reject) => {
- // console.log('here')
this.window.module = { exports: {} };
const script = document.createElement('script');
script.setAttribute('id', id);
@@ -379,8 +365,6 @@ host.BrowserHost = class {
const exports = this.window.module.exports;
delete this.window.module;
this.window.__modules__[id] = exports;
- // console.log('here')
- // console.log(exports)
resolve(exports);
}
else {
@@ -410,7 +394,6 @@ host.BrowserHost = class {
request(file, encoding, base) {
const url = base ? (base + '/' + file) : this._url(file);
- // console.log(url)
return this._request(url, null, encoding);
}
@@ -549,14 +532,8 @@ host.BrowserHost = class {
_open(file, files) {
this._view.show('welcome spinner');
const context = new host.BrowserHost.BrowserFileContext(this, file, files);
- // console.log(context);
- // console.log(model);
context.open().then(() => {
- // console.log(context);
- // console.log(model); // model is not defined
return this._view.open(context).then((model) => {
- // console.log("_open() in index.js is called");
- // console.log(model);
this._view.show(null);
this.document.title = files[0].name;
return model;
@@ -621,6 +598,7 @@ host.BrowserHost = class {
this._view.show('about');
}
+ // https://blog.csdn.net/Crazy_SunShine/article/details/80624366
_strMapToObj(strMap){
let obj = Object.create(null);
for (let [k, v] of strMap) {
@@ -860,17 +838,10 @@ host.BrowserHost.BrowserFileContext = class {
}
request(file, encoding, base) {
- // console.log(this)
- // console.log(file);
- // console.log(encoding);
- // console.log(base);
-
if (base !== undefined) {
return this._host.request(file, encoding, base);
}
- const blob = this._blobs[file];
- // console.log(blob);
-
+ const blob = this._blobs[file];
if (!blob) {
return Promise.reject(new Error("File not found '" + file + "'."));
}
@@ -920,7 +891,6 @@ host.BrowserHost.BrowserFileContext = class {
return this.request(this._file.name, null).then((stream) => {
this._stream = stream;
- // console.log(this)
});
}
};
diff --git a/static/logo.svg b/static/logo.svg
new file mode 100644
index 0000000..62e3b27
--- /dev/null
+++ b/static/logo.svg
@@ -0,0 +1,95 @@
+
+
+
diff --git a/templates/index.html b/templates/index.html
index 86829ed..1414878 100644
--- a/templates/index.html
+++ b/templates/index.html
@@ -5,15 +5,10 @@
-Netron
+onnx-modifier
-
-
-
-
-
@@ -267,124 +262,17 @@ button { font-family: -apple-system, BlinkMacSystemFont, "Segoe WPC", "Segoe UI"
+
-
+