yolov5

Commit Graph

Author	SHA1	Message	Date
Glenn Jocher	4bb7eb8b84	Dynamic normalization layer selection (#7392 ) * Dynamic normalization layer selection Based on actual available layers. Torch 1.7 compatible, resolves https://github.com/ultralytics/yolov5/issues/7381 * Update train.py	2 years ago
Vardan Agarwal	fa569cdae5	Add support for different normalization layers (#7377 ) * Add support for different normalization layers. * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Cleanup 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>	2 years ago
Glenn Jocher	bd2dda8e64	Update optimizer param group strategy (#7376 ) * Update optimizer param group strategy Avoid empty lists on missing BathNorm2d models as in https://github.com/ultralytics/yolov5/issues/7375 * fix init	2 years ago
Glenn Jocher	5f941a84ef	Print dataset scan only `if RANK in (-1, 0)` (#7337 ) * Print dataset scan only `if RANK in (-1, 0)` * [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	245d6459a9	Add callbacks (#7315 ) * Add `on_train_start()` callback * Update * Update	2 years ago
Glenn Jocher	f735458987	Use `tqdm.auto` (#7311 )	2 years ago
Glenn Jocher	2c3221844b	CLI `fire` prep updates (#7229 ) * CLI fire prep updates * revert unintentional TF export change	2 years ago
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>	2 years ago
Glenn Jocher	df9008ee69	Add train.py `--name cfg` option (#7202 ) Automatically names run as --cfg argument	2 years ago
Glenn Jocher	47f265dde2	Update min warmup iterations from 1k to 100 (#6768 )	2 years ago
Glenn Jocher	2692e67c5f	Updated VOC hyperparameters (#6732 ) * Update hyps * Update hyp.VOC.yaml * Update pathlib * Update hyps * Update hyps * Update hyps * Update hyps	2 years ago
Glenn Jocher	a936f5f219	Switch default LR scheduler from cos to linear (#6729 ) * Switch default LR scheduler from cos to linear Based on empirical results of training both ways on all YOLOv5 models. * linear bug fix	2 years ago
Glenn Jocher	3670d91b05	Fix `--evolve --bucket gs://...` (#6698 )	2 years ago
Glenn Jocher	7b80545e8e	Add `--cache val` (#6663 ) New `--cache val` argument will cache validation set only into RAM. Should help multi-GPU training speeds without consuming as much RAM as full `--cache ram`.	2 years ago
Glenn Jocher	6769021901	Validate with 2x `--workers` (#6658 )	2 years ago
Glenn Jocher	a5c9057dcc	Update train.py	2 years ago
Christoph Fürbacher	9cf80b7f60	return `opt` from `train.run()` (#6581 )	2 years ago
Matthias Vadcard	f40854b61b	Allow custom` --evolve` project names (#6567 ) * Update train.py As see in #6463, modification on train in evolve process to allow custom save directory. * fix val * PEP8 whitespace around operator * Cleanup Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	2 years ago
Glenn Jocher	006eb40ab7	Improved AutoBatch DDP error message (#6568 ) * Improved AutoBatch DDP error message * Cleanup	2 years ago
bilzard	aff0281969	Load checkpoint on CPU instead of on GPU (#6516 ) * Load checkpoint on CPU instead of on GPU * refactor: simplify code * Cleanup * Update train.py Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	2 years ago
Ayush Chaurasia	b73c62ebc5	W&B: Remember batchsize on resuming (#6512 ) * log best.pt metrics at train end * update * Update __init__.py * flush callbacks when using evolve * remember batch size on resuming * Update train.py Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	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
Otfot	db1f83be63	Fix `train.py` parameter groups desc error (#6318 ) * Fix `train.py` parameter groups desc error * Cleanup Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	2 years ago
Glenn Jocher	33a67b4918	Update P2-P7 `models/hub` variants (#6230 ) * Update p2-p7 `models/hub` variants * Update common.py * AutoAnchor camelcase corrections	2 years ago
Glenn Jocher	ec4b6dd2a3	Update export format docstrings (#6151 ) * Update export documentation * Cleanup * Update export.py * Update README.md * Update README.md * Update README.md * Update README.md * Update README.md * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update README.md * Update README.md * Update README.md * Update train.py * Update train.py Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>	2 years ago
bilzard	e1dc894364	Enable AdamW optimizer (#6152 )	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
Glenn Jocher	dc54ed5763	`--freeze` fix (#6044 ) Fix for https://github.com/ultralytics/yolov5/issues/6038	2 years ago
Felix You	361705d9be	Multi-layer capable `--freeze` argument (#6019 ) * support specfiy multiple frozen layers * fix bug * Cleanup Freeze section * Cleanup argument Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	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>	2 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>	2 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	2 years ago
Glenn Jocher	47fac9ff73	Update train.py comment to 'Model attributes' (#5670 )	2 years ago
Werner Duvaud	09d170381c	Default DataLoader `shuffle=True` for training (#5623 ) * Fix shuffle DataLoader argument * Add shuffle argument * Disable shuffle when rect * Cleanup, add rect warning * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Cleanup2 * Cleanup3 Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>	2 years ago
Glenn Jocher	7473f0f95d	DDP `WORLD_SIZE`-safe dataloader workers (#5631 ) * WORLD_SIZE-safe workers * Update with DDP comment	2 years ago
Glenn Jocher	61c50199a2	Update train, val `tqdm` to fixed width (#5367 ) * Update tqdm for fixed width * Update val.py * Update val.py * Try ncols= in train.py * NCOLS * NCOLS * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * bar_format * position 0 leave true * exp0 * auto * auto * Cleanup * Cleanup * Cleanup Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>	2 years ago
Glenn Jocher	79bca2bf64	`LOGGER` consolidation (#5569 ) * Logger consolidation * [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	e189fa15ea	`intersect_dicts()` in hubconf.py fix (#5542 )	2 years ago
Glenn Jocher	bcc085d83f	Common `is_coco` logic betwen train.py and val.py (#5521 )	2 years ago
Yonghye Kwon	99a45bad81	Write date in checkpoint file (#5514 ) * write date in checkpoint file write date in checkpoint file * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * isoformat 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>	2 years ago
Jirka Borovec	0155548384	precommit: isort (#5493 ) * precommit: isort * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update isort config * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update name 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>	2 years ago
Glenn Jocher	5866646cc8	Fix float zeros format (#5491 ) * Fix float zeros format * 255 to integer	2 years ago
Glenn Jocher	7b1f7aec46	Update `get_loggers()` (#4854 ) * Update `set_logging()` * Update export.py * pre-commit fixes * Update LoadImages * Update LoadStreams * Update print_args * Single LOGGER definition * yolo.py fix Co-authored-by: pre-commit <pre-commit@example.com>	2 years ago
Glenn Jocher	a4fece8c14	DDP `nl` fix (#5332 )	2 years ago
Glenn Jocher	ca19df5f7f	Add `autobatch` feature for best `batch-size` estimation (#5092 ) * Autobatch * fix mem * fix mem2 * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update * Update train.py * print result * Cleanup print result * swap fix in call * to 64 * use total * fix * fix * fix * fix * fix * Update * Update * Update * Update * Update * Update * Update * Cleanup printing * Update final printout * Update autobatch.py * Update autobatch.py * Update autobatch.py	2 years ago
Glenn Jocher	752d97df2c	Add `on_fit_epoch_end` callback (#5232 ) * Add `on_fit_epoch_end` callback * Add results to train * Update __init__.py	2 years ago
Malte Lorbach	153873e9e4	Fix `ROOT` as relative path (#5129 ) * use os.path.relpath instead of relative_to * use os.path.relpath instead of relative_to * Remove os.path from val.py * Remove os.path from train.py * Update detect.py import to os * Update export.py import to os Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	2 years ago
Nan	4a6dfffdaa	Pass `LOCAL_RANK` to `torch_distributed_zero_first()` (#5114 ) Co-authored-by: qiningonline <qiningonline@gmail.com>	2 years ago
Glenn Jocher	070af88108	Fix `yaml.safe_load()` ignore emoji errors (#5060 )	3 years ago

... 2 3 4 5 6 ...

365 Commits (master)