mirrors
/
mmyolo
mirror of https://github.com/open-mmlab/mmyolo.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmyolo/projects/easydeploy/tools
History
tripleMu a44495868d
[Feature] Support PyTorch model forward for TensorRT inference (#377) 
* Support pytorch model forward for TensorRT

* Add ort wrapper

* Fix import

* Add deploy image-demo

* raise NotImplementedError

* Fix onnxruntime output gpu tensor
 		2022-12-29 12:11:16 +08:00
..
build_engine.py [Feature] Support PyTorch model forward for TensorRT inference (#377) 2022-12-29 12:11:16 +08:00
export.py [Feature] Support PyTorch model forward for TensorRT inference (#377) 2022-12-29 12:11:16 +08:00
image-demo.py [Feature] Support PyTorch model forward for TensorRT inference (#377) 2022-12-29 12:11:16 +08:00