yolov5

Graphique des révisions

Auteur	SHA1	Message	Date
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	il y a 2 ans
Glenn Jocher	de9c25b35e	Use `export_formats()` in export.py (#6705 ) * Use `export_formats()` in export.py * list fix	il y a 2 ans
Raffaele Galliera	a297efc383	Edge TPU inference fix (#6686 ) * refactor: use edgetpu flag * fix: remove bitwise and assignation to tflite * Cleanup and fix tflite * Cleanup Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	il y a 2 ans
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.	il y a 2 ans
Glenn Jocher	56697c43ff	Fix CoreML P6 inference (#6700 ) * Fix CoreML P6 inference * [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
Glenn Jocher	3670d91b05	Fix `--evolve --bucket gs://...` (#6698 )	il y a 2 ans
Jiacong Fang	66e5d794c7	Fix TF exports >= 2GB (#6292 ) * Fix exporting saved_model: pb exceeds 2GB * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Replace TF v1.x API with TF v2.x API for saved_model export * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Clean up * Remove lambda in tf.function() * Revert "Remove lambda in tf.function()" to be compatible with TF v2.4 This reverts commit `46c7931f11`. * Fix for pre-commit.ci * Cleanup1 * Cleanup2 * Backwards compatibility update * Update common.py * Update common.py * Cleanup3 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
Glenn Jocher	07221f1591	FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6677 ) Revert to 21.10 on autobuild fail	il y a 2 ans
Glenn Jocher	18b5081166	FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6673 ) Reordered installation may help reduce resource usage in autobuild	il y a 2 ans
Glenn Jocher	25456284b4	Update Dockerfile reorder installs (#6672 ) Also `nvidia-tensorboard-plugin-dlprof`, `nvidia-tensorboard` are no longer installed in NVCR base.	il y a 2 ans
Glenn Jocher	09d263b376	FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6671 ) 22.10 returns 'no space left on device' error message. Seems like a bug at docker. Raised issue in https://github.com/docker/hub-feedback/issues/2209	il y a 2 ans
Glenn Jocher	bb054d5f55	FROM nvcr.io/nvidia/pytorch:22.01-py3 (#6670 )	il y a 2 ans
Glenn Jocher	9a8ebe671c	Update Dockerfile `torch==1.10.2+cu113` (#6669 )	il y a 2 ans
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	il y a 2 ans
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`.	il y a 2 ans
Glenn Jocher	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
Glenn Jocher	6769021901	Validate with 2x `--workers` (#6658 )	il y a 2 ans
Glenn Jocher	ee6c70ae43	Fix yolov3.yaml remove list (#6655 ) Per https://github.com/ultralytics/yolov3/issues/1887#issuecomment-1041135181	il y a 2 ans
imyhxy	ca0a00784d	Fixed wandb logger KeyError (#6637 )	il y a 2 ans
Glenn Jocher	538930023c	Fix ConfusionMatrix scale `vmin=0.0` (#6638 ) Fix attempt for https://github.com/ultralytics/yolov5/issues/6626	il y a 2 ans
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>	il y a 2 ans
Ayush Chaurasia	96d8f86085	W&B: don't log media in evolve (#6617 )	il y a 2 ans
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>	il y a 2 ans
Glenn Jocher	3eef80457f	Add YOLOv5n to Reproduce section (#6619 )	il y a 2 ans
Glenn Jocher	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
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	il y a 2 ans
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	il y a 2 ans
Glenn Jocher	a5c9057dcc	Update train.py	il y a 2 ans
Glenn Jocher	364b7c5111	Fix YouTube dislike button bug in `pafy` package (#6603 ) Per https://github.com/ultralytics/yolov5/issues/6583#issuecomment-1034421945 by @alicera	il y a 2 ans
Christoph Fürbacher	9cf80b7f60	return `opt` from `train.run()` (#6581 )	il y a 2 ans
Glenn Jocher	9c513ca629	Add `DATASETS_DIR` global in general.py (#6578 )	il y a 2 ans
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>	il y a 2 ans
Glenn Jocher	f627bc53f8	Fix `plot_labels()` colored histogram bug (#6574 ) * Fix `plot_labels()` colored histogram bug * Cleanup	il y a 2 ans
Glenn Jocher	22da63e30f	Fix zero-export handling with `if any(f):` (#6569 ) * Fix zero-export handling with `if any(f):` Partial fix for https://github.com/ultralytics/yolov5/issues/6563 * Cleanup	il y a 2 ans
Glenn Jocher	006eb40ab7	Improved AutoBatch DDP error message (#6568 ) * Improved AutoBatch DDP error message * Cleanup	il y a 2 ans
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	il y a 2 ans
Glenn Jocher	dc7e093071	Edge TPU TF imports fix (#6542 ) * Edge TPU TF imports fix Fix for https://github.com/ultralytics/yolov5/issues/6535#issuecomment-1030631526 * Update common.py	il y a 2 ans
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>	il y a 2 ans
Jirka Borovec	e1a6a0b100	flake8: code meanings (#6481 )	il y a 2 ans
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>	il y a 2 ans
greg2451	8fcdf3b60b	Fixing minor multi-streaming issues with TensoRT engine (#6504 ) * Update batch-size in model.warmup() + indentation for logging inference results * These changes are in response to PR comments * [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
Glenn Jocher	079b36d72b	Edge TPU `tf.lite.experimental.load_delegate` fix (#6536 ) * Edge TPU `tf.lite.experimental.load_delegate` fix Fix attempt for #6535 * [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
Glenn Jocher	9bc72a3ac2	Edge TPU export 'list index out of range' fix (#6533 )	il y a 2 ans
Glenn Jocher	8d05716b66	Edge TPU compiler `sudo` fix (#6531 ) * Edge TPU compiler sudo fix Allows for auto-install of Edge TPU compiler on non-sudo systems like the YOLOv5 Docker image. @kalenmike * Update export.py * Update export.py * Update export.py	il y a 2 ans
Glenn Jocher	a82292ec53	Social icons after text (#6473 ) * Social icons after text * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update README.md Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>	il y a 2 ans
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>	il y a 2 ans
Glenn Jocher	cb40c9afda	TODO issues exempt from stale action (#6530 )	il y a 2 ans
Glenn Jocher	19e0208fc9	Update hyp.scratch-high.yaml (#6525 ) Update `lrf: 0.1`, tested on YOLOv5x6 to 55.0 mAP@0.5:0.95, slightly higher than current.	il y a 2 ans
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>	il y a 2 ans
Glenn Jocher	4c40933266	Suppress export.run() TracerWarnings (#6499 ) Suppresses warnings when calling export.run() directly, not just CLI python export.py. Also adds Requirements examples for CPU and GPU backends	il y a 2 ans

1 2 3 4 5 ...

1809 Révisions (4de8b2488133b1f7d4db1e64c5eb2063c494b55a) Toutes les branches Chercher

1809 Révisions (4de8b2488133b1f7d4db1e64c5eb2063c494b55a)

Toutes les branches