Glenn Jocher
487edd5609
Dataset autodownload fstring updates
2 лет назад
Glenn Jocher
b53917de8d
Remove `tqdm.auto` ( #7599 )
2 лет назад
Glenn Jocher
5783de26fe
Objects365 dataset breakdown images vs zips ( #7335 )
2 лет назад
Glenn Jocher
0ca85ed65f
Update Objects365.yaml ( #7323 )
Updated dataset size to 712GB (includes undeleted zips).
2 лет назад
Glenn Jocher
32661f75ac
Add `retry=3` to `download()` ( #7313 )
* Add `retry=3` to `download()`
* Update general.py
* Update general.py
* Update general.py
* Update VOC.yaml
* Update VisDrone.yaml
2 лет назад
Glenn Jocher
f735458987
Use `tqdm.auto` ( #7311 )
2 лет назад
Max Strobel
741fac815e
fix: disable usage of root logger ( #7296 )
* fix: disable usage of root logger
`logging.basicConfig` configures Python's root logger. This prohibits
fine control of logging, overwrites logging configuration done outside
the package, and is not best practice. Instead, the used logger is now
configured directly, and the root logger is untouched.
Example:
If yolov5 is used as part of another project with some sophisticated
logging, the internal `logging.basicConfig` call overwrites all the
external configuration.
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update general.py
* Update general.py
* Comment kaggle
* Uncomment kaggle
Co-authored-by: Maximilian Strobel <Maximilian.Strobel@infineon.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 лет назад
Glenn Jocher
c759bbdf19
Add `check_requirements(('pycocotools>=2.0',))` ( #7295 )
Add `check_requirements(('pycocotools>=2.0',))`
2 лет назад
Glenn Jocher
b1300f3e0b
Add dataset sizes (zipped) ( #7293 )
2 лет назад
Glenn Jocher
5f97001ed4
Context manager `open(file) as f` fixes ( #7289 )
* Flask context manager `open()` fix
* Additional read context manager fixes
2 лет назад
Glenn Jocher
c3ae4e4af6
Multi-threaded VisDrone and VOC downloads ( #7108 )
* Multi-threaded VOC download
* Update VOC.yaml
* Update
* Update general.py
* Update general.py
2 лет назад
Glenn Jocher
cea994b3f6
Update hyp.VOC.yaml ( #6772 )
2 лет назад
Glenn Jocher
2692e67c5f
Updated VOC hyperparameters ( #6732 )
* Update hyps
* Update hyp.VOC.yaml
* Update pathlib
* Update hyps
* Update hyps
* Update hyps
* Update hyps
2 лет назад
Glenn Jocher
19e0208fc9
Update hyp.scratch-high.yaml ( #6525 )
Update `lrf: 0.1`, tested on YOLOv5x6 to 55.0 mAP@0.5:0.95, slightly higher than current.
2 лет назад
Glenn Jocher
26f0415287
Add dataset source citations ( #6032 )
2 лет назад
Glenn Jocher
abbdd4802e
train -> val comment fix ( #6024 )
2 лет назад
Glenn Jocher
628817dfae
Recommend `jar xf file.zip` for zips ( #5993 )
2 лет назад
Glenn Jocher
3883261143
New `DetectMultiBackend()` class ( #5549 )
* New `DetectMultiBackend()` class
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* pb to pt fix
* Cleanup
* explicit apply_classifier path
* Cleanup2
* Cleanup3
* Cleanup4
* Cleanup5
* Cleanup6
* val.py MultiBackend inference
* warmup fix
* to device fix
* pt fix
* device fix
* Val cleanup
* COCO128 URL to assets
* half fix
* detect fix
* detect fix 2
* remove half from DetectMultiBackend
* training half handling
* training half handling 2
* training half handling 3
* Cleanup
* Fix CI error
* Add torchscript _extra_files
* Add TorchScript
* Add CoreML
* CoreML cleanup
* New `DetectMultiBackend()` class
* pb to pt fix
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Cleanup
* explicit apply_classifier path
* Cleanup2
* Cleanup3
* Cleanup4
* Cleanup5
* Cleanup6
* val.py MultiBackend inference
* warmup fix
* to device fix
* pt fix
* device fix
* Val cleanup
* COCO128 URL to assets
* half fix
* detect fix
* detect fix 2
* remove half from DetectMultiBackend
* training half handling
* training half handling 2
* training half handling 3
* Cleanup
* Fix CI error
* Add torchscript _extra_files
* Add TorchScript
* Add CoreML
* CoreML cleanup
* revert default to pt
* Add Usage examples
* Cleanup val
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
3 лет назад
Jirka Borovec
ed887b5976
Add pre-commit CI actions ( #4982 )
* define pre-commit
* add CI code
* configure
* apply pre-commit
* fstring
* apply MD
* pre-commit
* Update torch_utils.py
* Update print strings
* notes
* Cleanup code-format.yml
* Update setup.cfg
* Update .pre-commit-config.yaml
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 лет назад
Cristi Fati
fee83c1634
Weights download script minor improvements ( #5213 )
* Add nano model to the download list
* Add possibility to also download the "*6" models
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 лет назад
Glenn Jocher
13f7275555
Update/inplace ops ( #5233 )
* Clip Objects365 autodownload labels (#5214 )
Fixes out of bounds labels that seem to affect ~10% of images in dataset.
* Inplace ops
3 лет назад
Glenn Jocher
0000334a6f
Update Objects365.yaml val count ( #5212 )
3 лет назад
Farley Lai
fc3606420d
Update Objects365.yaml to include the official validation set ( #5194 )
* Update Objects365.yaml
Download the official Objects365 validation set and convert the labels
* Enforce 4-space indent, reformat and cleanup
* shorten list comprehension
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 лет назад
Glenn Jocher
1c9f710b46
Add hyp.scratch-med.yaml ( #5196 )
* Add hyp.scratch-med.yaml
* Update hyp.scratch-med.yaml
3 лет назад
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
3 лет назад
Jebastin Nadar
1492632796
Fix zipfile name for coco128-segments ( #4914 )
3 лет назад
Glenn Jocher
24bea5e4b7
Standardize headers and docstrings ( #4417 )
* Implement new headers
* Reformat 1
* Reformat 2
* Reformat 3 - math
* Reformat 4 - yaml
3 лет назад
Glenn Jocher
3e7c59ad3b
Fix rename `utils.google_utils` to `utils.downloads` ( #4393 )
3 лет назад
Glenn Jocher
b60b62e874
PyCharm reformat ( #4209 )
* PyCharm reformat
* YAML reformat
* Markdown reformat
3 лет назад
Glenn Jocher
5d66e48723
Train from `--data path/to/dataset.zip` feature ( #4185 )
* Train from `--data path/to/dataset.zip` feature
* Update dataset_stats()
* cleanup
* cleanup2
3 лет назад
Glenn Jocher
3fef11706c
Update comments header ( #4184 )
3 лет назад
Glenn Jocher
0ad6301c96
Update script headers ( #4163 )
* Update download script headers
* cleanup
* bug fix attempt
* bug fix attempt2
* bug fix attempt3
* cleanup
3 лет назад
Glenn Jocher
f8e11483df
Update dataset headers ( #4162 )
3 лет назад
Glenn Jocher
c6c88dc601
Copy-Paste augmentation for YOLOv5 ( #3845 )
* Copy-paste augmentation initial commit
* if any segments
* Add obscuration rejection
* Add copy_paste hyperparameter
* Update comments
3 лет назад
Glenn Jocher
8e7f285051
Fix datasets for aws and get_coco.sh ( #3788 )
* merge master
* Update get_coco.sh
3 лет назад
Glenn Jocher
bd581b330b
Create hyp.scratch-p6.yaml ( #3787 )
3 лет назад
Glenn Jocher
f89941711c
NGA xView 2018 Dataset Auto-Download ( #3775 )
* update clip_coords for numpy
* uncomment
* cleanup
* Add autosplits
* fix
* cleanup
3 лет назад
Glenn Jocher
f79d7479da
Add optional dataset.yaml `path` attribute ( #3753 )
* Add optional dataset.yaml `path` attribute
@KalenMike
* pass locals to python scripts
* handle lists
* update coco128.yaml
* Capitalize first letter
* add test key
* finalize GlobalWheat2020.yaml
* finalize objects365.yaml
* finalize SKU-110K.yaml
* finalize SKU-110K.yaml
* finalize VisDrone.yaml
* NoneType fix
* update download comment
* voc to VOC
* update
* update VOC.yaml
* update VOC.yaml
* remove dashes
* delete get_voc.sh
* force coco and coco128 to ../datasets
* Capitalize Argoverse_HD.yaml
* Capitalize Objects365.yaml
* update Argoverse_HD.yaml
* coco segments fix
* VOC single-thread
* update Argoverse_HD.yaml
* update data_dict in test handling
* create root
3 лет назад
Glenn Jocher
fdc22398fa
Create `data/hyps` directory ( #3747 )
3 лет назад
Glenn Jocher
ac34834563
Update download_weights.sh comment ( #3662 )
3 лет назад
Glenn Jocher
2754adad46
Remove `/weights` directory ( #3659 )
* Remove `/weights` directory
* cleanup
3 лет назад
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
3 лет назад
Glenn Jocher
d6a71ae5f2
Update GlobalWheat2020.yaml test: # 1276 images ( #3101 )
3 лет назад
Jiacong Fang
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>
3 лет назад
Glenn Jocher
8cab907f60
download() ThreadPool update ( #3027 )
* download() ThreadPool update
* update train image count
* cid + 1
3 лет назад
Glenn Jocher
5189b3addb
Objects365 Update
3 лет назад
Glenn Jocher
37eaffec7d
Curl update ( #3004 )
* Curl update
* Curl update
3 лет назад
Glenn Jocher
f55730ee9c
Objects365 bug fix
3 лет назад
Glenn Jocher
54652fe3ff
Objects365 update
3 лет назад
Ferdinand Loesch
dbce1bc54c
Objects365 Dataset ( #2932 )
* add object365
* ADD CONVERSION SCRIPT
* fix transcript
* Reformat and simplify
* spelling
* Update get_objects365.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 лет назад