caf7ad0500
Allow 3-point segments ( #6938 )
May resolve https://github.com/ultralytics/yolov5/issues/6931
hace 2 años
6dd82c0252
Move `git_describe()` to general.py ( #6918 )
* Move `git_describe()` to general.py
* Move `git_describe()` to general.py
hace 2 años
e6e36aac10
Update bytes to GB with bitshift ( #6886 )
hace 2 años
4728840745
Update `--cache disk` deprecate `*_npy/` dirs ( #6876 )
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Cleanup
* Cleanup
hace 2 años
8a66ebad44
Add `*.ts` to `VID_FORMATS` ( #6859 )
hace 2 años
601dbb83f0
AutoAnchor improved initialization robustness ( #6854 )
* Update AutoAnchor
* Update AutoAnchor
* [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>
hace 2 años
bcc92e2169
Update sweep.yaml ( #6825 )
* Update sweep.yaml
Changed focal loss gamma search range between 1 and 4
* Update sweep.yaml
lowered the min value to match default
hace 2 años
63ddb6f0d0
Update autoanchor.py ( #6794 )
* Update autoanchor.py
* Update autoanchor.py
hace 2 años
0f819919ad
Update EMA decay `tau` ( #6769 )
* Update EMA
* Update EMA
* ratio invert
* fix ratio invert
* fix2 ratio invert
* warmup iterations to 100
* ema_k
* implement tau
* implement tau
hace 2 años
c161557563
Default `OMP_NUM_THREADS=8` ( #6770 )
hace 2 años
2692e67c5f
Updated VOC hyperparameters ( #6732 )
* Update hyps
* Update hyp.VOC.yaml
* Update pathlib
* Update hyps
* Update hyps
* Update hyps
* Update hyps
hace 2 años
c43f135571
Update `nw` to `max(nd, 1)` ( #6714 )
hace 2 años
4de8b24881
Suppress `torch` AMP-CPU warnings ( #6706 )
This is a torch bug, but they seem unable or unwilling to fix it so I'm creating a suppression in YOLOv5.
Resolves https://github.com/ultralytics/yolov5/issues/6692
hace 2 años
0365379016
Fix floating point in number of workers `nw` ( #6701 )
Integer division by a float yields a (rounded) float. This causes
the dataloader to crash when creating a range.
hace 2 años
2e5c67e537
Robust `scipy.cluster.vq.kmeans` too few points ( #6668 )
* Handle `scipy.cluster.vq.kmeans` too few points
Resolves #6664
* Update autoanchor.py
* Cleanup
hace 2 años
1ff43702a8
Validate with 2x `--workers` single-GPU/CPU fix ( #6659 )
Fix for #6658 for single-GPU and CPU training use cases
hace 2 años
6769021901
Validate with 2x `--workers` ( #6658 )
hace 2 años
ca0a00784d
Fixed wandb logger KeyError ( #6637 )
hace 2 años
538930023c
Fix ConfusionMatrix scale `vmin=0.0` ( #6638 )
Fix attempt for https://github.com/ultralytics/yolov5/issues/6626
hace 2 años
a45e472358
YOLOv5 Export Benchmarks ( #6613 )
* Add benchmarks.py
* Update
* Add requirements
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* dataset autodownload from root
* Update
* Redirect to /dev/null
* sudo --help
* Cleanup
* Add exports pd df
* Updates
* Updates
* Updates
* Cleanup
* dir handling fix
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup
* Cleanup2
* Cleanup3
* Cleanup model_type
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
hace 2 años
96d8f86085
W&B: don't log media in evolve ( #6617 )
hace 2 años
51f7750aad
W&B: Improve resume stability ( #6611 )
* log best.pt metrics at train end
* update
* Update __init__.py
* flush callbacks when using evolve
* remember batch size on resuming
* Update train.py
* improve stability of resume
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
hace 2 años
b40bdd6b2d
YouTube dependency fix `youtube_dl==2020.12.2` ( #6612 )
Per https://github.com/ultralytics/yolov5/issues/5860#issuecomment-1035320018 by @hdnh2006
hace 2 años
cb2ad9f685
Fix `ROOT / data` when running W&B `log_dataset()` ( #6606 )
* Fix missing data folder when running log_dataset
* Use ROOT/'data'
* PEP8 whitespace
hace 2 años
c21da596f3
Fix `hyp_evolve.yaml` indexing bug ( #6604 )
* Fix `hyp_evolve.yaml` indexing bug
Bug caused hyp_evolve.yaml to display latest generation result rather than best generation result.
* Update plots.py
* Update general.py
* Update general.py
* Update general.py
hace 2 años
364b7c5111
Fix YouTube dislike button bug in `pafy` package ( #6603 )
Per https://github.com/ultralytics/yolov5/issues/6583#issuecomment-1034421945 by @alicera
hace 2 años
9c513ca629
Add `DATASETS_DIR` global in general.py ( #6578 )
hace 2 años
f627bc53f8
Fix `plot_labels()` colored histogram bug ( #6574 )
* Fix `plot_labels()` colored histogram bug
* Cleanup
hace 2 años
b57abb17f2
Move trainloader functions to class methods ( #6559 )
* Move trainloader functions to class methods
* results = ThreadPool(NUM_THREADS).imap(self.load_image, range(n))
* Cleanup
hace 2 años
cba4303d32
Fix 6 Flake8 issues ( #6541 )
* F541
* F821
* F841
* E741
* E302
* E722
* Apply suggestions from code review
* Update general.py
* Update datasets.py
* Update export.py
* Update plots.py
* Update plots.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
hace 2 años
c3e599cfda
Update val_batch*.jpg for Chinese fonts ( #6526 )
* Update plots for Chinese fonts
* make is_chinese() non-str safe
* Add global FONT
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update general.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
hace 2 años
6445a8137e
Resolve dataset paths ( #6489 )
hace 2 años
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
hace 2 años
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
hace 2 años
fe7de6a82d
Revert "Remove `dataset_stats()` autodownload capability ( #6303 )" ( #6442 )
This reverts commit 3119b2f27c
hace 2 años
3b7ac28ed1
Add `*.asf` video support ( #6436 )
hace 2 años
d5966c93f1
Namespace `VERBOSE` env variable to `YOLOv5_VERBOSE` ( #6428 )
* Verbose updates
* Verbose updates
hace 2 años
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>
hace 2 años
482af479c0
Add `detect.py` GIF video inference ( #6410 )
* Add detect.py GIF video inference
* Cleanup
hace 2 años
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>
hace 2 años
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>
hace 2 años
e1893c894a
`DEVICE_COUNT` instead of `WORLD_SIZE` to calculate `nw` ( #6324 )
hace 2 años
4e841b9b16
Reuse `de_parallel()` rather than `is_parallel()` ( #6354 )
hace 2 años
9708cf56ea
New environment variable `VERBOSE` ( #6353 )
New environment variable `VERBOSE`
hace 2 años
fd55271c04
Console corrupted -> corrupt ( #6338 )
* Console corrupted -> corrupt
Minor style changes.
* Update export.py
hace 2 años
3119b2f27c
Remove `dataset_stats()` autodownload capability ( #6303 )
* Remove `dataset_stats()` autodownload capability
@kalenmike security update per Slack convo
* Update datasets.py
hace 2 años
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
hace 2 años
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
hace 2 años
e7bf38277f
Fix `device` count check ( #6290 )
* Fix device count check()
* Update torch_utils.py
* Update torch_utils.py
* Update hubconf.py
hace 2 años
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
hace 2 años
Glenn Jocher
Louis Combaldieu
Samuel Yvon
imyhxy
Ayush Chaurasia
or-toledano
Jirka Borovec
toschi23
Jonathan Samelson
Matthias
sitecao
johnk2hawaii
Henry