1. 14 12月, 2021 1 次提交
  2. 13 12月, 2021 2 次提交
  3. 12 12月, 2021 1 次提交
  4. 11 12月, 2021 4 次提交
  5. 10 12月, 2021 2 次提交
  6. 09 12月, 2021 2 次提交
  7. 08 12月, 2021 4 次提交
  8. 07 12月, 2021 2 次提交
  9. 04 12月, 2021 2 次提交
    • Glenn Jocher's avatar
      Single-command multiple-model export (#5882) · 1075488d
      Glenn Jocher 提交于
      * Export multiple models in series
      
      Export multiple models in series by adding additional `*.pt` files to the `--weights` argument, i.e.:
      
      ```bash
      python export.py --include tflite --weights yolov5n.pt  # export 1 model
      python export.py --include tflite --weights yolov5n.pt yolov5s.pt yolov5m.pt yolov5l.pt yolov5x.pt  # export 5 models
      ```
      
      * Update export.py
      
      * Update README.md
      1075488d
    • Glenn Jocher's avatar
      `AutoShape()` models as `DetectMultiBackend()` instances (#5845) · 7bf04d9b
      Glenn Jocher 提交于
      * Update AutoShape()
      
      * autodownload ONNX
      
      * Cleanup
      
      * Finish updates
      
      * Add Usage
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * Update
      
      * fix device
      
      * Update hubconf.py
      
      * Update common.py
      
      * smart param selection
      
      * autodownload all formats
      
      * autopad only pytorch models
      
      * new_shape edits
      
      * stride tensor fix
      
      * Cleanup
      7bf04d9b
  10. 03 12月, 2021 2 次提交
  11. 02 12月, 2021 5 次提交
  12. 01 12月, 2021 1 次提交
  13. 30 11月, 2021 1 次提交
  14. 29 11月, 2021 3 次提交
  15. 27 11月, 2021 1 次提交
  16. 26 11月, 2021 3 次提交
  17. 25 11月, 2021 3 次提交
  18. 22 11月, 2021 1 次提交
    • 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