1. 25 11月, 2021 1 次提交
  2. 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
  3. 20 11月, 2021 3 次提交
  4. 19 11月, 2021 1 次提交
  5. 18 11月, 2021 1 次提交
  6. 17 11月, 2021 1 次提交
  7. 16 11月, 2021 4 次提交
  8. 15 11月, 2021 1 次提交
  9. 14 11月, 2021 1 次提交
  10. 13 11月, 2021 2 次提交
  11. 12 11月, 2021 2 次提交
  12. 11 11月, 2021 2 次提交
  13. 10 11月, 2021 5 次提交
  14. 09 11月, 2021 1 次提交
    • Glenn Jocher's avatar
      New `DetectMultiBackend()` class (#5549) · 38832611
      Glenn Jocher 提交于
      * New `DetectMultiBackend()` class
      
      * [pre-commit.ci] auto fixes from pre-commit.com hooks
      
      for more information, see https://pre-commit.ci
      
      * pb to pt fix
      
      * Cleanup
      
      * explicit apply_classifier path
      
      * Cleanup2
      
      * Cleanup3
      
      * Cleanup4
      
      * Cleanup5
      
      * Cleanup6
      
      * val.py MultiBackend inference
      
      * warmup fix
      
      * to device fix
      
      * pt fix
      
      * device fix
      
      * Val cleanup
      
      * COCO128 URL to assets
      
      * half fix
      
      * detect fix
      
      * detect fix 2
      
      * remove half from DetectMultiBackend
      
      * training half handling
      
      * training half handling 2
      
      * training half handling 3
      
      * Cleanup
      
      * Fix CI error
      
      * Add torchscript _extra_files
      
      * Add TorchScript
      
      * Add CoreML
      
      * CoreML cleanup
      
      * New `DetectMultiBackend()` class
      
      * pb to pt fix
      
      * [pre-commit.ci] auto fixes from pre-commit.com hooks
      
      for more information, see https://pre-commit.ci
      
      * Cleanup
      
      * explicit apply_classifier path
      
      * Cleanup2
      
      * Cleanup3
      
      * Cleanup4
      
      * Cleanup5
      
      * Cleanup6
      
      * val.py MultiBackend inference
      
      * warmup fix
      
      * to device fix
      
      * pt fix
      
      * device fix
      
      * Val cleanup
      
      * COCO128 URL to assets
      
      * half fix
      
      * detect fix
      
      * detect fix 2
      
      * remove half from DetectMultiBackend
      
      * training half handling
      
      * training half handling 2
      
      * training half handling 3
      
      * Cleanup
      
      * Fix CI error
      
      * Add torchscript _extra_files
      
      * Add TorchScript
      
      * Add CoreML
      
      * CoreML cleanup
      
      * revert default to pt
      
      * Add Usage examples
      
      * Cleanup val
      Co-authored-by: 's avatarpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
      38832611
  15. 08 11月, 2021 2 次提交
  16. 07 11月, 2021 2 次提交
  17. 06 11月, 2021 8 次提交
    • Glenn Jocher's avatar
      Fix for *.yaml emojis on load (#5543) · 60c8a4f6
      Glenn Jocher 提交于
      Fix for Colab hub error:
      
      
      ```python
      import yaml
      
      with open('yolov5s.yaml', errors='ignore') as f:
           d = yaml.safe_load(f)  # model dict
      
      print(d)
      
      ---------------------------------------------------------------------------
      ReaderError                               Traceback (most recent call last)
      <ipython-input-8-1150b5143073> in <module>()
            2 
            3 with open('yolov5s.yaml', errors='ignore') as f:
      ----> 4      d = yaml.safe_load(f)  # model dict
            5 
            6 print(d)
      
      6 frames
      /usr/local/lib/python3.7/dist-packages/yaml/reader.py in check_printable(self, data)
          142             position = self.index+(len(self.buffer)-self.pointer)+match.start()
          143             raise ReaderError(self.name, position, ord(character),
      --> 144                     'unicode', "special characters are not allowed")
          145 
          146     def update(self, length):
      
      ReaderError: unacceptable character #x1f680: special characters are not allowed
        in "yolov5s.yaml", position 9
      ```
      60c8a4f6
    • Glenn Jocher's avatar
      `intersect_dicts()` in hubconf.py fix (#5542) · e189fa15
      Glenn Jocher 提交于
      e189fa15
    • Glenn Jocher's avatar
      Update `models/hub/*.yaml` files for v6.0n release (#5540) · fa2344cd
      Glenn Jocher 提交于
      * Update model yamls for v6.0
      
      * Add python models/yolo.py --test
      
      * Ghost fix
      fa2344cd
    • Glenn Jocher's avatar
      Update Issue Templates with 💡 ProTip! (#5539) · 76d90d89
      Glenn Jocher 提交于
      * Update bug-report.yml
      
      * Update question.yml
      
      * Update bug-report.yml
      76d90d89
    • Glenn Jocher's avatar
      Update autobatch.py (#5538) · cb18cac3
      Glenn Jocher 提交于
      * Update autobatch.py
      
      * Update autobatch.py
      
      * Update autobatch.py
      cb18cac3
    • Glenn Jocher's avatar
      Update autobatch.py (#5536) · 60e42e16
      Glenn Jocher 提交于
      60e42e16
    • Deep Patel's avatar
      Suppress ONNX export trace warning (#5437) · 33643799
      Deep Patel 提交于
      Checking for `onnx_dynamic` first should suppress the warning:
      
      ```log
      TracerWarning: Converting a tensor to a Python boolean might cause the trace to be incorrect. We can't record the data flow of Python values, so this value will be treated as a constant in the future. This means that the trace might not generalize to other inputs!
        if self.grid[i].shape[2:4] != x[i].shape[2:4] or self.onnx_dynamic
      ```
      33643799
    • Wonbeom Jang's avatar
      Update train.py (#5451) · d895a7f7
      Wonbeom Jang 提交于
      * correct --resume True error
      
      * delete temp file
      
      * Update train.py
      Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
      d895a7f7