1. 12 6月, 2021 2 次提交
    • Glenn Jocher's avatar
      Update `dataset_stats()` (#3593) · 7a565f13
      Glenn Jocher 提交于
      @KalenMike this is a PR to add image filenames and labels to our stats dictionary and to save the dictionary to JSON. Save location is next to the train labels.cache file. The single JSON contains all stats for entire dataset.
      
      Usage example:
      ```python
      from utils.datasets import *
      
      dataset_stats('coco128.yaml', verbose=True)
      ```
      7a565f13
    • Glenn Jocher's avatar
      train.py GPU memory fix (#3590) · 4984cf54
      Glenn Jocher 提交于
      * train.py GPU memory fix
      
      * ema
      
      * cuda
      
      * cuda
      
      * zeros input
      
      * to device
      
      * batch index 0
      4984cf54
  2. 11 6月, 2021 3 次提交
  3. 10 6月, 2021 8 次提交
  4. 09 6月, 2021 16 次提交
  5. 08 6月, 2021 6 次提交
    • Glenn Jocher's avatar
      Merge `develop` branch into `master` (#3518) · f3c3d2ce
      Glenn Jocher 提交于
      * update ci-testing.yml (#3322)
      
      * update ci-testing.yml
      
      * update greetings.yml
      
      * bring back os matrix
      
      * update ci-testing.yml (#3322)
      
      * update ci-testing.yml
      
      * update greetings.yml
      
      * bring back os matrix
      
      * Enable direct `--weights URL` definition (#3373)
      
      * Enable direct `--weights URL` definition
      
      @KalenMike this PR will enable direct --weights URL definition. Example use case:
      ```
      python train.py --weights https://storage.googleapis.com/bucket/dir/model.pt
      ```
      
      * cleanup
      
      * bug fixes
      
      * weights = attempt_download(weights)
      
      * Update experimental.py
      
      * Update hubconf.py
      
      * return bug fix
      
      * comment mirror
      
      * min_bytes
      
      * Update tutorial.ipynb (#3368)
      
      add Open in Kaggle badge
      
      * `cv2.imread(img, -1)` for IMREAD_UNCHANGED (#3379)
      
      * Update datasets.py
      
      * comment
      Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
      
      * COCO evolution fix (#3388)
      
      * COCO evolution fix
      
      * cleanup
      
      * update print
      
      * print fix
      
      * Create `is_pip()` function (#3391)
      
      Returns `True` if file is part of pip package. Useful for contextual behavior modification.
      
      ```python
      def is_pip():
          # Is file in a pip package?
          return 'site-packages' in Path(__file__).absolute().parts
      ```
      
      * Revert "`cv2.imread(img, -1)` for IMREAD_UNCHANGED (#3379)" (#3395)
      
      This reverts commit 21a9607e.
      
      * Update FLOPs description (#3422)
      
      * Update README.md
      
      * Changing FLOPS to FLOPs.
      Co-authored-by: 's avatarBuildTools <unconfigured@null.spigotmc.org>
      
      * Parse URL authentication (#3424)
      
      * Parse URL authentication
      
      * urllib.parse.unquote()
      
      * improved error handling
      
      * improved error handling
      
      * remove %3F
      
      * update check_file()
      
      * Add FLOPs title to table (#3453)
      
      * Suppress jit trace warning + graph once (#3454)
      
      * Suppress jit trace warning + graph once
      
      Suppress harmless jit trace warning on TensorBoard add_graph call. Also fix multiple add_graph() calls bug, now only on batch 0.
      
      * Update train.py
      
      * Update MixUp augmentation `alpha=beta=32.0` (#3455)
      
      Per VOC empirical results https://github.com/ultralytics/yolov5/issues/3380#issuecomment-853001307 by @developer0hye
      
      * Add `timeout()` class (#3460)
      
      * Add `timeout()` class
      
      * rearrange order
      
      * Faster HSV augmentation (#3462)
      
      remove datatype conversion process that can be skipped
      
      * Add `check_git_status()` 5 second timeout (#3464)
      
      * Add check_git_status() 5 second timeout
      
      This should prevent the SSH Git bug that we were discussing @KalenMike
      
      * cleanup
      
      * replace timeout with check_output built-in timeout
      
      * Improved `check_requirements()` offline-handling (#3466)
      
      Improve robustness of `check_requirements()` function to offline environments (do not attempt pip installs when offline).
      
      * Add `output_names` argument for ONNX export with dynamic axes (#3456)
      
      * Add output names & dynamic axes for onnx export
      
      Add output_names and dynamic_axes names for all outputs in torch.onnx.export. The first four outputs of the model will have names output0, output1, output2, output3
      
      * use first output only + cleanup
      Co-authored-by: 's avatarSamridha Shrestha <samridha.shrestha@g42.ai>
      Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
      
      * Revert FP16 `test.py` and `detect.py` inference to FP32 default (#3423)
      
      * fixed inference bug ,while use half precision
      
      * replace --use-half with --half
      
      * replace space and PEP8 in detect.py
      
      * PEP8 detect.py
      
      * update --half help comment
      
      * Update test.py
      
      * revert space
      Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
      
      * Add additional links/resources to stale.yml message (#3467)
      
      * Update stale.yml
      
      * cleanup
      
      * Update stale.yml
      
      * reformat
      
      * Update stale.yml HUB URL (#3468)
      
      * Stale `github.actor` bug fix (#3483)
      
      * Explicit `model.eval()` call `if opt.train=False` (#3475)
      
      * call model.eval() when opt.train is False
      
      call model.eval() when opt.train is False
      
      * single-line if statement
      
      * cleanup
      Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
      
      * check_requirements() exclude `opencv-python` (#3495)
      
      Fix for 3rd party or contrib versions of installed OpenCV as in https://github.com/ultralytics/yolov5/issues/3494.
      
      * Earlier `assert` for cpu and half option (#3508)
      
      * early assert for cpu and half option
      
      early assert for cpu and half option
      
      * Modified comment
      
      Modified comment
      
      * Update tutorial.ipynb (#3510)
      
      * Reduce test.py results spacing (#3511)
      
      * Update README.md (#3512)
      
      * Update README.md
      
      Minor modifications
      
      * 850 width
      
      * Update greetings.yml
      
      revert greeting change as PRs will now merge to master.
      Co-authored-by: 's avatarPiotr Skalski <SkalskiP@users.noreply.github.com>
      Co-authored-by: 's avatarSkalskiP <piotr.skalski92@gmail.com>
      Co-authored-by: 's avatarPeretz Cohen <pizzaz93@users.noreply.github.com>
      Co-authored-by: 's avatartudoulei <34886368+tudoulei@users.noreply.github.com>
      Co-authored-by: 's avatarchocosaj <chocosaj@users.noreply.github.com>
      Co-authored-by: 's avatarBuildTools <unconfigured@null.spigotmc.org>
      Co-authored-by: 's avatarYonghye Kwon <developer.0hye@gmail.com>
      Co-authored-by: 's avatarSam_S <SamSamhuns@users.noreply.github.com>
      Co-authored-by: 's avatarSamridha Shrestha <samridha.shrestha@g42.ai>
      Co-authored-by: 's avataredificewang <609552430@qq.com>
      f3c3d2ce
    • Glenn Jocher's avatar
      Update greetings.yml · c058a61e
      Glenn Jocher 提交于
      revert greeting change as PRs will now merge to master.
      c058a61e
    • Glenn Jocher's avatar
      Update README.md (#3512) · abb2a96e
      Glenn Jocher 提交于
      * Update README.md
      
      Minor modifications
      
      * 850 width
      abb2a96e
    • Glenn Jocher's avatar
      Reduce test.py results spacing (#3511) · d986145b
      Glenn Jocher 提交于
      d986145b
    • Glenn Jocher's avatar
      Update tutorial.ipynb (#3510) · eede7dc4
      Glenn Jocher 提交于
      eede7dc4
    • Yonghye Kwon's avatar
      Earlier `assert` for cpu and half option (#3508) · 8d1ddc93
      Yonghye Kwon 提交于
      * early assert for cpu and half option
      
      early assert for cpu and half option
      
      * Modified comment
      
      Modified comment
      8d1ddc93
  6. 07 6月, 2021 4 次提交
  7. 05 6月, 2021 1 次提交