1722 Commits

Author SHA1 Message Date
Glenn Jocher
a6037838a3 Merge master 2021-12-12 23:52:55 +01:00
Diego Montes
19c56e60b1
Fix imgsz bug (#5948)
* fix imgsz bug

* Update detect.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-12-12 23:39:14 +01:00
Yono Mittlefehldt
8f354362cd
Fix Detections class tolist() method (#5945)
* Fix tolist() to add the file for each Detection

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Fix PEP8 requirement for 2 spaces before an inline comment

* [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>
2021-12-11 18:40:37 +01:00
Glenn Jocher
ea3d131010 Update default --workers 8 2021-12-11 14:34:12 +01:00
Glenn Jocher
6c2ac617d9 Merge master 2021-12-11 14:32:39 +01:00
Glenn Jocher
8f875d93a2
Refactor NUM_THREADS (#5954) 2021-12-11 14:32:18 +01:00
Glenn Jocher
ecd0f6d9b9 Add mnist-fashion 2021-12-11 13:36:02 +01:00
Glenn Jocher
d88ee1e629 Merge master 2021-12-11 13:08:42 +01:00
Glenn Jocher
14af287e6c Add FILE ROOT globals 2021-12-11 13:00:42 +01:00
Diego Montes
2c6317547a
Add nms and agnostic nms to export.py (#5938)
* add nms and agnostic nms to export.py

* fix agnostic implies nms

* reorder args to group TF args

* PEP8 120 char

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-12-10 18:24:32 +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
Pascal Maillard
922fbd8209
fix .gitignore not tracking existing folders (#5946)
* fix .gitignore not tracking existing folders

fix .gitignore so that the files that are in the repository are actually being tracked.

Everything in the data/ folder is ignored, which also means the subdirectories are ignored. Fix so that the subdirectories and their contents are still tracked.

* Remove data/trainings

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-12-10 17:32:09 +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
Diego Montes
4fb6dd4b26
Fix ONNX opset inconsistency with parseargs and run args (#5937) 2021-12-09 23:10:16 +01:00
Glenn Jocher
5bdb28ed10
Default PyTorch Hub to autocast(False) (#5926) 2021-12-08 23:15:14 +01:00
Glenn Jocher
3e0008d53d Add --adam to Usage: example 2021-12-08 17:48:05 +01:00
Glenn Jocher
2b40bd928a Merge master 2021-12-08 17:32:20 +01:00
Glenn Jocher
c77a5a84e3
Absolute '/content/sample_data' (#5922) 2021-12-08 17:31:31 +01:00
Glenn Jocher
3feecfaeb7 Improve stability by disabling AMP 2021-12-08 17:19:58 +01:00
Glenn Jocher
a3d5f1d3e3
Revert "Update plot_lr_scheduler() (#5864)" (#5920)
This reverts commit 360eec69101ad9ffada78326b715b724b9b7eb0d.
2021-12-08 16:46:24 +01:00
Glenn Jocher
1207356411 Implement scaled loss backward 2021-12-08 16:33:16 +01:00
Glenn Jocher
d82766dd09 Merge master 2021-12-08 16:18:42 +01:00
Glenn Jocher
91e13b24f4 Smart DetectMultiBackend() unwrapping 2021-12-08 16:15:31 +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
Glenn Jocher
581dc301a7
Add ONNX inference providers (#5918)
* Add ONNX inference providers

Fix for https://github.com/ultralytics/yolov5/issues/5916

* Update common.py
2021-12-08 13:37:33 +01:00
greg2451
554f782537
Add *.engine (TensorRT extensions) to .gitignore (#5911)
* Add *.engine (TensorRT extensions) to .gitignore

* Update .dockerignore

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-12-07 18:20:16 +01:00
Ayush Chaurasia
3f152e5807
Update wandb_utils.py (#5908) 2021-12-07 16:09:11 +01:00
Li Zeng
fa05f8c977
Detections().tolist() explicit argument fix (#5907)
debugged for missigned Detections attributes
2021-12-07 16:01:41 +01:00
Glenn Jocher
2ed726baff pbar fix 2021-12-06 20:55:05 +01:00
Glenn Jocher
4a922e4f3c Merge master 2021-12-06 20:51:06 +01:00
Glenn Jocher
07ae9f799a DetectMultiBackend fix 2021-12-06 20:50:38 +01:00
Glenn Jocher
1075488d89
Single-command multiple-model export (#5882)
* Export multiple models in series

Export multiple models in series by adding additional `*.pt` files to the `--weights` argument, i.e.:

```bash
python export.py --include tflite --weights yolov5n.pt  # export 1 model
python export.py --include tflite --weights yolov5n.pt yolov5s.pt yolov5m.pt yolov5l.pt yolov5x.pt  # export 5 models
```

* Update export.py

* Update README.md
2021-12-04 16:28:40 +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
iumyx2612
92a7391039
Add --workers 8 argument to val.py (#5857)
* Update val.py

Add an option to choose number of workers if not called by train.py

* Update comment

* 120 char line width

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-12-02 16:49:50 +01:00
Glenn Jocher
00e308f7be
Update TorchScript suffix to *.torchscript (#5856) 2021-12-02 16:06:45 +01:00
Vishnu Pradeep
30db14fea8
Update val.py (#5838)
* Update val.py

Solving Non-ASCII character '\xf0' error during runtime

* Update val.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-12-02 13:20:58 +01:00
gggmt
1679aacdc7
Update LoadImages ret_val=False handling (#5852)
Video errors may occur.
2021-12-02 10:57:39 +01:00
Yu Zhang
e8f8f2b903
Avoid inplace modifyingimgs in LoadStreams (#5850)
When OpenCV retrieving image fail, original code would modify source images **inplace**, which may result in plotting bounding boxes on a black image. That is, before inference, source image `im0s[i]` is OK, but after inference before `Process predictions`,  `im0s[i]` may have been changed.
2021-12-02 10:51:19 +01:00
Glenn Jocher
9451512a86 Merge master 2021-12-01 15:38:41 +01:00
Glenn Jocher
bc484579d7
Handle non-TTY wandb.errors.UsageError (#5839)
* `try: except (..., wandb.errors.UsageError)`

* bug fix
2021-12-01 15:38:02 +01:00
imyhxy
a4207a202d
Fix TensorRT potential unordered binding addresses (#5826)
* feat: change file suffix in pythonic way

* fix: enforce binding addresses order

* fix: enforce binding addresses order
2021-11-30 13:52:22 +01:00
Glenn Jocher
5ca5dd4c87
Update dataset_stats() to cv2.INTER_AREA (#5821) 2021-11-29 13:15:36 +01:00
dependabot[bot]
8277033b65
Bump actions/cache from 2.1.6 to 2.1.7 (#5816)
Bumps [actions/cache](https://github.com/actions/cache) from 2.1.6 to 2.1.7.
- [Release notes](https://github.com/actions/cache/releases)
- [Commits](https://github.com/actions/cache/compare/v2.1.6...v2.1.7)

---
updated-dependencies:
- dependency-name: actions/cache
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
2021-11-29 12:04:36 +01:00
Glenn Jocher
94d8fec6d8
GCP VM from Image example (#5814) 2021-11-28 18:12:46 +01:00
Glenn Jocher
fcd180d336
Refactor new model.warmup() method (#5810)
* Refactor new `model.warmup()` method

* Add half
2021-11-27 12:29:45 +01:00
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>
2021-11-26 13:37:28 +01:00
Phil2020
53349dac8e
Scope TF imports in DetectMultiBackend() (#5792)
* tensorflow or tflite exclusively as interpreter

As per bug report https://github.com/ultralytics/yolov5/issues/5709 I think there should be only one attempt to assign interpreter, and it appears tflite is only ever needed for the case of edgetpu model.

* Scope imports

* Nested definition line fix

* Update common.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-11-25 17:54:00 +01:00
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>
2021-11-25 17:19:21 +01:00