[Feature] Add tensorrt evaluation #328
Conversation
Codecov Report
@@ Coverage Diff @@
## master #328 +/- ##
==========================================
- Coverage 80.76% 80.03% -0.73%
==========================================
Files 171 178 +7
Lines 8526 9332 +806
Branches 1234 1357 +123
==========================================
+ Hits 6886 7469 +583
- Misses 1488 1674 +186
- Partials 152 189 +37
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
|
|
||
| We provide methods to evaluate TensorRT and ONNX models in `tools/deploy_test.py`. To evaluate ONNX models, onnxruntime should be installed first. | ||
|
|
||
| #### Prerequisite |
There was a problem hiding this comment.
We may need to install TensorRT as well for the evaluation of TensorRT models.
docs/tools_scripts.md
Outdated
|
|
||
| #### Results and Models | ||
|
|
||
| <table class="tg"> |
There was a problem hiding this comment.
Please keep the format of html table the same as before. For instance, the class attribute.
You can check generated HTML pages by following instructions:
- Generate HTML pages.
pip install recommonmark sphinx_markdown_tables sphinx_rtd_theme
cd docs
make html- Then view the related HTML files with the browser.
There was a problem hiding this comment.
Well, I used Table Generator. But your advise will be taken.
|
Some minor style change |
* trt_test * remove history * remove history * remove history * use html table * use html table * use html table * eval tensorrt * merge * merge into one table * specify all the prerequistes * render table * fix lint * minor style change
Mainly changes the following files:
.gitignore: add onnx and tensorrt file and local history
docs/tools_scripts.md: add tensorrt evaluation results
tools/deploy_test.py: add tensorrt evaluation parts