2c3221844b
CLI `fire` prep updates ( #7229 )
* CLI fire prep updates
* revert unintentional TF export change
il y a 2 ans
c3d5ac151e
precommit: yapf ( #5494 )
* precommit: yapf
* align isort
* fix
# Conflicts:
# utils/plots.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update setup.cfg
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update setup.cfg
* Update setup.cfg
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update wandb_utils.py
* Update augmentations.py
* Update setup.cfg
* Update yolo.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update val.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* simplify colorstr
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* val run fix
* export.py last comma
* Update export.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update hubconf.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* PyTorch Hub tuple fix
* PyTorch Hub tuple fix2
* PyTorch Hub tuple fix3
* Update setup
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>
il y a 2 ans
3373aab56c
NMS unused variable fix ( #7161 )
* NMS unused variable fix
* Update general.py
il y a 2 ans
26bfd44465
Adjust NMS time limit warning to batch size ( #7156 )
il y a 2 ans
a4c661873f
Fix `detect.py --view-img` for non-ASCII paths ( #7093 )
* Update detect.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update general.py
* Update detect.py
* Update general.py
* Update general.py
* Update general.py
* Update general.py
* Update general.py
* Update general.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update general.py
* Update general.py
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>
il y a 2 ans
d115bbf509
Fix `cv2.imwrite` on non-ASCII paths ( #7139 )
* Fix imwrite on non-ASCII paths
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update general.py
* Update __init__.py
* Update __init__.py
* Update datasets.py
* Update hubconf.py
* Update detect.py
* [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>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
il y a 2 ans
bc3ed957ce
`np.fromfile()` Chinese image paths fix ( #6979 )
* :tada: :new: now can read Chinese image path.
use "cv2.imdecode(np.fromfile(f, np.uint8), cv2.IMREAD_COLOR)" instead of "cv2.imread(f)" for Chinese image path.
* Update datasets.py
* Update __init__.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
il y a 2 ans
c3ae4e4af6
Multi-threaded VisDrone and VOC downloads ( #7108 )
* Multi-threaded VOC download
* Update VOC.yaml
* Update
* Update general.py
* Update general.py
il y a 2 ans
6134ec5d94
Model summary `pathlib` fix ( #7104 )
Stems not working correctly for YOLOv5l with current .rstrip() implementation. After fix:
```
YOLOv5l summary: 468 layers, 46563709 parameters, 46563709 gradients, 109.3 GFLOPs
```
il y a 2 ans
ee0b3b2a95
Update benchmarks significant digits ( #7103 )
il y a 2 ans
05aae17333
`torch.split()` 1.7.0 compatibility fix ( #7102 )
* Update loss.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update loss.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
il y a 2 ans
a600baed8e
Update model summary to display model name ( #7101 )
il y a 2 ans
a2d617ece9
Update loss for FP16 `tobj` ( #7088 )
il y a 2 ans
6f128031d0
Update loss.py with `if self.gr < 1:` ( #7087 )
* Update loss.py with `if self.gr < 1:`
* Update loss.py
il y a 2 ans
9cd89b75cc
Fix2 `check_anchor_order()` in pixel-space not grid-space ( #7067 )
Follows https://github.com/ultralytics/yolov5/pull/7060 which provided only a partial solution to this issue. #7060 resolved occurences in yolo.py, this applies the same fix in autoanchor.py.
il y a 2 ans
178c109576
Add non-zero `da` `check_anchor_order()` condition ( #7066 )
il y a 2 ans
9e75cbf4c1
Update __init__.py ( #7065 )
il y a 2 ans
e278fd63ec
Update W&B message to `LOGGER.info()` ( #7064 )
il y a 2 ans
f327eee614
Fix `check_anchor_order()` in pixel-space not grid-space ( #7060 )
* Update `check_anchor_order()`
Use mean area per output layer for added stability.
* Check in pixel-space not grid-space fix
il y a 2 ans
529fbc1814
Use PIL to eliminate chroma subsampling in crops ( #7008 )
* use pillow to save higher-quality jpg (w/o color subsampling)
* Cleanup and doc issue
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
il y a 2 ans
9f4d11379b
Fix incomplete URL substring sanitation ( #7056 )
Resolves code scanning alert in https://github.com/ultralytics/yolov5/issues/7055
il y a 2 ans
b0ba101ac0
`ComputeLoss()` indexing/speed improvements ( #7048 )
* device as class attribute
* Update loss.py
* Update loss.py
* improve zeros
* tensor split
il y a 2 ans
7c6a33564a
fix: add default PIL font as fallback ( #7010 )
* fix: add default font as fallback
Add default font as fallback if the downloading of the Arial.ttf font
fails for some reason, e.g. no access to public internet.
* Update plots.py
Co-authored-by: Maximilian Strobel <Maximilian.Strobel@infineon.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
il y a 2 ans
3f634d43c8
Conditional `Timeout()` by OS (disable on Windows) ( #7013 )
* Conditional `Timeout()` by OS (disable on Windows)
* Update general.py
il y a 2 ans
c09fb2aa95
Update TQDM bar format ( #6988 )
il y a 2 ans
932dc78496
YOLOv5 Export Benchmarks for GPU ( #6963 )
* Add benchmarks.py GPU support
* Updates
* Updates
* Updates
* Updates
* Add --half
* Add TRT requirements
* Cleanup
* Add TF to warmup types
* Update export.py
* Update export.py
* Update benchmarks.py
il y a 2 ans
caf7ad0500
Allow 3-point segments ( #6938 )
May resolve https://github.com/ultralytics/yolov5/issues/6931
il y a 2 ans
6dd82c0252
Move `git_describe()` to general.py ( #6918 )
* Move `git_describe()` to general.py
* Move `git_describe()` to general.py
il y a 2 ans
e6e36aac10
Update bytes to GB with bitshift ( #6886 )
il y a 2 ans
4728840745
Update `--cache disk` deprecate `*_npy/` dirs ( #6876 )
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Cleanup
* Cleanup
il y a 2 ans
8a66ebad44
Add `*.ts` to `VID_FORMATS` ( #6859 )
il y a 2 ans
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>
il y a 2 ans
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
il y a 2 ans
63ddb6f0d0
Update autoanchor.py ( #6794 )
* Update autoanchor.py
* Update autoanchor.py
il y a 2 ans
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
il y a 2 ans
c161557563
Default `OMP_NUM_THREADS=8` ( #6770 )
il y a 2 ans
2692e67c5f
Updated VOC hyperparameters ( #6732 )
* Update hyps
* Update hyp.VOC.yaml
* Update pathlib
* Update hyps
* Update hyps
* Update hyps
* Update hyps
il y a 2 ans
c43f135571
Update `nw` to `max(nd, 1)` ( #6714 )
il y a 2 ans
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
il y a 2 ans
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.
il y a 2 ans
2e5c67e537
Robust `scipy.cluster.vq.kmeans` too few points ( #6668 )
* Handle `scipy.cluster.vq.kmeans` too few points
Resolves #6664
* Update autoanchor.py
* Cleanup
il y a 2 ans
1ff43702a8
Validate with 2x `--workers` single-GPU/CPU fix ( #6659 )
Fix for #6658 for single-GPU and CPU training use cases
il y a 2 ans
6769021901
Validate with 2x `--workers` ( #6658 )
il y a 2 ans
ca0a00784d
Fixed wandb logger KeyError ( #6637 )
il y a 2 ans
538930023c
Fix ConfusionMatrix scale `vmin=0.0` ( #6638 )
Fix attempt for https://github.com/ultralytics/yolov5/issues/6626
il y a 2 ans
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>
il y a 2 ans
96d8f86085
W&B: don't log media in evolve ( #6617 )
il y a 2 ans
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>
il y a 2 ans
b40bdd6b2d
YouTube dependency fix `youtube_dl==2020.12.2` ( #6612 )
Per https://github.com/ultralytics/yolov5/issues/5860#issuecomment-1035320018 by @hdnh2006
il y a 2 ans
cb2ad9f685
Fix `ROOT / data` when running W&B `log_dataset()` ( #6606 )
* Fix missing data folder when running log_dataset
* Use ROOT/'data'
* PEP8 whitespace
il y a 2 ans
Glenn Jocher
Jirka Borovec
Zengyf-CVer
RcINS
yeshanliu
Philip Gutjahr
Max Strobel
Louis Combaldieu
Samuel Yvon
imyhxy
Ayush Chaurasia
or-toledano