• imyhxy's avatar
    Export, detect and validation with TensorRT engine file (#5699) · 7a398034
    imyhxy 提交于
    * Export and detect with TensorRT engine file
    
    * Resolve `isort`
    
    * Make validation works with TensorRT engine
    
    * feat: update export docstring
    
    * feat: change suffix from *.trt to *.engine
    
    * feat: get rid of pycuda
    
    * feat: make compatiable with val.py
    
    * feat: support detect with fp16 engine
    
    * Add Lite to Edge TPU string
    
    * Remove *.trt comment
    
    * Revert to standard success logger.info string
    
    * Fix Deprecation Warning
    
    ```
    export.py:310: DeprecationWarning: Use build_serialized_network instead.
      with builder.build_engine(network, config) as engine, open(f, 'wb') as t:
    ```
    
    * Revert deprecation warning fix
    
    @imyhxy it seems we can't apply the deprecation warning fix because then export fails, so I'm reverting my previous change here.
    
    * Update export.py
    
    * Update export.py
    
    * Update common.py
    
    * export onnx to file before building TensorRT engine file
    
    * feat: triger ONNX export failed early
    
    * feat: load ONNX model from file
    Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
    7a398034
名称
最后提交
最后更新
.github 正在载入提交数据...
data 正在载入提交数据...
models 正在载入提交数据...
utils 正在载入提交数据...
.dockerignore 正在载入提交数据...
.gitattributes 正在载入提交数据...
.gitignore 正在载入提交数据...
.pre-commit-config.yaml 正在载入提交数据...
CONTRIBUTING.md 正在载入提交数据...
Dockerfile 正在载入提交数据...
LICENSE 正在载入提交数据...
README.md 正在载入提交数据...
detect.py 正在载入提交数据...
export.py 正在载入提交数据...
hubconf.py 正在载入提交数据...
requirements.txt 正在载入提交数据...
setup.cfg 正在载入提交数据...
train.py 正在载入提交数据...
tutorial.ipynb 正在载入提交数据...
val.py 正在载入提交数据...