Export, detect and validation with TensorRT engine file#5699

Merged

glenn-jocher merged 19 commits intoultralytics:masterfrom

imyhxy:yt

Nov 22, 2021

Commits on Nov 22, 2021

Export and detect with TensorRT engine file

imyhxy
authored and
glenn-jocher
committed
Resolve `isort`

imyhxy
authored and
glenn-jocher
committed
Make validation works with TensorRT engine

imyhxy
authored and
glenn-jocher
committed
feat: update export docstring

imyhxy
authored and
glenn-jocher
committed
feat: change suffix from *.trt to *.engine

imyhxy
authored and
glenn-jocher
committed
feat: get rid of pycuda

imyhxy
authored and
glenn-jocher
committed
feat: make compatiable with val.py

imyhxy
authored and
glenn-jocher
committed
feat: support detect with fp16 engine

imyhxy
authored and
glenn-jocher
committed
Add Lite to Edge TPU string
glenn-jocher
committed
Remove *.trt comment
glenn-jocher
committed
Revert to standard success logger.info string
glenn-jocher
committed
Fix Deprecation Warning
glenn-jocher
committed
Revert deprecation warning fix
glenn-jocher
committed
Update export.py
glenn-jocher
committed
Update export.py
glenn-jocher
committed
Update common.py
glenn-jocher
committed
export onnx to file before building TensorRT engine file

imyhxy
authored and
glenn-jocher
committed
feat: triger ONNX export failed early

imyhxy
authored and
glenn-jocher
committed
feat: load ONNX model from file

imyhxy
authored and
glenn-jocher
committed