mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity
1,556 Commits 10 Branches 10 Tags
Commit Graph

1556 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
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 Lai
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
NauchtanRobotics
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 Chaurasia
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 Chaurasia
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 Chechik
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
v6.0 		2021-10-11 23:47:18 -07:00
Andrei Ionut Damian
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 Nadar
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 Lorbach
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 Nadar
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
Nan
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 mohammadi
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 Orel
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 Michael
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
Glenn Jocher
42354d7025
Update README.md 2021-09-29 16:56:10 -07:00
Kalen Michael
76d301bd21
Fix URL parsing bug (#4998) 
* added callbacks

* added back callback to main

* added save_dir to callback output

* merged in upstream

* removed ghost code

* fixed parsing error for google temp links

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-09-29 10:48:45 -07:00