5 Commits
Author SHA1 Message Date
tripleMuandAlexey 8dc755a359 Add grid concat and fuse such operators (#389)
* Add grid concat and fuse so many op

* Fix model

* Fix other detector

* Update yolo.py

* Update yolo.py

Co-authored-by: Alexey <AlexeyAB@users.noreply.github.com>
2022-08-09 07:26:05 +03:00
tripleMu 6bacefff5c Unified output names (#283) 2022-07-29 04:58:57 +03:00
tripleMu a7c00297d5 Support dynamic batch for TensorRT and onnxruntime (#329)
* Support dynamic batch for TensorRT and onnxruntime

* Fix output name

* Add some images

* Add dynamic-batch usage notebook

* Add example notebook for onnxruntime and tensorrt
2022-07-29 01:01:43 +03:00
tripleMu 7a7cba744a Fix yolo for torch.onnx.export (#314) 2022-07-26 18:03:20 +03:00
tripleMuandAlexey 1c59e43d9d Add end2end yolov7 onnx export for TensorRT8.0+ and onnxruntime(testing now) (#273)
* Add end2end yolov7 onnx export for TensorRT8.0+

* Add usage in README

* Update yolo.py

* Update yolo.py

* Add tensorrt onnxruntime examples

* Add usage in README

Co-authored-by: Alexey <AlexeyAB@users.noreply.github.com>
2022-07-23 05:36:51 +03:00