yolov5

Commit Graph

Tekijä	SHA1	Viesti	Päivämäärä
Glenn Jocher	c215878f11	YOLOv5 Apple Metal Performance Shader (MPS) support (#7878 ) * Apple Metal Performance Shader (MPS) device support Following https://pytorch.org/blog/introducing-accelerated-pytorch-training-on-mac/ Should work with Apple M1 devices with PyTorch nightly installed with command `--device mps`. Usage examples: ```bash python train.py --device mps python detect.py --device mps python val.py --device mps ``` * Update device strategy to fix MPS issue	2 vuotta sitten
xylieong	27911dc824	OpenVINO metadata fix2 (#7954 ) * Bug Fixed: OpenVINO metadata * Bug Fixed: OpenVINO metadata * Update export.py * Update export.py Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	2 vuotta sitten
Glenn Jocher	0dd66e2dc7	OpenVINO metadata fix (#7952 ) * Rename OpenVINO meta.yaml to model name * Rename OpenVINO meta.yaml to model name * Rename OpenVINO meta.yaml to model name * fix	2 vuotta sitten
xylieong	a3a652c933	Add OpenVINO metadata to export (#7947 ) * Write .yaml file when exporting model to openvino Write .yaml file automatically when exporting model to openvino to be used during inference * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update export.py * Update export.py * Load metadata on inference * Update common.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>	2 vuotta sitten
Glenn Jocher	eb1217f3ac	Add PyTorch AMP check (#7917 ) * Add PyTorch AMP check * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Cleanup * Cleanup * Cleanup * Robust for DDP * Fixes * Add amp enabled boolean to check_train_batch_size * Simplify * space to prefix Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>	2 vuotta sitten
Glenn Jocher	5774a1514d	Add `DWConvTranspose2d()` module (#7881 ) * Add DWConvTranspose2d() module * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Add DWConvTranspose2d() module * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Fix * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Fix * Fix Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>	2 vuotta sitten
Glenn Jocher	1b8e70f037	Add TFDWConv() `depth_multiplier` (#7858 ) Enabled grouped non c1 == c2 convolutions in TF YOLOv5 models.	2 vuotta sitten
Glenn Jocher	b52fd48b4a	TFDWConv() `depthwise_initializer` fix (#7845 )	2 vuotta sitten
Nanohana	05d4fc9db4	Replace `openvino-dev` with OpenVINO Runtime inference (#7843 ) * Uses OpenVINO runtime instead of openvino-dev * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * export with openvino package * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Revert export.py * Update common.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>	2 vuotta sitten
Glenn Jocher	2b64b45cd8	Fix TFDWConv() `c1 == c2` check (#7842 )	2 vuotta sitten
Glenn Jocher	fb7fa5be8b	New TensorFlow `TFCrossConv()` module (#7827 ) * New TensorFlow `TFCrossConv()` module * Move from experimental to common * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Add C3x * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Add to C3x to yolo.py * Add to C3x to tf.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * TFC3x bug fix * TFC3x bug fix * TFC3x bug fix * Add TFDWConv g==c1==c2 check * Add comment * Update tf.py * [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 vuotta sitten
Glenn Jocher	4d59f65db5	New TensorFlow `TFDWConv()` module (#7824 ) * New TensorFlow `TFDWConv()` module Analog to DWConv() module: `8aa2085a7e/models/common.py (L53-L57)` * Fix and new activations() function * Update tf.py	2 vuotta sitten
Glenn Jocher	8aa2085a7e	Refactor modules (#7823 )	2 vuotta sitten
Glenn Jocher	9d8ed37df7	Rename `utils/datasets.py` > `utils/dataloaders.py` (#7799 )	2 vuotta sitten
Glenn Jocher	d059d1da03	Report fused model summary by default (#7722 )	2 vuotta sitten
Glenn Jocher	950a85d9f6	TensorRT PyTorch Hub inference fix (#7560 ) Solution proposed in https://github.com/ultralytics/yolov5/issues/7128 to TRT PyTorch Hub CUDA illegal memory errors.	2 vuotta sitten
Glenn Jocher	c16671fc74	Add `print(results)` override for PyTorch Hub results (#7559 )	2 vuotta sitten
Glenn Jocher	be67572279	Enable `results.print()` when `_verbose=False` (#7558 ) Follows implementation of _verbose flag for PyTorch Hub models. Currently these are so silent that result.print() does nothing 😂	2 vuotta sitten
Glenn Jocher	3f3852e2ff	Fix val.py Ensemble() (#7490 )	2 vuotta sitten
rglkt	aa542ce6a6	DetectMultiBackend() default `stride=32` (#7342 ) * set common default stride as 32 * restore default stride, and set it on argument optional * fix wrong use of opt * fix missing parameter of stride * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * fix format of parameters * Update val.py * Update common.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>	2 vuotta sitten
Glenn Jocher	3bb233a7fb	Add ONNX export metadata (#7353 )	2 vuotta sitten
Glenn Jocher	302b00b5f4	Update `_make_grid()` (#7346 )	2 vuotta sitten
Nrupatunga	b7faeda0f2	Fix Tf export for BottleneckCSP (#7330 )	2 vuotta sitten
Glenn Jocher	5f97001ed4	Context manager `open(file) as f` fixes (#7289 ) * Flask context manager `open()` fix * Additional read context manager fixes	2 vuotta sitten
pre-commit-ci[bot]	7882950577	[pre-commit.ci] pre-commit suggestions (#7279 ) * [pre-commit.ci] pre-commit suggestions updates: - [github.com/asottile/pyupgrade: v2.31.0 → v2.31.1](https://github.com/asottile/pyupgrade/compare/v2.31.0...v2.31.1) - [github.com/pre-commit/mirrors-yapf: v0.31.0 → v0.32.0](https://github.com/pre-commit/mirrors-yapf/compare/v0.31.0...v0.32.0) * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update yolo.py * Update activations.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update activations.py * Update tf.py * Update tf.py * [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 vuotta sitten
Glenn Jocher	ea72b84f5e	Integrate offset into grid (#7262 ) Eliminate 1 op during training and inference.	2 vuotta sitten
Glenn Jocher	8bc839ed8e	TorchScript single-output fix (#7261 )	2 vuotta sitten
Glenn Jocher	05cf0d1a44	Export single output only (#7259 ) * Update * Update * [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 vuotta sitten
Glenn Jocher	ad0e4d5d19	`torch.split()` replace slicing on out-of-place inference (#7258 )	2 vuotta sitten
Glenn Jocher	4f839b7970	Refactor out-of-place `Detect()` for reduced ops (#7257 )	2 vuotta sitten
Glenn Jocher	2c3221844b	CLI `fire` prep updates (#7229 ) * CLI fire prep updates * revert unintentional TF export change	2 vuotta sitten
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 vuotta sitten
Glenn Jocher	cf4f3c3455	yolo.py profiling updates (#7178 ) * yolo.py profiling updates * [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 vuotta sitten
Glenn Jocher	ee77632393	Revert `C3()` change (#7172 )	2 vuotta sitten
Glenn Jocher	1832264dd6	Update	2 vuotta sitten
Glenn Jocher	b2194b9015	`yolo.py --profile` updates (#7170 )	2 vuotta sitten
Glenn Jocher	7830e91b9a	`yolo.py --profile` default GPU batch size 16	2 vuotta sitten
Glenn Jocher	e19f87eb4b	Sidestep `os.path.relpath()` Windows bug (#7158 ) * Sidestep os.path.relpath() Windows bug os.path.relpath() seems to have a major bug on Windows due to Windows horrible path handling. This fix attempts to sidestep the issue. ``` File "C:\Users\mkokg/.cache\torch\hub\ultralytics_yolov5_master\export.py", line 64, in ROOT = Path(os.path.relpath(ROOT, Path.cwd())) # relative File "C:\Users\mkokg\AppData\Local\Programs\Python\Python310\lib\ntpath.py", line 718, in relpath raise ValueError("path is on mount %r, start on mount %r" % ( ValueError: path is on mount 'C:', start on mount 'D:' ``` * Update yolo.py * Update yolo.py * Update yolo.py * Update export.py	2 vuotta sitten
Glenn Jocher	a0a4adf6de	Add PyTorch Hub `results.save(labels=False)` option (#7129 ) Resolves https://github.com/ultralytics/yolov5/issues/388#issuecomment-1077121821	2 vuotta sitten
Glenn Jocher	ecc2c7ba73	Remove named arguments where possible (#7105 ) * Remove named arguments where possible Speed improvements. * Update yolo.py * Update yolo.py * Update yolo.py	2 vuotta sitten
Glenn Jocher	9b771a3e71	Revert "Update detect.py non-inplace with `y.tensor_split()` (#7062 )" (#7074 ) This reverts commit `d5e363f29d`.	2 vuotta sitten
Glenn Jocher	0529b77232	Update common.py lists for tuples (#7063 ) Improved profiling.	2 vuotta sitten
Glenn Jocher	d5e363f29d	Update detect.py non-inplace with `y.tensor_split()` (#7062 )	2 vuotta sitten
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	2 vuotta sitten
Mrinal Jain	4effd064b1	Consistent saved_model output format (#7032 )	2 vuotta sitten
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	2 vuotta sitten
Glenn Jocher	2d45de617e	Model `ema` key backward compatibility fix (#6972 ) Fix for older model loading issue in `d3d9cbce22 (commitcomment-68622388)`	2 vuotta sitten
paradigm	c13d4ce7ef	EdgeTPU optimizations (#6808 ) * removed transpose op for better edgetpu support * fix for training case * enabled experimental new quantizer flag * precalculate add and mul ops at compile time Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	2 vuotta sitten
Glenn Jocher	701e1177ac	Tensor initialization on device improvements (#6959 ) * Update common.py speed improvements Eliminate .to() ops where possible for reduced data transfer overhead. Primarily affects warmup and PyTorch Hub inference. * Updates * Updates * Update detect.py * Update val.py	2 vuotta sitten
Glenn Jocher	52c1399fdc	DetectMultiBackend() return `device` update (#6958 ) Fixes ONNX validation that returns outputs on CPU.	2 vuotta sitten

1 2 3 4 5 ...

352 Commitit (bf1a6dda7b270eec066615646b2d30ec601b918a)