1. 13 Oct, 2021 2 commits
  2. 07 Oct, 2021 1 commit
  3. 05 Oct, 2021 1 commit
  4. 24 Sep, 2021 1 commit
  5. 23 Sep, 2021 1 commit
  6. 22 Sep, 2021 4 commits
  7. 21 Sep, 2021 1 commit
  8. 20 Sep, 2021 1 commit
  9. 26 Aug, 2021 1 commit
  10. 25 Aug, 2021 2 commits
  11. 24 Aug, 2021 2 commits
  12. 22 Aug, 2021 2 commits
  13. 18 Aug, 2021 2 commits
  14. 17 Aug, 2021 2 commits
    • Glenn Jocher's avatar
      Fix default `--weights yolov5s.pt` (#4458) · 6dd7dd8d
      Glenn Jocher authored
      6dd7dd8d
    • Jiacong Fang's avatar
      Add TensorFlow and TFLite export (#1127) · 808bcad3
      Jiacong Fang authored
      
      
      * Add models/tf.py for TensorFlow and TFLite export
      
      * Set auto=False for int8 calibration
      
      * Update requirements.txt for TensorFlow and TFLite export
      
      * Read anchors directly from PyTorch weights
      
      * Add --tf-nms to append NMS in TensorFlow SavedModel and GraphDef export
      
      * Remove check_anchor_order, check_file, set_logging from import
      
      * Reformat code and optimize imports
      
      * Autodownload model and check cfg
      
      * update --source path, img-size to 320, single output
      
      * Adjust representative_dataset
      
      * Put representative dataset in tfl_int8 block
      
      * detect.py TF inference
      
      * weights to string
      
      * weights to string
      
      * cleanup tf.py
      
      * Add --dynamic-batch-size
      
      * Add xywh normalization to reduce calibration error
      
      * Update requirements.txt
      
      TensorFlow 2.3.1 -> 2.4.0 to avoid int8 quantization error
      
      * Fix imports
      
      Move C3 from models.experimental to models.common
      
      * Add models/tf.py for TensorFlow and TFLite export
      
      * Set auto=False for int8 calibration
      
      * Update requirements.txt for TensorFlow and TFLite export
      
      * Read anchors directly from PyTorch weights
      
      * Add --tf-nms to append NMS in TensorFlow SavedModel and GraphDef export
      
      * Remove check_anchor_order, check_file, set_logging from import
      
      * Reformat code and optimize imports
      
      * Autodownload model and check cfg
      
      * update --source path, img-size to 320, single output
      
      * Adjust representative_dataset
      
      * detect.py TF inference
      
      * Put representative dataset in tfl_int8 block
      
      * weights to string
      
      * weights to string
      
      * cleanup tf.py
      
      * Add --dynamic-batch-size
      
      * Add xywh normalization to reduce calibration error
      
      * Update requirements.txt
      
      TensorFlow 2.3.1 -> 2.4.0 to avoid int8 quantization error
      
      * Fix imports
      
      Move C3 from models.experimental to models.common
      
      * implement C3() and SiLU()
      
      * Fix reshape dim to support dynamic batching
      
      * Add epsilon argument in tf_BN, which is different between TF and PT
      
      * Set stride to None if not using PyTorch, and do not warmup without PyTorch
      
      * Add list support in check_img_size()
      
      * Add list input support in detect.py
      
      * sys.path.append('./') to run from yolov5/
      
      * Add int8 quantization support for TensorFlow 2.5
      
      * Add get_coco128.sh
      
      * Remove --no-tfl-detect in models/tf.py (Use tf-android-tfl-detect branch for EdgeTPU)
      
      * Update requirements.txt
      
      * Replace torch.load() with attempt_load()
      
      * Update requirements.txt
      
      * Add --tf-raw-resize to set half_pixel_centers=False
      
      * Add --agnostic-nms for TF class-agnostic NMS
      
      * Cleanup after merge
      
      * Cleanup2 after merge
      
      * Cleanup3 after merge
      
      * Add tf.py docstring with credit and usage
      
      * pb saved_model and tflite use only one model in detect.py
      
      * Add use cases in docstring of tf.py
      
      * Remove redundant `stride` definition
      
      * Remove keras direct import
      
      * Fix `check_requirements(('tensorflow>=2.4.1',))`
      Co-authored-by: default avatarGlenn Jocher <glenn.jocher@ultralytics.com>
      808bcad3
  15. 16 Aug, 2021 2 commits
  16. 15 Aug, 2021 3 commits
  17. 14 Aug, 2021 6 commits
  18. 13 Aug, 2021 2 commits
  19. 11 Aug, 2021 4 commits
    • Glenn Jocher's avatar
    • Ayush Chaurasia's avatar
      W&B: Fix for 4360 (#4388) · a4e45530
      Ayush Chaurasia authored
      
      
      * Improve docstrings and run names
      
      * default wandb login prompt with timeout
      
      * return key
      
      * Update api_key check logic
      
      * Properly support zipped dataset feature
      
      * update docstring
      
      * Revert tuorial change
      
      * extend changes to log_dataset
      
      * add run name
      
      * bug fix
      
      * bug fix
      
      * Update comment
      
      * fix import check
      
      * remove unused import
      
      * Hardcore .yaml file extension
      
      * reduce code
      
      * Reformat using pycharm
      
      * Remove redundant try catch
      
      * More refactoring and bug fixes
      
      * retry
      
      * Reformat using pycharm
      
      * respect LOGGERS include list
      
      * Fix
      
      * fix
      Co-authored-by: default avatarGlenn Jocher <glenn.jocher@ultralytics.com>
      a4e45530
    • Ayush Chaurasia's avatar
      W&B: Add advanced features tutorial (#4384) · 75d323dd
      Ayush Chaurasia authored
      
      
      * Improve docstrings and run names
      
      * default wandb login prompt with timeout
      
      * return key
      
      * Update api_key check logic
      
      * Properly support zipped dataset feature
      
      * update docstring
      
      * Revert tuorial change
      
      * extend changes to log_dataset
      
      * add run name
      
      * bug fix
      
      * bug fix
      
      * Update comment
      
      * fix import check
      
      * remove unused import
      
      * Hardcore .yaml file extension
      
      * reduce code
      
      * Reformat using pycharm
      
      * Remove redundant try catch
      
      * More refactoring and bug fixes
      
      * retry
      
      * Reformat using pycharm
      
      * respect LOGGERS include list
      
      * Initial readme update
      
      * Update README.md
      
      * Update README.md
      
      * Update README.md
      
      * Update README.md
      
      * Update README.md
      Co-authored-by: default avatarGlenn Jocher <glenn.jocher@ultralytics.com>
      75d323dd
    • Glenn Jocher's avatar
      Update README.md (#4387) · 11e4aebf
      Glenn Jocher authored
      11e4aebf