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
1f8d716ec9
yolo.py header ( #3347 )
2021-05-26 12:06:08 +02:00
aad99b63d6
TensorBoard DP/DDP graph fix ( #3325 )
2021-05-25 11:45:24 +02:00
407dc5008e
Update README.md ( #3320 )
2021-05-24 17:17:32 +02:00
61ea23c3fe
Implement `@torch.no_grad()` decorator ( #3312 )
...
* `@torch.no_grad()` decorator
* Update detect.py
2021-05-24 13:23:09 +02:00
73a92dc1b6
Explicit `git clone` master ( #3311 )
2021-05-24 12:42:36 +02:00
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
ee24ae11ea
Comment Cython ( #3297 )
...
Ultralytics REST API fix.
2021-05-23 17:53:54 +02:00
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
0e2f2cbb51
Update LoadStreams init fallbacks ( #3295 )
2021-05-23 14:55:42 +02:00
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
dd7f0b7e05
Fix TypeError: 'PosixPath' object is not iterable ( #3285 )
2021-05-21 23:35:31 +02:00
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
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
10d56d784e
Assert `--image-weights` not combined with DDP ( #3275 )
2021-05-21 14:46:42 +02:00
7b36e38cf8
Check CoreML models.train() mode ( #3262 )
...
* Check CoreML models.train() mode
* Update export.py
2021-05-20 15:22:36 +02:00
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
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
b7cd1f540d
TensorBoard add_graph() fix ( #3236 )
2021-05-19 13:51:55 +02:00
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
13a1c72699
Update datasets.py ( #3216 )
2021-05-17 22:24:26 +02:00
ffb47ffbeb
Silent Patches Bug Fix ( #3214 )
2021-05-17 17:47:30 +02:00
36b0a9e7d0
Add CAP_PROP_FRAME_COUNT for YouTube sources ( #3193 )
2021-05-17 12:27:40 +02:00
ae53f50ae7
Plot labels histogram colors ( #3192 )
2021-05-17 11:44:22 +02:00
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
9ab561dbfc
Parameterize ONNX `--opset-version` ( #3154 )
2021-05-16 16:13:03 +02:00
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
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
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
1935266951
Scope imports for torch.hub.list() improvement ( #3144 )
2021-05-12 20:18:32 +02:00
b098600763
Update requirements.txt `onnx>=1.9.0` ( #3143 )
2021-05-12 19:56:12 +02:00
f12cef85a3
TorchScript, ONNX, CoreML Export tutorial title ( #3142 )
2021-05-12 19:49:12 +02:00
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
bd6f6a784a
Add yolov5/__init__.py ( #3127 )
2021-05-12 18:50:35 +02:00
7918eed040
Update https://ultralytics.com/images/zidane.jpg ( #3140 )
2021-05-12 18:30:31 +02:00
c1c7eb023f
Update JSON response ( #3139 )
2021-05-12 18:23:50 +02:00
52c05707d6
Update Colors() ( #3046 )
...
* Update Colors()
* update colors
* update colors
2021-05-12 17:41:11 +02:00
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
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
d6a71ae5f2
Update GlobalWheat2020.yaml test: # 1276 images ( #3101 )
2021-05-10 16:04:10 +02:00
60fe54449d
Update train.py ( #3099 )
2021-05-10 15:26:23 +02:00
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
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
91547edec1
Update detect.py ( #3087 )
...
* Update detect.py
* Update detect.py
2021-05-09 13:01:58 +02:00
251aeafcb1
Update P5 + P6 model ensembling ( #3082 )
2021-05-08 19:06:12 +02:00
d2a17289c9
Explicitly convert artifact path to posix_path ( #3067 )
...
* Explicitly convert artifact path to posix_path
* Remove redudant str() casting
2021-05-07 19:53:56 +02:00
e97d129db4
Update export.py with --train mode argument ( #3066 )
2021-05-07 13:10:07 +02:00
f2de1ad2aa
Comment fix ( #3058 )
...
Co-authored-by: kepler62f <>
2021-05-07 12:28:55 +02:00
3ef3a95cfa
Do not optimize CoreML TorchScript model ( #3055 )
2021-05-06 17:23:33 +02:00
b8be76f915
Add get_coco128.sh for downloading the coco128 dataset ( #3047 )
...
* Add get_coco128.sh
* Update get_coco128.sh
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-05-06 13:44:14 +02:00
Glenn Jocher
yeric1789
Charles Frye
Adrian Holovaty
Christoph Gerum
Cristi Fati
Yonghye Kwon
KC-Zhang
Ayush Chaurasia
kepler62f
Jiacong Fang
