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

1148 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
SkalskiP
b78e30dd2b Merge remote-tracking branch 'origin/develop' into develop 2021-05-28 14:44:45 +02:00
Piotr Skalski
3fea068384 update ci-testing.yml (#3322) 
* update ci-testing.yml

* update greetings.yml

* bring back os matrix
 2021-05-28 14:44:24 +02:00
Piotr Skalski
bb131238aa
update ci-testing.yml (#3322) 
* update ci-testing.yml

* update greetings.yml

* bring back os matrix
 2021-05-27 17:01:36 +02:00
Glenn Jocher
4d4a2b0520
Ignore blank lines in *.txt labels (#3366) 
Fix for https://github.com/ultralytics/yolov5/issues/958#issuecomment-849512083
 2021-05-27 14:31:26 +02:00
WangChaofeng
ef4d53818d
ONNX export in .train() mode fix (#3362) 2021-05-27 14:10:14 +02:00
Glenn Jocher
2435bfe896
Add URL download to check_file() (#3330) 
* Add URL file download to check_file()

* cleanup

* pathlib bug fix
 2021-05-26 15:51:49 +02:00
Glenn Jocher
c6b5bfca85
Updated cache v0.2 with hashlib (#3350) 
* Update cache v0.2 to include parent hash

Possible fix for https://github.com/ultralytics/yolov5/issues/3349

* Update datasets.py
 2021-05-26 14:26:52 +02:00
Glenn Jocher
1f8d716ec9
yolo.py header (#3347) 2021-05-26 12:06:08 +02:00
Glenn Jocher
aad99b63d6
TensorBoard DP/DDP graph fix (#3325) 2021-05-25 11:45:24 +02:00
Glenn Jocher
407dc5008e
Update README.md (#3320) 2021-05-24 17:17:32 +02:00
Glenn Jocher
61ea23c3fe
Implement @torch.no_grad() decorator (#3312) 
* `@torch.no_grad()` decorator

* Update detect.py
 2021-05-24 13:23:09 +02:00
Glenn Jocher
73a92dc1b6
Explicit git clone master (#3311) 2021-05-24 12:42:36 +02:00
Glenn Jocher
9f3a388cea
Improved check_requirements() robustness (#3298) 
Add try: except clause on missing requirements install to catch install failures. Prompted by Ultralytics YOLOv5 API failure on Cython.
 2021-05-23 18:00:23 +02:00
Glenn Jocher
ee24ae11ea
Comment Cython (#3297) 
Ultralytics REST API fix.
 2021-05-23 17:53:54 +02:00
yeric1789
a039b7796d
PyTorch Hub crops = results.crop() return values (#3282) 
* Changing save_one_box 

Made to work with other changes to common.py

* PEP8 and single line BGR

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-05-23 16:00:55 +02:00
Glenn Jocher
0e2f2cbb51
Update LoadStreams init fallbacks (#3295) 2021-05-23 14:55:42 +02:00
Charles Frye
19100ba007
Improves docs and handling of entities and resuming by WandbLogger (#3264) 
* adds latest tag to match wandb defaults

* adds entity handling, 'last' tag

* fixes bug causing finished runs to resume

* removes redundant "last" tag for wandb artifact
 2021-05-21 23:42:53 +02:00
Glenn Jocher
dd7f0b7e05
Fix TypeError: 'PosixPath' object is not iterable (#3285) 2021-05-21 23:35:31 +02:00
Glenn Jocher
683cefead4
YouTube stream ending fix (#3277) 
* YouTube stream ending fix

Properly terminates YouTube streams on video end. Should resolve issues #2769 and #3220.

* Update datasets.py
 2021-05-21 16:51:07 +02:00
Glenn Jocher
f3402353fb
check batch_size % utilized_device_count (#3276) 
Bug fix to check batch_size divisibility of utilized CUDA device count vs total system CUDA device count.
 2021-05-21 15:20:03 +02:00
Glenn Jocher
10d56d784e
Assert --image-weights not combined with DDP (#3275) 2021-05-21 14:46:42 +02:00
Glenn Jocher
7b36e38cf8
Check CoreML models.train() mode (#3262) 
* Check CoreML models.train() mode

* Update export.py
 2021-05-20 15:22:36 +02:00
Glenn Jocher
7b863a34dc
Add Cython (#3217) 
Cython required by pycocotools in certain cases, i.e. https://github.com/cocodataset/cocoapi/issues/172
 2021-05-19 19:58:53 +02:00
yeric1789
7b0eb952ca
plot_one_box() default color=(128, 128, 128) (#3240) 
* Color can be none by default

* `plot_one_box()` default `color=(128, 128, 128)`

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-05-19 19:47:36 +02:00
Glenn Jocher
b7cd1f540d
TensorBoard add_graph() fix (#3236) 2021-05-19 13:51:55 +02:00
Adrian Holovaty
3f74cd9ed1
Parameterize max_det + inference default at 1000 (#3215) 
* Added max_det parameters in various places

* 120 character line

* PEP8

* 120 character line

* Update inference default to 1000 instances

* Update inference default to 1000 instances

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-05-17 22:47:44 +02:00
Glenn Jocher
13a1c72699
Update datasets.py (#3216) 2021-05-17 22:24:26 +02:00
Glenn Jocher
ffb47ffbeb
Silent Patches Bug Fix (#3214) 2021-05-17 17:47:30 +02:00
Glenn Jocher
36b0a9e7d0
Add CAP_PROP_FRAME_COUNT for YouTube sources (#3193) 2021-05-17 12:27:40 +02:00
Glenn Jocher
ae53f50ae7
Plot labels histogram colors (#3192) 2021-05-17 11:44:22 +02:00
Christoph Gerum
b133baa336
Add device argument to PyTorch Hub models (#3104) 
* Allow to manual selection of device for torchhub models

* single line device

nested torch.device(torch.device(device)) ok

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-05-16 17:41:26 +02:00
Cristi Fati
9ab561dbfc
Parameterize ONNX --opset-version (#3154) 2021-05-16 16:13:03 +02:00
Yonghye Kwon
be86c21c73
rename class autoShape -> AutoShape (#3173) 
* rename class autoShape -> AutoShape

follow other class naming convention

* rename class autoShape -> AutoShape

follow other classes' naming convention

* rename class autoShape -> AutoShape
 2021-05-16 15:46:45 +02:00
Glenn Jocher
17b0f71538
SKU-110K CVPR2019 Dataset Auto-Download (#3167) 
* SKU-110K CVPR2019 Dataset Auto-Download

This PR adds the **SKU-110K CVPR2019 Dataset** https://github.com/eg4000/SKU110K_CVPR19 to the list of YOLOv5 🚀  auto-download list.

- [x] [COCO](https://cocodataset.org/#home)
- [x] [COCO128](https://www.kaggle.com/ultralytics/coco128)
- [x] [VOC](http://host.robots.ox.ac.uk/pascal/VOC/)
- [x] [Argoverse-HD](http://www.cs.cmu.edu/~mengtial/proj/streaming/)
- [x] [VisDrone2019-DET](https://github.com/VisDrone/VisDrone-Dataset)
- [x] [GlobalWheat-2020](http://www.global-wheat.com/)
- [ ] [KITTI](https://www.kaggle.com/twaldo/kitti-object-detection)
- [x] [SKU110](https://github.com/eg4000/SKU110K_CVPR19) (grocery store items)
- [ ] [CityScapes](https://www.cityscapes-dataset.com/)
- [x] [Objects365](https://www.objects365.org/overview.html)
- [ ] [OpenImages](https://storage.googleapis.com/openimages/web/index.html)

```yaml
# SKU-110K retail items dataset https://github.com/eg4000/SKU110K_CVPR19
# Train command: python train.py --data SKU-110K.yaml
# Default dataset location is next to YOLOv5:
#   /parent_folder
#     /datasets/SKU-110K
#     /yolov5
```

* Update SKU-110K.yaml

* Update SKU-110K.yaml
 2021-05-14 17:05:26 +02:00
Glenn Jocher
06372b1465
Scope all hubconf.py imports for torch.hub.list() (#3145) 
* Scope all hubconf.py imports for torch.hub.list()

* Update hubconf.py
 2021-05-12 20:28:26 +02:00
Glenn Jocher
1935266951
Scope imports for torch.hub.list() improvement (#3144) 2021-05-12 20:18:32 +02:00
Glenn Jocher
b098600763
Update requirements.txt onnx>=1.9.0 (#3143) 2021-05-12 19:56:12 +02:00
Glenn Jocher
f12cef85a3
TorchScript, ONNX, CoreML Export tutorial title (#3142) 2021-05-12 19:49:12 +02:00
Cristi Fati
d9b4e6b748
Add --include torchscript onnx coreml argument (#3137) 
* Allow users to skip exporting in formats that they don't care about

* Correct comments

* Update export.py

renamed --skip-format to --exclude

* Switched format from exclude to include (as instructed by @glenn-jocher)

* cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-05-12 19:46:32 +02:00
KC-Zhang
bd6f6a784a
Add yolov5/__init__.py (#3127) 2021-05-12 18:50:35 +02:00
Glenn Jocher
7918eed040
Update https://ultralytics.com/images/zidane.jpg (#3140) 2021-05-12 18:30:31 +02:00
Glenn Jocher
c1c7eb023f
Update JSON response (#3139) 2021-05-12 18:23:50 +02:00
Glenn Jocher
52c05707d6
Update Colors() (#3046) 
* Update Colors()

* update colors

* update colors
 2021-05-12 17:41:11 +02:00
Glenn Jocher
abfcf9eb79
Replace print() with logging.info() in trainloader (#3103) 
Might indirectly help #3095 by providing better visibility on source of corruption.
 2021-05-10 17:07:16 +02:00
Glenn Jocher
25f8ab835e
detect.py streaming source --save-crop bug fix (#3102) 
* detect.py streaming source --save-crop bug fix

Possible fix for #3100.

* () parenthesis
 2021-05-10 16:39:37 +02:00
Glenn Jocher
d6a71ae5f2
Update GlobalWheat2020.yaml test: # 1276 images (#3101) 2021-05-10 16:04:10 +02:00
Glenn Jocher
60fe54449d
Update train.py (#3099) 2021-05-10 15:26:23 +02:00
Glenn Jocher
525f4f86a9
Add --optimize argument (#3093) 
Fix for c++ runtime errors in https://github.com/ultralytics/yolov5/issues/2973
 2021-05-10 12:13:44 +02:00
Glenn Jocher
57b0d3a605
Add check_python() (#3088) 
* Add check_python()

Checks python version against minimum version of 3.7.0.

* remove packaging dependency

* refactor import
 2021-05-09 15:33:38 +02:00
Glenn Jocher
91547edec1
Update detect.py (#3087) 
* Update detect.py

* Update detect.py
 2021-05-09 13:01:58 +02:00