1. 13 Aug, 2021 1 commit
  2. 02 Aug, 2021 2 commits
    • junji hashimoto's avatar
      Feature `python train.py --cache disk` (#4049) · 2d990632
      junji hashimoto authored
      
      
      * Add cache-on-disk and cache-directory to cache images on disk
      
      * Fix load_image with cache_on_disk
      
      * Add no_cache flag for load_image
      
      * Revert the parts('logging' and a new line) that do not need to be modified
      
      * Add the assertion for shapes of cached images
      
      * Add a suffix string for cached images
      
      * Fix boundary-error of letterbox for load_mosaic
      
      * Add prefix as cache-key of cache-on-disk
      
      * Update cache-function on disk
      
      * Add psutil in requirements.txt
      
      * Update train.py
      
      * Cleanup1
      
      * Cleanup2
      
      * Skip existing npy
      
      * Include re-space
      
      * Export return character fix
      
      Co-authored-by: default avatarGlenn Jocher <glenn.jocher@ultralytics.com>
      2d990632
    • Glenn Jocher's avatar
      Update 'results saved to' string (#4275) · 388016e9
      Glenn Jocher authored
      388016e9
  3. 25 Jul, 2021 1 commit
  4. 24 Jul, 2021 1 commit
  5. 20 Jul, 2021 2 commits
  6. 22 Jun, 2021 1 commit
  7. 21 Jun, 2021 1 commit
  8. 19 Jun, 2021 2 commits
    • Glenn Jocher's avatar
      Update DDP for `torch.distributed.run` with `gloo` backend (#3680) · fad27c00
      Glenn Jocher authored
      * Update DDP for `torch.distributed.run`
      
      * Add LOCAL_RANK
      
      * remove opt.local_rank
      
      * backend="gloo|nccl"
      
      * print
      
      * print
      
      * debug
      
      * debug
      
      * os.getenv
      
      * gloo
      
      * gloo
      
      * gloo
      
      * cleanup
      
      * fix getenv
      
      * cleanup
      
      * cleanup destroy
      
      * try nccl
      
      * return opt
      
      * add --local_rank
      
      * add timeout
      
      * add init_method
      
      * gloo
      
      * move destroy
      
      * move destroy
      
      * move print(opt) under if RANK
      
      * destroy only RANK 0
      
      * move destroy inside train()
      
      * restore destroy outside train()
      
      * update print(opt)
      
      * cleanup
      
      * nccl
      
      * gloo with 60 second timeout
      
      * update namespace printing
      fad27c00
    • lb-desupervised's avatar
      Slightly modify CLI execution (#3687) · bfb2276b
      lb-desupervised authored
      
      
      * Slightly modify CLI execution
      
      This simple change makes it easier to run the primary functions of this
      repo (train/detect/test) from within Python. An object which represents
      `opt` can be constructed and fed to the `main` function of each of these
      modules, rather than having to call the lower level functions directly,
      or run the module as a script.
      
      * Update export.py
      
      Add CLI parsing update for more convenient module usage within Python.
      
      Co-authored-by: default avatarLewis Belcher <lb@desupervised.io>
      bfb2276b
  9. 10 Jun, 2021 1 commit
  10. 09 Jun, 2021 1 commit
  11. 08 Jun, 2021 1 commit
    • Glenn Jocher's avatar
      Merge `develop` branch into `master` (#3518) · f3c3d2ce
      Glenn Jocher authored
      * 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: default 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: default 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: default avatarSamridha Shrestha <samridha.shrestha@g42.ai>
      Co-authored-by: default 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: default 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: default 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: default avatarPiotr Skalski <SkalskiP@users.noreply.github.com>
      Co-authored-by: default avatarSkalskiP <piotr.skalski92@gmail.com>
      Co-authored-by: default avatarPeretz Cohen <pizzaz93@users.noreply.github.com>
      Co-authored-by: default avatartudoulei <34886368+tudoulei@users.noreply.github.com>
      Co-authored-by: default avatarchocosaj <chocosaj@users.noreply.github.com>
      Co-authored-by: default avatarBuildTools <unconfigured@null.spigotmc.org>
      Co-authored-by: default avatarYonghye Kwon <developer.0hye@gmail.com>
      Co-authored-by: default avatarSam_S <SamSamhuns@users.noreply.github.com>
      Co-authored-by: default avatarSamridha Shrestha <samridha.shrestha@g42.ai>
      Co-authored-by: default avataredificewang <609552430@qq.com>
      f3c3d2ce
  12. 07 Jun, 2021 1 commit
  13. 06 Jun, 2021 1 commit
  14. 04 Jun, 2021 1 commit
  15. 27 May, 2021 1 commit
  16. 20 May, 2021 1 commit
  17. 16 May, 2021 1 commit
  18. 12 May, 2021 1 commit
  19. 10 May, 2021 1 commit
  20. 07 May, 2021 1 commit
  21. 06 May, 2021 1 commit
  22. 03 May, 2021 1 commit
  23. 02 May, 2021 1 commit
  24. 27 Apr, 2021 1 commit
  25. 23 Apr, 2021 4 commits
  26. 20 Apr, 2021 1 commit
  27. 16 Apr, 2021 1 commit
    • Glenn Jocher's avatar
      ONNX Simplifier (#2815) · 1f3e482b
      Glenn Jocher authored
      * ONNX Simplifier
      
      Add ONNX Simplifier to ONNX export pipeline in export.py. Will auto-install onnx-simplifier if onnx is installed but onnx-simplifier is not.
      
      * Update general.py
      1f3e482b
  28. 11 Apr, 2021 1 commit
  29. 06 Mar, 2021 1 commit
  30. 22 Feb, 2021 1 commit
  31. 17 Dec, 2020 1 commit
  32. 11 Oct, 2020 1 commit
  33. 06 Oct, 2020 1 commit
  34. 04 Oct, 2020 1 commit