Glenn Jocher
6445a8137e
Resolve dataset paths ( #6489 )
2 years ago
Glenn Jocher
d8b5beb0b0
Fix2 `select_device()` for Multi-GPU ( #6461 )
* Fix2 select_device() for Multi-GPU
* Cleanup
* Cleanup
* Simplify error message
* Improve assert
* Update torch_utils.py
2 years ago
Glenn Jocher
856d4e5733
Fix `select_device()` for Multi-GPU ( #6434 )
* Fix `select_device()` for Multi-GPU
Possible fix for https://github.com/ultralytics/yolov5/issues/6431
* Update torch_utils.py
* Update torch_utils.py
* Update torch_utils.py
* Update torch_utils.py
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
2 years ago
Glenn Jocher
fe7de6a82d
Revert "Remove `dataset_stats()` autodownload capability ( #6303 )" ( #6442 )
This reverts commit 3119b2f27c
.
2 years ago
toschi23
3b7ac28ed1
Add `*.asf` video support ( #6436 )
2 years ago
Glenn Jocher
d5966c93f1
Namespace `VERBOSE` env variable to `YOLOv5_VERBOSE` ( #6428 )
* Verbose updates
* Verbose updates
2 years ago
Jonathan Samelson
ed9bac8392
Rename logger from 'utils.logger' to 'yolov5' ( #6421 )
* Gave a more explicit name to the logger
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
Glenn Jocher
482af479c0
Add `detect.py` GIF video inference ( #6410 )
* Add detect.py GIF video inference
* Cleanup
2 years ago
Matthias
8efe97719c
Add `stop_training=False` flag to callbacks ( #6365 )
* New flag 'stop_training' in util.callbacks.Callbacks class to prematurely stop training from callback handler
* Removed most of the new checks, leaving only the one after calling 'on_train_batch_end'
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
Ayush Chaurasia
9bcc32a5bf
Flush callbacks when on `--evolve` ( #6374 )
* log best.pt metrics at train end
* update
* Update __init__.py
* flush callbacks when using evolve
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
sitecao
e1893c894a
`DEVICE_COUNT` instead of `WORLD_SIZE` to calculate `nw` ( #6324 )
2 years ago
imyhxy
4e841b9b16
Reuse `de_parallel()` rather than `is_parallel()` ( #6354 )
2 years ago
johnk2hawaii
9708cf56ea
New environment variable `VERBOSE` ( #6353 )
New environment variable `VERBOSE`
2 years ago
Glenn Jocher
fd55271c04
Console corrupted -> corrupt ( #6338 )
* Console corrupted -> corrupt
Minor style changes.
* Update export.py
2 years ago
Glenn Jocher
3119b2f27c
Remove `dataset_stats()` autodownload capability ( #6303 )
* Remove `dataset_stats()` autodownload capability
@kalenmike security update per Slack convo
* Update datasets.py
2 years ago
Glenn Jocher
436ffc417a
`select_device()` cleanup ( #6302 )
* `select_device()` cleanup
* Update torch_utils.py
* Update torch_utils.py
* Update torch_utils.py
* Update torch_utils.py
* Update torch_utils.py
2 years ago
Henry
a1a9c6884c
Fixing bug multi-gpu training ( #6299 )
* Fixing bug multi-gpu training
This solves this issue: https://github.com/ultralytics/yolov5/issues/6297#issue-1103853348
* Update torch_utils.py for pep8
2 years ago
Glenn Jocher
e7bf38277f
Fix `device` count check ( #6290 )
* Fix device count check()
* Update torch_utils.py
* Update torch_utils.py
* Update hubconf.py
2 years ago
Glenn Jocher
af001349e4
Add `is_kaggle()` function ( #6285 )
* Add `is_kaggle()` function
Return True if environment is Kaggle Notebook.
* Remove root loggers only if is_kaggle() == True
* Update general.py
2 years ago
Glenn Jocher
9e9219fe17
Fix `nan`-robust stream FPS ( #6198 )
Fix for Webcam stop working suddenly (Issue #6197 )
2 years ago
Glenn Jocher
b4a29b5a8d
Update NMS `max_wh=7680` for 8k images ( #6178 )
2 years ago
Chen Gen
affa284352
Refactor/reduce G/C/D/IoU `if: else` statements ( #6087 )
* Refactor the code to reduece else
* Update metrics.py
* Cleanup
Co-authored-by: Cmos <gen.chen@ubisoft.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
Awsaf
7b6938d5b5
Log best results ( #6085 )
* log best result in summary
* comment added
* add space for `flake8`
* log `best/epoch`
* fix `dimension` for epoch
ValueError: all the input arrays must have same number of dimensions
* log `best/` in `utils.logger.__init__`
* fix pre-commit
1. missing whitespace around operator
2. over-indented
2 years ago
Ayush Chaurasia
9155eb8641
W&B: Log best results after training ends ( #6120 )
* log best.pt metrics at train end
* update
* Update __init__.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
Ayush Chaurasia
db6ec66a60
W&B: track batch size after autobatch ( #6039 )
* track batch size after autobatch
* remove redundant import
* Update __init__.py
* Update __init__.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
JieLi
afa5cfb0f8
Reduce G/D/CIoU logic operations ( #6074 )
Consider that the default value is CIOU,adjust the order of judgment could reduce the number of judgments.
And “elif CIoU:” didn't need 'if'.
Co-authored-by: 李杰 <360751194@qq.comqq.com>
2 years ago
Glenn Jocher
b8a4babd60
Simplify `set_logging()` indexing ( #6042 )
2 years ago
Glenn Jocher
0db9d5b6a2
Kaggle `LOGGER` fix ( #6041 )
2 years ago
Glenn Jocher
407a905747
Check TensorRT>=8.0.0 version ( #6021 )
* Check TensorRT>=8.0.0 version
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Glenn Jocher
da9a1b719b
Allow `--weights URL` ( #5991 )
2 years ago
Mrinal Jain
b7d18f3636
Increase `ar_thr` from 20 to 100 for better detection on slender (high aspect ratio) objects ( #5556 )
* Making `ar_thr` available as a hyperparameter
* Disabling ar_thr as hyperparameter and computing from the dataset instead
* Fixing bug in ar_thr computation
* Fix `ar_thr` to 100
2 years ago
Glenn Jocher
c9a46a60e0
Update callbacks.py with `__init__()` ( #5979 )
Add __init__() function.
2 years ago
Glenn Jocher
8f875d93a2
Refactor NUM_THREADS ( #5954 )
2 years ago
iumyx2612
a42af30d8a
Update `strip_optimizer()` ( #5949 )
Replace 'training_result' with 'best_fitness' in strip_optimizer() to match key with ckpt from train.py
2 years ago
Felix You
c45f9f678d
Make `select_device()` robust to `batch_size=-1` ( #5940 )
* Find out a bug. When set batch_size = -1 to use the autobatch.
reproduce:
* Fix type conflict
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
Glenn Jocher
c77a5a84e3
Absolute '/content/sample_data' ( #5922 )
2 years ago
Glenn Jocher
a3d5f1d3e3
Revert "Update `plot_lr_scheduler()` ( #5864 )" ( #5920 )
This reverts commit 360eec6910
.
2 years ago
Glenn Jocher
7d56d45124
Add hardware checks to `notebook_init()` ( #5919 )
* Update notebook
* Update notebook
* update string
* update string
* Updates
* Updates
* Updates
* check both ipython and psutil
* remove sample_data if is_colab
* cleanup
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 years ago
Ayush Chaurasia
3f152e5807
Update wandb_utils.py ( #5908 )
2 years ago
Glenn Jocher
7bf04d9bbf
`AutoShape()` models as `DetectMultiBackend()` instances ( #5845 )
* Update AutoShape()
* autodownload ONNX
* Cleanup
* Finish updates
* Add Usage
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* Update
* fix device
* Update hubconf.py
* Update common.py
* smart param selection
* autodownload all formats
* autopad only pytorch models
* new_shape edits
* stride tensor fix
* Cleanup
2 years ago
Glenn Jocher
d885799c71
Update `nl` after `cutout()` ( #5873 )
2 years ago
Can
360eec6910
Update `plot_lr_scheduler()` ( #5864 )
shallow copy modify originals
2 years ago
Glenn Jocher
00e308f7be
Update TorchScript suffix to `*.torchscript` ( #5856 )
2 years ago
gggmt
1679aacdc7
Update `LoadImages` `ret_val=False` handling ( #5852 )
Video errors may occur.
2 years ago
Yu Zhang
e8f8f2b903
Avoid inplace modifying`imgs` in `LoadStreams` ( #5850 )
When OpenCV retrieving image fail, original code would modify source images **inplace**, which may result in plotting bounding boxes on a black image. That is, before inference, source image `im0s[i]` is OK, but after inference before `Process predictions`, `im0s[i]` may have been changed.
2 years ago
Glenn Jocher
bc484579d7
Handle non-TTY `wandb.errors.UsageError` ( #5839 )
* `try: except (..., wandb.errors.UsageError)`
* bug fix
2 years ago
Glenn Jocher
5ca5dd4c87
Update `dataset_stats()` to `cv2.INTER_AREA` ( #5821 )
2 years ago
Glenn Jocher
7c6bae0ae6
Remove NCOLS from tqdm ( #5804 )
* Remove NCOLS from tqdm
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
3 years ago
Ayush Chaurasia
f2ca30a407
W&B: refactor W&B tables ( #5737 )
* update
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* reformat
* Single-line argparser argument
* Update README.md
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update README.md
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
4ca4aec46f
Do not save hyp.yaml and opt.yaml on evolve ( #5775 )
* Do not save hyp.yaml and opt.yaml on evolve
* Update general.py
3 years ago