Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix outputs order of tensorrt #18

Merged
merged 2 commits into from
Jul 14, 2022
Merged

Fix outputs order of tensorrt #18

merged 2 commits into from
Jul 14, 2022

Conversation

jiangjiajun
Copy link
Collaborator

  • 模型转为TensorRT后,输出顺序发生了变化。 通过OnnxReader还原输出顺序

@jiangjiajun jiangjiajun changed the title Fix output order of tensorrt Fix outputs order of tensorrt Jul 14, 2022
@jiangjiajun jiangjiajun changed the base branch from release/0.1.0 to develop July 14, 2022 11:18
@jiangjiajun jiangjiajun merged commit 90061e1 into develop Jul 14, 2022
@jiangjiajun jiangjiajun deleted the fix_trt branch July 14, 2022 11:20
DefTruth referenced this pull request in DefTruth/FastDeploy Jul 18, 2022
* update .gitignore

* Added checking for cmake include dir

* fixed missing trt_backend option bug when init from trt

* remove un-need data layout and add pre-check for dtype

* changed RGB2BRG to BGR2RGB in ppcls model

* Update CMakeLists.txt

* Add multi-label function for yolov5 (#15)

* Add multi-label function for yolov5

* Update README.md

Update doc

* fix compile for python

* Fix doc and wrong variable name (#17)

* Add multi-label function for yolov5

* Update README.md

Update doc

* Update fastdeploy_runtime.cc

fix variable option.trt_max_shape wrong name

* Update runtime_option.md

Update resnet model dynamic shape setting name from images to x

* add model_zoo yolov6 c++/python demo

* fixed CMakeLists.txt typos

* update yolov6 cpp/README.md

* add yolov6 c++ and yolov6 pybind (#16)

* update .gitignore

* Added checking for cmake include dir

* fixed missing trt_backend option bug when init from trt

* remove un-need data layout and add pre-check for dtype

* changed RGB2BRG to BGR2RGB in ppcls model

* add model_zoo yolov6 c++/python demo

* fixed CMakeLists.txt typos

* update yolov6 cpp/README.md

* Fix outputs order of tensorrt (#18)

* fix trt output order

* Update trt_backend.cc

* add yolox c++/pybind and model_zoo demo

* move some helpers to private

* fixed CMakeLists.txt typos

Co-authored-by: Jason <jiangjiajun@baidu.com>
Co-authored-by: huangjianhui <852142024@qq.com>
DefTruth referenced this pull request in DefTruth/FastDeploy Jul 18, 2022
* update .gitignore

* Added checking for cmake include dir

* fixed missing trt_backend option bug when init from trt

* remove un-need data layout and add pre-check for dtype

* changed RGB2BRG to BGR2RGB in ppcls model

* Update CMakeLists.txt

* Add multi-label function for yolov5 (#15)

* Add multi-label function for yolov5

* Update README.md

Update doc

* fix compile for python

* Fix doc and wrong variable name (#17)

* Add multi-label function for yolov5

* Update README.md

Update doc

* Update fastdeploy_runtime.cc

fix variable option.trt_max_shape wrong name

* Update runtime_option.md

Update resnet model dynamic shape setting name from images to x

* add model_zoo yolov6 c++/python demo

* fixed CMakeLists.txt typos

* update yolov6 cpp/README.md

* add yolov6 c++ and yolov6 pybind (#16)

* update .gitignore

* Added checking for cmake include dir

* fixed missing trt_backend option bug when init from trt

* remove un-need data layout and add pre-check for dtype

* changed RGB2BRG to BGR2RGB in ppcls model

* add model_zoo yolov6 c++/python demo

* fixed CMakeLists.txt typos

* update yolov6 cpp/README.md

* Fix outputs order of tensorrt (#18)

* fix trt output order

* Update trt_backend.cc

* add yolox c++/pybind and model_zoo demo

* move some helpers to private

* fixed CMakeLists.txt typos

Co-authored-by: Jason <jiangjiajun@baidu.com>
Co-authored-by: huangjianhui <852142024@qq.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant