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,722 Commits 10 Branches 10 Tags
Commit Graph

1722 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
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
136640eee8 Improve doctstring, add usage examples 2021-10-09 15:22:12 -07:00
Glenn Jocher
76259b15b0 Hyp fix 2021-10-09 14:40:42 -07:00
Glenn Jocher
f9c254b207 Merge master 2021-10-09 14:38:28 -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
Glenn Jocher
cd35a009ba
Revert torch.hub.load() test (#4986) 
Temporarily reverts https://github.com/ultralytics/yolov5/pull/4978 until torch 1.10 is released, which should resolve `urllib.error.HTTPError: HTTP Error 403: rate limit exceeded` errors generated by torch hub from GitHub actions runners.
 2021-09-28 20:11:20 -07:00
Glenn Jocher
9988059b10
Clickable CI badge (#4985) 2021-09-28 17:46:14 -07:00
Glenn Jocher
6b19f728a1 Created using Colaboratory 2021-09-28 17:33:48 -07:00
Glenn Jocher
cff7d2af45 Created using Colaboratory 2021-09-28 17:29:05 -07:00
Glenn Jocher
0c93ec762a
Faster --img 64 CI tests (#4979) 2021-09-28 12:41:28 -07:00
Glenn Jocher
94705a9528
Add Hub custom models to CI tests (#4978) 
* Update ci-testing.yml for Hub custom model tests

* Update ci-testing.yml
 2021-09-28 12:28:18 -07:00
Diego Montes
0c87478713
Update wandb_utils.py (#4953) 
`is_valset_wandb_artifact` and `is_trainset_wandb_artifact` were referenced before assignment causing wandb to be unusable.
 2021-09-28 12:22:31 -07:00
Glenn Jocher
38c779b099 Created using Colaboratory 2021-09-28 10:04:56 -07:00
Kalen Michael
29acedf7dd
Add Roboflow to README (#4972) 
* added callbacks

* added back callback to main

* added save_dir to callback output

* merged in upstream

* removed ghost code

* added integrations section removed competition

* attempt column build

* attempt column build

* moved new to header

* removed center column

* utm

* Update README.md

* Update README.md

* updated logo sizes

* resized logos

* fixed links

* Update README.md

* Update README.md

* Update README.md

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-09-28 09:35:46 -07:00