Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
1,558 Commits 18 Branches 0 Tags
fd3e2ad8efb2595d3329833ce5e50d7a49e07440
Commit Graph
1558 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
dependabot[bot] fd3e2ad8ef Bump actions/stale from 3 to 4 (#5253) 
Bumps [actions/stale](https://github.com/actions/stale) from 3 to 4.
- [Release notes](https://github.com/actions/stale/releases)
- [Changelog](https://github.com/actions/stale/blob/main/CHANGELOG.md)
- [Commits](https://github.com/actions/stale/compare/v3...v4)

---
updated-dependencies:
- dependency-name: actions/stale
  dependency-type: direct:production
  update-type: version-update:semver-major
...

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

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2021-10-19 11:37:54 +02:00
dependabot[bot] aa26504137 Bump actions/cache from 1 to 2.1.6 (#5252) 
Bumps [actions/cache](https://github.com/actions/cache) from 1 to 2.1.6.
- [Release notes](https://github.com/actions/cache/releases)
- [Commits](https://github.com/actions/cache/compare/v1...v2.1.6)

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

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

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2021-10-19 11:31:24 +02:00
dependabot[bot] fb12faafa3 Bump cirrus-actions/rebase from 1.4 to 1.5 (#5251) 
Bumps [cirrus-actions/rebase](https://github.com/cirrus-actions/rebase) from 1.4 to 1.5.
- [Release notes](https://github.com/cirrus-actions/rebase/releases)
- [Commits](https://github.com/cirrus-actions/rebase/compare/1.4...1.5)

---
updated-dependencies:
- dependency-name: cirrus-actions/rebase
  dependency-type: direct:production
  update-type: version-update:semver-minor
...

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

Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2021-10-19 11:28:52 +02:00
Zhiqiang Wang 8e87396705 Add dependabot for GH actions (#5250) 2021-10-19 11:23:30 +02:00
Glenn Jocher 4f5e274eee Update rebase.yml (#5245) 2021-10-19 09:46:05 +02:00
Glenn Jocher 752d97df2c Add on_fit_epoch_end callback (#5232) 
* Add `on_fit_epoch_end` callback

* Add results to train

* Update __init__.py
 2021-10-18 14:28:22 +02:00
Glenn Jocher 13f7275555 Update/inplace ops (#5233) 
* Clip Objects365 autodownload labels (#5214)

Fixes out of bounds labels that seem to affect ~10% of images in dataset.

* Inplace ops
 2021-10-18 14:24:48 +02:00
Glenn Jocher 0000334a6f Update Objects365.yaml val count (#5212) 2021-10-15 12:40:40 -07:00
Glenn Jocher 991c654e81 Autofix duplicate label handling (#5210) 
* Autofix duplicate labels

PR changes duplicate label handling from report error and ignore image-label pair to report warning and autofix image-label pair. 

This should fix this common issue for users and allow everyone to get started and get a model trained faster and easier than before.

* sign fix

* Cleanup

* Increment cache version

* all to any fix
 2021-10-15 12:32:08 -07:00
Farley LaiandGlenn Jocher fc3606420d Update Objects365.yaml to include the official validation set (#5194) 
* Update Objects365.yaml

Download the official Objects365 validation set and convert the labels

* Enforce 4-space indent, reformat and cleanup

* shorten list comprehension

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-14 20:48:47 -07:00
Glenn Jocher 1c9f710b46 Add hyp.scratch-med.yaml (#5196) 
* Add hyp.scratch-med.yaml

* Update hyp.scratch-med.yaml
 2021-10-14 20:40:30 -07:00
Glenn Jocher 6b31cfd793 Highlight contributors in README (#5173) 
* Highlight contributors in README

* Adjust Contact section

* Update README.md
 2021-10-14 12:49:16 -07:00
Glenn Jocher 7ee5aed0b3 Improved check_suffix() robustness to '' and "" (#5192) 
* Improved check_suffix() robustness to `''` and `""`

* Cleanup
 2021-10-14 12:00:39 -07:00
NauchtanRoboticsandGlenn Jocher 0be58f16f0 Adjust legend labels for classes without instances (#5174) 
* legend-labels Adjust legend labels for classes without instances.

* #5158 Re-indexed series names: only classes containing data.

* #5158 Re-indexed series names: only classes containing data.

* Cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-14 11:57:00 -07:00
Ayush ChaurasiaandGlenn Jocher 1172ac15fb W&B: DDP fix (#5176) 
* fix dpp with wandb disabled

* PyCharm reformat

* Fix

* fix

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-13 20:11:04 -07:00
Glenn Jocher e520e820f1 Update autodownload fallbacks to v6.0 assets (#5177) 2021-10-13 19:58:38 -07:00
Ayush ChaurasiaandGlenn Jocher d0bfeb3980 W&B: fix DDP with wandb disabled (#5163) 
* fix dpp with wandb disabled

* PyCharm reformat

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-13 16:17:15 -07:00
Glenn Jocher a346926996 Add class filtering to LoadImagesAndLabels() dataloader (#5172) 
* Add train class filter feature to datasets.py

Allows for training on a subset of total classes if `include_class` list is defined on datasets.py L448:
```python
        include_class = []  # filter labels to include only these classes (optional)
```

* segments fix
 2021-10-13 15:48:54 -07:00
Glenn Jocher b754525e99 Check 'onnxruntime-gpu' if torch.has_cuda (#5087) 
* Check `'onnxruntime-gpu' if torch.has_cuda`

* fix indent
 2021-10-12 22:25:38 -07:00
Glenn Jocher 20a809dee5 Update stale.yml (#5156) 2021-10-12 17:54:56 -07:00
Glenn Jocher 26784afd99 Use YOLOv5n for CI testing (#5154) 2021-10-12 08:42:26 -07:00
Yoni ChechikandGlenn Jocher 34da872ab6 fix tf conversion in new v6 models (#5153) 
* fix `tf` conversion in new v6 (#5147)

* sort imports

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-12 08:38:54 -07:00
Glenn Jocher 956be8e642 YOLOv5 release v6.0 (#5141) 
* Update P5 models

* Update P6 models

* Update with GFLOPs and Params

* Update with GFLOPs and Params

* Update README

* Update

* Update README

* Update

* Update

* Add times

* Update README

* Update results

* Update results

* Update results

* Update hyps

* Update plots

* Update plots

* Update README.md

* Add nano models to hubconf.py
 2021-10-11 23:47:18 -07:00
Andrei Ionut DamianandGlenn Jocher 938efe5dcf update detect.py in order to support torch script (#5109) 
* update detect.py in order to support torch script

This change assumes the torchscrip file was previously saved with `export.py`

* update `detect.py` for torchscript support

Simple update for torchscript support. Assumes the torchscript file has been generated with `export.py`

* Cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-11 21:56:06 -07:00
Glenn Jocher 0bf24cf641 Add OpenCV DNN option for ONNX inference (#5136) 
* Add OpenCV DNN option for ONNX inference

Usage:

```bash
python detect.py --weights yolov5s.onnx  # ONNX Runtime inference
python detect.py --weights yolov5s.onnx -dnn  # OpenCV DNN inference
```

* DNN prediction to tensor

* Update detect.py
 2021-10-11 12:39:20 -07:00
Jebastin NadarandGlenn Jocher 9d75e42f98 Refactor Detect() anchors for ONNX <> OpenCV DNN compatibility (#4833) 
* refactor anchors and anchor_grid in Detect Layer

* fix CI failures by adding compatibility

* fix tf failure

* fix different devices errors

* Cleanup

* fix anchors overwriting issue

* better refactoring

* Remove self.anchor_grid shape check (redundant with self.grid check)

Also PEP8 / 120 line width

* Convert _make_grid() from static to dynamic method

* Remove anchor_grid.to(device)

clone() should already clone to same device as self.anchors

* fix different devices error

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-11 09:58:42 -07:00
Malte LorbachandGlenn Jocher 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>
 2021-10-11 09:47:24 -07:00
Glenn Jocher 4cf7d487a6 Update plots.py (#5127) 2021-10-11 01:59:22 -07:00
Glenn Jocher 7d37b3c52e Update val.py pad = 0.0 if task == speed else 0.5 (#5121) 
* Update val.py `pad = 0.0 if task == speed else 0.5`

* Cleanup
 2021-10-10 23:20:42 -07:00
Glenn Jocher ba4b79de8b Update val.py --speed and --study usages (#5120) 2021-10-10 21:15:28 -07:00
Glenn Jocher 4f9718abe6 Pass --device for --task study (#5118) 2021-10-10 15:07:26 -07:00
Jebastin NadarandGlenn Jocher a0e15046c3 Fix different devices bug when moving model from GPU to CPU (#5110) 
* fix different devices bug

* extend _apply() instead of to() for a general fix

* Only apply if Detect() is last layer

Co-authored-by: Jebastin Nadar <njebastin10@gmail.com>

* Indent fix

* Add comment to yolo.py

* Add comment to common.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-10 14:24:05 -07:00
Nanandqiningonline 4a6dfffdaa Pass LOCAL_RANK to torch_distributed_zero_first() (#5114) 
Co-authored-by: qiningonline <qiningonline@gmail.com>
 2021-10-09 18:41:50 -07:00
hamid mohammadiandGlenn Jocher 97b6b14abe Add requests to requirements.txt (#5112) 
* Requirements updated

1. Requests added to requirements.txt. That might not be included in all docker base images, adding it to the requirements is safer.
2. Added a minimum version to Pandas. It's a good practice to have versions for all dependencies.

* Sort alphabetically

* Update requirements.txt

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-09 15:51:01 -07:00
Glenn Jocher 276b674513 Fix SKU-110K HUB: OSError (#5106) 2021-10-09 00:12:59 -07:00
Egor OrelandGlenn Jocher 48b00dbc0d Re-open IP-camera videostream if disconnected (#5074) 
* Re-open IP-camera videostream if disconnected. Provide IP-streaming analysis stability

* Re-open IP-camera videostream if disconnected. Provide IP-streaming analysis stability

* `self.imgs[i] *= 0` bug fix and cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-10-07 23:26:43 -07:00
Glenn Jocher 57ce73cd4f Update datasets.py comments (#5088) 2021-10-07 16:04:08 -07:00
Glenn Jocher ce8e5dc8bb Update Dockerfile to ADD Arial.ttf (#5084) 
* Update Dockerfile to `ADD` Arial.ttf

Fix for `Arial.ttf download in Docker` #5078

* Cleanup

* Cleanup2
 2021-10-07 14:16:33 -07:00
Glenn Jocher 070af88108 Fix yaml.safe_load() ignore emoji errors (#5060) 2021-10-05 13:41:52 -07:00
Glenn Jocher 5afc9c25ef Implement --save-period locally (#5047) 
This PR adds a new training argument `--save-period` to save training checkpoints every `x` epochs. To save training every 50 epochs for example:
```
python train.py --save-period 50  # saves epoch50.pt, epoch100.pt, epoch150.pt, ... etc.
```

This saved checkpoints in addition to existing last.pt and best.pt checkpoints and does not affect their behavior. Default value is -1, i.e. disabled.
 2021-10-04 18:48:46 -07:00
Glenn Jocher d133968b1e Clip TTA Augmented Tails (#5028) 
* Clip TTA Augmented Tails

Experimental TTA update.

* Update yolo.py

* Update yolo.py

* Update yolo.py

* Update yolo.py
 2021-10-04 15:48:55 -07:00
Zhiqiang Wang 1922ddeac0 Fix pylint: do not use bare 'except' (#5025) 
* Fix E722, do not use bare 'except'

* Remove used codes

* Add FileNotFoundError in LoadImagesAndLabels

* Remove AssertionError

* Ignore LoadImagesAndLabels

* Ignore downloads.py

* Ignore torch_utils.py

* Ignore train.py

* Ignore datasets.py

* Enable utils/download.py

* Fixing exception in thop

* Remove unused code

* Fixing exception in LoadImagesAndLabels

* Fixing exception in exif_size

* Fixing exception in parse_model

* Ignore exceptions in requests

* Revert the exception as suggested

* Revert the exception as suggested
 2021-10-03 17:54:40 -07:00
Glenn Jocher b0ade48457 Fix missing opt.device on --task study (#5031) 2021-10-02 17:55:55 -07:00
Glenn Jocher cce7e78fae Created using Colaboratory 2021-09-30 19:10:55 -07:00
Glenn Jocher 89a6e9ffa7 Compute loss on final val (#5017) 2021-09-30 15:16:11 -07:00
Glenn Jocher b20e381492 Update README.md (#5015) 2021-09-30 14:06:00 -07:00
Glenn Jocher 452e84703c Update train.py (#5014) 2021-09-30 13:57:53 -07:00
Kalen MichaelandGlenn Jocher cce313d67d Add YOLOv5 Survey link to README.md (#5000) 
* added callbacks

* added back callback to main

* added save_dir to callback output

* merged in upstream

* removed ghost code

* added survey

* Update README.md

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-09-30 13:25:48 -07:00
Glenn Jocher ba99092304 Update W&B README (#5006) 2021-09-30 00:35:00 -07:00
Glenn Jocher 1916226841 Created using Colaboratory 2021-09-29 22:18:16 -07:00