-
Notifications
You must be signed in to change notification settings - Fork 127
Support super resolution models and benchmarking #291
base: master
Are you sure you want to change the base?
Conversation
@@ -0,0 +1,5 @@ | |||
# Licenses |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
should this be license or readme?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Since we put tfjs in our benchmark, we need to specify tfjs's license here to state that we are using some other project. I think i included a README in the same folder.
@@ -0,0 +1,38 @@ | |||
# Benchmarks |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
could we generalize this sub project to other models besides SR, which would be very useful feature?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yes i put a TODO in the main index.js file to improve this benchmark to take arbitrary models and configs. For that, we would need a benchmark driver similar to our test-runner-cli. I will do a followup change just for the benchmarking tool.
depth-to-space kernel bug fixes
migrating this to the new repo...
This PR adds the following support:
depth to space kernel to run office super resolution model
6d tensor support in coordinate library to run model zoo super resolution
a super resolution benchmarking tool that compares the performance of onnxjs to the latest tfjs with profiling info.
This pr also exposes a logger config(only contains a log level field) to the inference session so that logger's log level can be configured from onnxjs api.
***All the required changes are verified currently. This PR is for viewing diffs only. Changes are being migrated to the new ORT repo.