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
b18ca31ff5
use check_file for --data ( #3035 )
2021-05-05 14:28:19 +02:00
85b6fdda0f
FROM nvcr.io/nvidia/pytorch:21.03-py3
...
Issues with 21.04
2021-05-05 01:00:12 +02:00
4c7d686d7e
FROM nvcr.io/nvidia/pytorch:21.04-py3 ( #3030 )
2021-05-05 00:16:35 +02:00
8cab907f60
download() ThreadPool update ( #3027 )
...
* download() ThreadPool update
* update train image count
* cid + 1
2021-05-04 23:37:31 +02:00
5189b3addb
Objects365 Update
2021-05-04 02:25:14 +02:00
b292837e36
Fix ONNX export using --grid --simplify --dynamic simultaneously ( #2982 )
...
* Update yolo.py
* Update export.py
* fix export grid
* Update export.py, remove detect export attribute
* rearrange if order
* remove --grid, default inplace=False
* rename exp_dynamic to onnx_dynamic, comment
* replace bs with 1 in anchor_grid[i] index 0
* Update export.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-05-03 19:01:29 +02:00
41cc7caee6
Add NMS threshold checks ( #3020 )
...
* Add NMS threshold checks
* fix
2021-05-03 15:07:20 +02:00
fcd57028d2
Add is_colab() function ( #3018 )
2021-05-03 12:37:50 +02:00
e2a80c6c0f
Add support for FP16 (half) to export.py ( #3010 )
...
* Added support for fp16 (half) to export.py
* minimize code additions
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-05-02 22:42:33 +02:00
31ee54c28c
Hubconf.py bug fix ( #3007 )
2021-05-01 18:15:17 +02:00
7c89c829e3
cleanup
2021-05-01 17:48:12 +02:00
c1a44ed9c5
Update hubconf.py for unified loading ( #3005 )
2021-05-01 17:35:02 +02:00
37eaffec7d
Curl update ( #3004 )
...
* Curl update
* Curl update
2021-05-01 17:29:51 +02:00
4d7bca7648
Add unzip flag to download() ( #3002 )
2021-05-01 12:37:41 +02:00
f55730ee9c
Objects365 bug fix
2021-05-01 01:03:09 +02:00
54652fe3ff
Objects365 update
2021-05-01 00:16:58 +02:00
801b469878
Improve performance of dataset Logger ( #2943 )
...
* Improve performance of Dataset Logger
* Fix scaling bug
2021-04-30 21:54:26 +02:00
d08575ee5e
PyTorch Hub load directly when possible ( #2986 )
2021-04-30 14:59:51 +02:00
9b91db6d1a
Created using Colaboratory
2021-04-30 13:15:31 +02:00
41f5cc5637
YOLOv5 AWS Inferentia Inplace compatibility updates ( #2953 )
...
* Added flag to enable/disable all inplace and assignment operations
* Removed shape print statements
* Scope Detect/Model import to avoid circular dependency
* PEP8
* create _descale_pred()
* replace lost space
* replace list with tuple
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-04-30 12:54:48 +02:00
Glenn Jocher
Christoph Gerum
Cristi Fati
Yonghye Kwon
KC-Zhang
Ayush Chaurasia
kepler62f
Jiacong Fang
jylink
Hodovo
jluntamazon
