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

1161 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Yonghye Kwon
c37f072ba7
Faster HSV augmentation (#3462) 
remove datatype conversion process that can be skipped
 2021-06-04 20:02:20 +02:00
Glenn Jocher
d40481acc5
Add timeout() class (#3460) 
* Add `timeout()` class

* rearrange order
 2021-06-04 15:23:33 +02:00
Glenn Jocher
8e3b4a0bf3
Update MixUp augmentation alpha=beta=32.0 (#3455) 
Per VOC empirical results https://github.com/ultralytics/yolov5/issues/3380#issuecomment-853001307 by @developer0hye
 2021-06-04 12:47:53 +02:00
Glenn Jocher
4aa2959101
Suppress jit trace warning + graph once (#3454) 
* Suppress jit trace warning + graph once

Suppress harmless jit trace warning on TensorBoard add_graph call. Also fix multiple add_graph() calls bug, now only on batch 0.

* Update train.py
 2021-06-04 12:37:41 +02:00
Glenn Jocher
af2bc3a1c3
Add FLOPs title to table (#3453) 2021-06-04 11:46:40 +02:00
Glenn Jocher
f8651c388f
Parse URL authentication (#3424) 
* Parse URL authentication

* urllib.parse.unquote()

* improved error handling

* improved error handling

* remove %3F

* update check_file()
 2021-06-03 18:44:58 +02:00
chocosaj
3cb9ad4fc4
Update FLOPs description (#3422) 
* Update README.md

* Changing FLOPS to FLOPs.

Co-authored-by: BuildTools <unconfigured@null.spigotmc.org>
 2021-06-03 12:31:51 +02:00
Glenn Jocher
fdbe527dc0
Revert "cv2.imread(img, -1) for IMREAD_UNCHANGED (#3379)" (#3395) 
This reverts commit 21a9607e00f1365b21d8c4bd81bdbf5fc0efea24.
 2021-05-31 10:39:00 +02:00
Glenn Jocher
d833ab3d25
Create is_pip() function (#3391) 
Returns `True` if file is part of pip package. Useful for contextual behavior modification.

```python
def is_pip():
    # Is file in a pip package?
    return 'site-packages' in Path(__file__).absolute().parts
```
 2021-05-30 20:52:42 +02:00
Glenn Jocher
4b52e19a61
COCO evolution fix (#3388) 
* COCO evolution fix

* cleanup

* update print

* print fix
 2021-05-29 22:49:34 +02:00
tudoulei
21a9607e00
cv2.imread(img, -1) for IMREAD_UNCHANGED (#3379) 
* Update datasets.py

* comment

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
 2021-05-29 21:12:01 +02:00
Peretz Cohen
57f773b0ae
Update tutorial.ipynb (#3368) 
add Open in Kaggle badge
 2021-05-29 20:49:24 +02:00
Glenn Jocher
ba6f3f974b
Enable direct --weights URL definition (#3373) 
* Enable direct `--weights URL` definition

@KalenMike this PR will enable direct --weights URL definition. Example use case:
```
python train.py --weights https://storage.googleapis.com/bucket/dir/model.pt
```

* cleanup

* bug fixes

* weights = attempt_download(weights)

* Update experimental.py

* Update hubconf.py

* return bug fix

* comment mirror

* min_bytes
 2021-05-28 15:18:44 +02:00
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