Glenn Jocher
2c3221844b
CLI `fire` prep updates ( #7229 )
...
* CLI fire prep updates
* revert unintentional TF export change
2022-03-31 17:11:43 +02:00
Jirka Borovec
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>
2022-03-31 16:52:34 +02:00
Glenn Jocher
3373aab56c
NMS unused variable fix ( #7161 )
...
* NMS unused variable fix
* Update general.py
2022-03-26 16:52:58 +01:00
Glenn Jocher
26bfd44465
Adjust NMS time limit warning to batch size ( #7156 )
2022-03-26 11:45:28 +01:00
Zengyf-CVer
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>
2022-03-25 13:40:55 +01:00
RcINS
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>
2022-03-25 13:25:30 +01:00
yeshanliu
bc3ed957ce
`np.fromfile()` Chinese image paths fix ( #6979 )
...
* 🎉 🆕 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>
2022-03-23 15:35:15 +01:00
Glenn Jocher
c3ae4e4af6
Multi-threaded VisDrone and VOC downloads ( #7108 )
...
* Multi-threaded VOC download
* Update VOC.yaml
* Update
* Update general.py
* Update general.py
2022-03-23 01:19:37 +01:00
Glenn Jocher
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
```
2022-03-22 20:05:07 +01:00
Glenn Jocher
ee0b3b2a95
Update benchmarks significant digits ( #7103 )
2022-03-22 18:02:35 +01:00
Glenn Jocher
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>
2022-03-22 17:36:05 +01:00
Glenn Jocher
a600baed8e
Update model summary to display model name ( #7101 )
2022-03-22 15:41:19 +01:00
Glenn Jocher
a2d617ece9
Update loss for FP16 `tobj` ( #7088 )
2022-03-21 19:18:34 +01:00
Glenn Jocher
6f128031d0
Update loss.py with `if self.gr < 1:` ( #7087 )
...
* Update loss.py with `if self.gr < 1:`
* Update loss.py
2022-03-21 18:35:36 +01:00
Glenn Jocher
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.
2022-03-20 18:55:13 +01:00
Glenn Jocher
178c109576
Add non-zero `da` `check_anchor_order()` condition ( #7066 )
2022-03-20 18:37:27 +01:00
Glenn Jocher
9e75cbf4c1
Update __init__.py ( #7065 )
2022-03-20 18:17:04 +01:00
Glenn Jocher
e278fd63ec
Update W&B message to `LOGGER.info()` ( #7064 )
2022-03-20 18:15:22 +01:00
Glenn Jocher
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
2022-03-20 16:27:51 +01:00
Philip Gutjahr
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>
2022-03-20 15:46:29 +01:00
Glenn Jocher
9f4d11379b
Fix incomplete URL substring sanitation ( #7056 )
...
Resolves code scanning alert in https://github.com/ultralytics/yolov5/issues/7055
2022-03-20 14:18:05 +01:00
Glenn Jocher
b0ba101ac0
`ComputeLoss()` indexing/speed improvements ( #7048 )
...
* device as class attribute
* Update loss.py
* Update loss.py
* improve zeros
* tensor split
2022-03-20 01:04:48 +01:00
Max Strobel
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>
2022-03-17 16:37:09 +01:00
Glenn Jocher
3f634d43c8
Conditional `Timeout()` by OS (disable on Windows) ( #7013 )
...
* Conditional `Timeout()` by OS (disable on Windows)
* Update general.py
2022-03-16 15:33:54 +01:00
Glenn Jocher
c09fb2aa95
Update TQDM bar format ( #6988 )
2022-03-15 16:32:56 +01:00
Glenn Jocher
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
2022-03-14 15:07:13 +01:00
Glenn Jocher
caf7ad0500
Allow 3-point segments ( #6938 )
...
May resolve https://github.com/ultralytics/yolov5/issues/6931
2022-03-10 18:41:47 +01:00
Glenn Jocher
6dd82c0252
Move `git_describe()` to general.py ( #6918 )
...
* Move `git_describe()` to general.py
* Move `git_describe()` to general.py
2022-03-09 18:22:53 +01:00
Glenn Jocher
e6e36aac10
Update bytes to GB with bitshift ( #6886 )
2022-03-07 19:26:37 +01:00
Glenn Jocher
4728840745
Update `--cache disk` deprecate `*_npy/` dirs ( #6876 )
...
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Updates
* Cleanup
* Cleanup
2022-03-06 16:16:17 +01:00
Glenn Jocher
8a66ebad44
Add `*.ts` to `VID_FORMATS` ( #6859 )
2022-03-04 14:10:13 +01:00
Glenn Jocher
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>
2022-03-04 10:32:18 +01:00
Louis Combaldieu
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
2022-03-04 09:39:23 +01:00
Glenn Jocher
63ddb6f0d0
Update autoanchor.py ( #6794 )
...
* Update autoanchor.py
* Update autoanchor.py
2022-02-26 19:15:12 +01:00
Glenn Jocher
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
2022-02-25 12:33:09 +01:00
Glenn Jocher
c161557563
Default `OMP_NUM_THREADS=8` ( #6770 )
2022-02-25 00:09:14 +01:00
Glenn Jocher
2692e67c5f
Updated VOC hyperparameters ( #6732 )
...
* Update hyps
* Update hyp.VOC.yaml
* Update pathlib
* Update hyps
* Update hyps
* Update hyps
* Update hyps
2022-02-22 12:11:18 +01:00
Glenn Jocher
c43f135571
Update `nw` to `max(nd, 1)` ( #6714 )
2022-02-20 21:47:46 +01:00
Glenn Jocher
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
2022-02-19 16:48:33 +01:00
Samuel Yvon
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.
2022-02-18 21:06:21 +01:00
Glenn Jocher
2e5c67e537
Robust `scipy.cluster.vq.kmeans` too few points ( #6668 )
...
* Handle `scipy.cluster.vq.kmeans` too few points
Resolves #6664
* Update autoanchor.py
* Cleanup
2022-02-17 12:55:03 +01:00
Glenn Jocher
1ff43702a8
Validate with 2x `--workers` single-GPU/CPU fix ( #6659 )
...
Fix for #6658 for single-GPU and CPU training use cases
2022-02-16 14:50:37 +01:00
Glenn Jocher
6769021901
Validate with 2x `--workers` ( #6658 )
2022-02-16 14:44:44 +01:00
imyhxy
ca0a00784d
Fixed wandb logger KeyError ( #6637 )
2022-02-14 17:08:36 +01:00
Glenn Jocher
538930023c
Fix ConfusionMatrix scale `vmin=0.0` ( #6638 )
...
Fix attempt for https://github.com/ultralytics/yolov5/issues/6626
2022-02-14 14:54:55 +01:00
Glenn Jocher
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>
2022-02-12 16:05:43 +01:00
Ayush Chaurasia
96d8f86085
W&B: don't log media in evolve ( #6617 )
2022-02-12 13:04:50 +01:00
Ayush Chaurasia
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>
2022-02-12 13:03:09 +01:00
Glenn Jocher
b40bdd6b2d
YouTube dependency fix `youtube_dl==2020.12.2` ( #6612 )
...
Per https://github.com/ultralytics/yolov5/issues/5860#issuecomment-1035320018 by @hdnh2006
2022-02-11 13:46:20 +01:00
or-toledano
cb2ad9f685
Fix `ROOT / data` when running W&B `log_dataset()` ( #6606 )
...
* Fix missing data folder when running log_dataset
* Use ROOT/'data'
* PEP8 whitespace
2022-02-10 16:17:47 +01:00
Glenn Jocher
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
2022-02-10 14:51:29 +01:00
Glenn Jocher
364b7c5111
Fix YouTube dislike button bug in `pafy` package ( #6603 )
...
Per https://github.com/ultralytics/yolov5/issues/6583#issuecomment-1034421945 by @alicera
2022-02-10 13:13:44 +01:00
Glenn Jocher
9c513ca629
Add `DATASETS_DIR` global in general.py ( #6578 )
2022-02-08 22:20:44 +01:00
Glenn Jocher
f627bc53f8
Fix `plot_labels()` colored histogram bug ( #6574 )
...
* Fix `plot_labels()` colored histogram bug
* Cleanup
2022-02-08 14:03:50 +01:00
Glenn Jocher
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
2022-02-07 18:59:26 +01:00
Jirka Borovec
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>
2022-02-05 15:22:59 +01:00
Glenn Jocher
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>
2022-02-03 19:09:24 +01:00
Glenn Jocher
6445a8137e
Resolve dataset paths ( #6489 )
2022-02-01 15:54:51 +01:00
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
2022-01-28 08:18:01 +01:00
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
2022-01-26 23:47:00 +01:00
Glenn Jocher
fe7de6a82d
Revert "Remove `dataset_stats()` autodownload capability ( #6303 )" ( #6442 )
...
This reverts commit 3119b2f27c
.
2022-01-26 19:10:59 +01:00
toschi23
3b7ac28ed1
Add `*.asf` video support ( #6436 )
2022-01-26 14:26:02 +01:00
Glenn Jocher
d5966c93f1
Namespace `VERBOSE` env variable to `YOLOv5_VERBOSE` ( #6428 )
...
* Verbose updates
* Verbose updates
2022-01-25 17:57:27 -05:00
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>
2022-01-25 07:21:06 -08:00
Glenn Jocher
482af479c0
Add `detect.py` GIF video inference ( #6410 )
...
* Add detect.py GIF video inference
* Cleanup
2022-01-24 13:11:11 -08:00
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>
2022-01-22 16:37:21 -10:00
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>
2022-01-20 13:22:47 -10:00
sitecao
e1893c894a
`DEVICE_COUNT` instead of `WORLD_SIZE` to calculate `nw` ( #6324 )
2022-01-20 13:06:26 -10:00
imyhxy
4e841b9b16
Reuse `de_parallel()` rather than `is_parallel()` ( #6354 )
2022-01-20 10:50:17 -10:00
johnk2hawaii
9708cf56ea
New environment variable `VERBOSE` ( #6353 )
...
New environment variable `VERBOSE`
2022-01-19 14:32:19 -10:00
Glenn Jocher
fd55271c04
Console corrupted -> corrupt ( #6338 )
...
* Console corrupted -> corrupt
Minor style changes.
* Update export.py
2022-01-18 10:49:26 -10:00
Glenn Jocher
3119b2f27c
Remove `dataset_stats()` autodownload capability ( #6303 )
...
* Remove `dataset_stats()` autodownload capability
@kalenmike security update per Slack convo
* Update datasets.py
2022-01-18 10:04:49 -10:00
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
2022-01-14 15:48:15 -10:00
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
2022-01-14 10:11:06 -10:00
Glenn Jocher
e7bf38277f
Fix `device` count check ( #6290 )
...
* Fix device count check()
* Update torch_utils.py
* Update torch_utils.py
* Update hubconf.py
2022-01-13 21:23:03 -10:00
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
2022-01-13 12:39:42 -10:00
Glenn Jocher
9e9219fe17
Fix `nan`-robust stream FPS ( #6198 )
...
Fix for Webcam stop working suddenly (Issue #6197 )
2022-01-04 19:32:42 -08:00
Glenn Jocher
b4a29b5a8d
Update NMS `max_wh=7680` for 8k images ( #6178 )
2022-01-03 10:54:52 -08:00
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>
2021-12-30 13:59:29 -08:00
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
2021-12-30 13:47:53 -08:00
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>
2021-12-30 11:43:19 -08:00
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>
2021-12-23 14:23:50 +01:00
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>
2021-12-23 11:53:00 +01:00
Glenn Jocher
b8a4babd60
Simplify `set_logging()` indexing ( #6042 )
2021-12-20 17:42:52 +01:00
Glenn Jocher
0db9d5b6a2
Kaggle `LOGGER` fix ( #6041 )
2021-12-20 17:30:46 +01:00
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>
2021-12-17 14:59:46 +01:00
Glenn Jocher
da9a1b719b
Allow `--weights URL` ( #5991 )
2021-12-15 15:27:08 +01:00
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
2021-12-15 14:42:23 +01:00
Glenn Jocher
c9a46a60e0
Update callbacks.py with `__init__()` ( #5979 )
...
Add __init__() function.
2021-12-14 15:47:49 +01:00
Glenn Jocher
8f875d93a2
Refactor NUM_THREADS ( #5954 )
2021-12-11 14:32:18 +01:00
iumyx2612
a42af30d8a
Update `strip_optimizer()` ( #5949 )
...
Replace 'training_result' with 'best_fitness' in strip_optimizer() to match key with ckpt from train.py
2021-12-10 18:06:27 +01:00
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>
2021-12-10 15:27:20 +01:00
Glenn Jocher
c77a5a84e3
Absolute '/content/sample_data' ( #5922 )
2021-12-08 17:31:31 +01:00
Glenn Jocher
a3d5f1d3e3
Revert "Update `plot_lr_scheduler()` ( #5864 )" ( #5920 )
...
This reverts commit 360eec6910
.
2021-12-08 16:46:24 +01:00
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>
2021-12-08 14:57:03 +01:00
Ayush Chaurasia
3f152e5807
Update wandb_utils.py ( #5908 )
2021-12-07 16:09:11 +01:00
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
2021-12-04 15:00:07 +01:00
Glenn Jocher
d885799c71
Update `nl` after `cutout()` ( #5873 )
2021-12-03 15:28:14 +01:00
Can
360eec6910
Update `plot_lr_scheduler()` ( #5864 )
...
shallow copy modify originals
2021-12-03 13:37:45 +01:00
Glenn Jocher
00e308f7be
Update TorchScript suffix to `*.torchscript` ( #5856 )
2021-12-02 16:06:45 +01:00