1. 09 12月, 2021 1 次提交
  2. 08 12月, 2021 4 次提交
  3. 07 12月, 2021 2 次提交
  4. 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
  5. 03 12月, 2021 2 次提交
  6. 02 12月, 2021 5 次提交
  7. 01 12月, 2021 1 次提交
  8. 30 11月, 2021 1 次提交
  9. 29 11月, 2021 3 次提交
  10. 27 11月, 2021 1 次提交
  11. 26 11月, 2021 3 次提交
  12. 25 11月, 2021 3 次提交
  13. 22 11月, 2021 3 次提交
    • 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
    • Zengyf-CVer's avatar
      Save *.npy features on detect.py `--visualize` (#5701) · f17c86b7
      Zengyf-CVer 提交于
      * Add feature map to save npy files
      
      Add feature map to save npy files,export npy files with 32 feature maps per layer.
      
      * [pre-commit.ci] auto fixes from pre-commit.com hooks
      
      for more information, see https://pre-commit.ci
      
      * Update plots.py
      
      * Update plots.py
      
      * [pre-commit.ci] auto fixes from pre-commit.com hooks
      
      for more information, see https://pre-commit.ci
      
      * Update plots.py
      Co-authored-by: 's avatarpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
      Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
      f17c86b7
    • Glenn Jocher's avatar
      Created using Colaboratory · d6ae1c83
      Glenn Jocher 提交于
      d6ae1c83
  14. 20 11月, 2021 3 次提交
  15. 19 11月, 2021 1 次提交
  16. 18 11月, 2021 1 次提交
  17. 17 11月, 2021 1 次提交
  18. 16 11月, 2021 3 次提交