1. 12 10月, 2021 5 次提交
    • Glenn Jocher's avatar
      YOLOv5 release v6.0 (#5141) · 956be8e6
      Glenn Jocher 提交于
      * Update P5 models
      
      * Update P6 models
      
      * Update with GFLOPs and Params
      
      * Update with GFLOPs and Params
      
      * Update README
      
      * Update
      
      * Update README
      
      * Update
      
      * Update
      
      * Add times
      
      * Update README
      
      * Update results
      
      * Update results
      
      * Update results
      
      * Update hyps
      
      * Update plots
      
      * Update plots
      
      * Update README.md
      
      * Add nano models to hubconf.py
      956be8e6
    • Andrei Ionut Damian's avatar
      update `detect.py` in order to support torch script (#5109) · 938efe5d
      Andrei Ionut Damian 提交于
      * update detect.py in order to support torch script
      
      This change assumes the torchscrip file was previously saved with `export.py`
      
      * update `detect.py` for torchscript support
      
      Simple update for torchscript support. Assumes the torchscript file has been generated with `export.py`
      
      * Cleanup
      Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
      938efe5d
    • Glenn Jocher's avatar
      Add OpenCV DNN option for ONNX inference (#5136) · 0bf24cf6
      Glenn Jocher 提交于
      * Add OpenCV DNN option for ONNX inference
      
      Usage:
      
      ```bash
      python detect.py --weights yolov5s.onnx  # ONNX Runtime inference
      python detect.py --weights yolov5s.onnx -dnn  # OpenCV DNN inference
      ```
      
      * DNN prediction to tensor
      
      * Update detect.py
      0bf24cf6
    • Jebastin Nadar's avatar
      Refactor `Detect()` anchors for ONNX <> OpenCV DNN compatibility (#4833) · 9d75e42f
      Jebastin Nadar 提交于
      * refactor anchors and anchor_grid in Detect Layer
      
      * fix CI failures by adding compatibility
      
      * fix tf failure
      
      * fix different devices errors
      
      * Cleanup
      
      * fix anchors overwriting issue
      
      * better refactoring
      
      * Remove self.anchor_grid shape check (redundant with self.grid check)
      
      Also PEP8 / 120 line width
      
      * Convert _make_grid() from static to dynamic method
      
      * Remove anchor_grid.to(device)
      
      clone() should already clone to same device as self.anchors
      
      * fix different devices error
      Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
      9d75e42f
    • Malte Lorbach's avatar
      Fix `ROOT` as relative path (#5129) · 153873e9
      Malte Lorbach 提交于
      * use os.path.relpath instead of relative_to
      
      * use os.path.relpath instead of relative_to
      
      * Remove os.path from val.py
      
      * Remove os.path from train.py
      
      * Update detect.py import to os
      
      * Update export.py import to os
      Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
      153873e9
  2. 11 10月, 2021 5 次提交
  3. 10 10月, 2021 2 次提交
  4. 09 10月, 2021 1 次提交
  5. 08 10月, 2021 3 次提交
  6. 06 10月, 2021 1 次提交
  7. 05 10月, 2021 2 次提交
    • Glenn Jocher's avatar
      Implement `--save-period` locally (#5047) · 5afc9c25
      Glenn Jocher 提交于
      This PR adds a new training argument `--save-period` to save training checkpoints every `x` epochs. To save training every 50 epochs for example:
      ```
      python train.py --save-period 50  # saves epoch50.pt, epoch100.pt, epoch150.pt, ... etc.
      ```
      
      This saved checkpoints in addition to existing last.pt and best.pt checkpoints and does not affect their behavior. Default value is -1, i.e. disabled.
      5afc9c25
    • Glenn Jocher's avatar
      Clip TTA Augmented Tails (#5028) · d133968b
      Glenn Jocher 提交于
      * Clip TTA Augmented Tails
      
      Experimental TTA update.
      
      * Update yolo.py
      
      * Update yolo.py
      
      * Update yolo.py
      
      * Update yolo.py
      d133968b
  8. 04 10月, 2021 1 次提交
    • Zhiqiang Wang's avatar
      Fix pylint: do not use bare 'except' (#5025) · 1922ddea
      Zhiqiang Wang 提交于
      * Fix E722, do not use bare 'except'
      
      * Remove used codes
      
      * Add FileNotFoundError in LoadImagesAndLabels
      
      * Remove AssertionError
      
      * Ignore LoadImagesAndLabels
      
      * Ignore downloads.py
      
      * Ignore torch_utils.py
      
      * Ignore train.py
      
      * Ignore datasets.py
      
      * Enable utils/download.py
      
      * Fixing exception in thop
      
      * Remove unused code
      
      * Fixing exception in LoadImagesAndLabels
      
      * Fixing exception in exif_size
      
      * Fixing exception in parse_model
      
      * Ignore exceptions in requests
      
      * Revert the exception as suggested
      
      * Revert the exception as suggested
      1922ddea
  9. 03 10月, 2021 1 次提交
  10. 01 10月, 2021 5 次提交
  11. 30 9月, 2021 4 次提交
  12. 29 9月, 2021 9 次提交
  13. 28 9月, 2021 1 次提交