yolov5

Revīziju grafs

Autors	SHA1	Ziņojums	Datums
Glenn Jocher	f00071416f	Refactor collections and fstrings (#7821 ) * Update torch_utils.py * Additional code refactoring * tuples to sets * Cleanup	pirms 2 gadiem
Glenn Jocher	9d8ed37df7	Rename `utils/datasets.py` > `utils/dataloaders.py` (#7799 )	pirms 2 gadiem
Glenn Jocher	d4ea61e043	FROM nvcr.io/nvidia/pytorch:22.04-py3 (#7680 ) * FROM nvcr.io/nvidia/pytorch:22.04-py3 * Update Docker * Update Docker * Update Docker * Update Docker * Update TRT auto-install * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Cleanup * Cleanup * Cleanup cpu * Cleanup cpu Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>	pirms 2 gadiem
David Matos	bff6e51bc4	Disallow `--dynamic` when `--half` is passed (#7669 ) * disallow dynamic arg when half is given * Update export.py Co-authored-by: David Matos <david@track32.nl> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	pirms 2 gadiem
Duncan Moss	177da7f348	Add `--half` support for OpenVINO exports (#7615 ) * feature(export): add half support for openvino models * Update export.py * Update export.py Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	pirms 2 gadiem
Glenn Jocher	1f1ec1c3e7	Restrict TRT autoinstall to Linux-only (#7549 ) May partially resolve concerns in https://github.com/ultralytics/yolov5/pull/7537#discussion_r856843711	pirms 2 gadiem
Glenn Jocher	f62609eb52	Update check_requirements() with `cmds=()` argument (#7543 )	pirms 2 gadiem
Glenn Jocher	cc1d7df03c	Autoinstall TensorRT if missing (#7537 ) * Autoinstall TensorRT if missing May resolve https://github.com/ultralytics/yolov5/issues/7464 * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update export.py * Update export.py * Update export.py * Update export.py Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>	pirms 2 gadiem
Cedric Perauer	7926afccde	Add `--half` support for FP16 CoreML exports with (#7446 ) * add fp16 for coreml using --half * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * fix * Cleanup * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update export.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>	pirms 2 gadiem
Glenn Jocher	3eefab1bb1	Remove `tensorrt` pip install check (#7439 )	pirms 2 gadiem
Glenn Jocher	2da2466168	Fix EdgeTPU output directory (#7399 ) * Fix EdgeTPU output directory Outputs to same directory as --weights * Update export.py	pirms 2 gadiem
Glenn Jocher	5333b55e74	Remove OpenVINO ONNX `opset<=12` check (#7398 ) No longer needed.	pirms 2 gadiem
Glenn Jocher	3bb233a7fb	Add ONNX export metadata (#7353 )	pirms 2 gadiem
Glenn Jocher	446e6f563a	Rename 'MacOS' to 'macOS' (#7349 )	pirms 2 gadiem
Glenn Jocher	d257c75c84	Update export.py (#7301 ) * Update export.py Simplify code. * [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>	pirms 2 gadiem
Glenn Jocher	5f97001ed4	Context manager `open(file) as f` fixes (#7289 ) * Flask context manager `open()` fix * Additional read context manager fixes	pirms 2 gadiem
leeflix	8d0291f3af	Enable TensorFlow ops for `--nms` and `--agnostic-nms` (#7281 ) * enable TensorFlow ops if flag --nms or --agnostic-nms is used * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update export.py * Update export.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>	pirms 2 gadiem
Glenn Jocher	8bc839ed8e	TorchScript single-output fix (#7261 )	pirms 2 gadiem
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>	pirms 2 gadiem
Glenn Jocher	ffcbd8ca97	Export with official `nn.SiLU()` (#7256 ) * Update * Update time_limit	pirms 2 gadiem
Glenn Jocher	734ab033fd	SavedModel TF Serve Fix (#7228 ) * SavedModel TF Serve Fix Fix for https://github.com/ultralytics/yolov5/issues/7205 proposed by @tylertroy * Update export.py	pirms 2 gadiem
Glenn Jocher	2c3221844b	CLI `fire` prep updates (#7229 ) * CLI fire prep updates * revert unintentional TF export change	pirms 2 gadiem
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>	pirms 2 gadiem
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	pirms 2 gadiem
Mrinal Jain	4effd064b1	Consistent saved_model output format (#7032 )	pirms 2 gadiem
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	pirms 2 gadiem
Glenn Jocher	99de551f97	pt model to cpu on TF export	pirms 2 gadiem
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>	pirms 2 gadiem
Glenn Jocher	acc58c1dcf	Fix TRT `max_workspace_size` deprecation notice (#6856 ) * Fix TRT `max_workspace_size` deprecation notice * Update export.py * Update export.py	pirms 2 gadiem
DavidB	596de6d5a0	Default FP16 TensorRT export (#6798 ) * Assert engine precision #6777 * Default to FP32 inputs for TensorRT engines * Default to FP16 TensorRT exports #6777 * Remove wrong line #6777 * Automatically adjust detect.py input precision #6777 * Automatically adjust val.py input precision #6777 * Add missing colon * Cleanup * Cleanup * Remove default trt_fp16_input definition * Experiment * Reorder detect.py if statement to after half checks * Update common.py * Update export.py * Cleanup Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	pirms 2 gadiem
Louis Combaldieu	b2adc7c39a	Fix export for 1-channel images (#6780 ) Export failed for 1-channel input shape, 1-liner fix	pirms 2 gadiem
Glenn Jocher	de9c25b35e	Use `export_formats()` in export.py (#6705 ) * Use `export_formats()` in export.py * list fix	pirms 2 gadiem
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>	pirms 2 gadiem
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>	pirms 2 gadiem
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	pirms 2 gadiem
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>	pirms 2 gadiem
Glenn Jocher	9bc72a3ac2	Edge TPU export 'list index out of range' fix (#6533 )	pirms 2 gadiem
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	pirms 2 gadiem
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	pirms 2 gadiem
Glenn Jocher	842d049e1b	Suppress `torch.jit.TracerWarning` on export (#6498 ) * Suppress torch.jit.TracerWarning TracerWarnings can be safely ignored. * Cleanup	pirms 2 gadiem
Glenn Jocher	5e4ff195b2	Improved `export.py` usage examples (#6495 ) * Improved `export.py` usage examples * Cleanup	pirms 2 gadiem
Glenn Jocher	750c42e43e	`export.py` automatic `forward_export` (#6352 ) * `export.py` automatic `forward_export` * [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>	pirms 2 gadiem
Glenn Jocher	0cf932bf63	`export.py` return exported files/dirs (#6343 ) * `export.py` return exported files/dirs * Path to str	pirms 2 gadiem
Glenn Jocher	e2e95b2d8e	TensorRT `assert im.device.type != 'cpu'` on export (#6340 ) * TensorRT `assert im.device.type != 'cpu'` on export * Update export.py	pirms 2 gadiem
Glenn Jocher	fd55271c04	Console corrupted -> corrupt (#6338 ) * Console corrupted -> corrupt Minor style changes. * Update export.py	pirms 2 gadiem
Glenn Jocher	80473a6551	Update `export.py` with Detect, Validate usages (#6280 )	pirms 2 gadiem
Jinwoong Yoo	9b13a594e9	Fix `cmd` string on `tfjs` export (#6243 ) * Fix cmd string on tfjs export * Cleanup Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>	pirms 2 gadiem
Glenn Jocher	6865d19a92	TensorRT 7 export fix (#6235 )	pirms 2 gadiem
Glenn Jocher	f80c463010	Attempt `edgetpu-compiler` autoinstall (#6223 ) * Attempt `edgetpu-compiler` autoinstall Attempt to install edgetpu-compiler dependency if missing on Linux. * Update export.py * Update export.py	pirms 2 gadiem
Glenn Jocher	b4ac3df6ff	Add `edgetpu_compiler` checks (#6218 ) * Add `edgetpu_compiler` checks * Update export.py * Update export.py * Update export.py * Update export.py * Update export.py * Update export.py	pirms 2 gadiem

1 2

99 Revīzijas (1b8e70f037abdb6a243c453906ca6fdeacbecafa)