Glenn Jocher
5783de26fe
Objects365 dataset breakdown images vs zips ( #7335 )
2 jaren geleden
Glenn Jocher
0ca85ed65f
Update Objects365.yaml ( #7323 )
Updated dataset size to 712GB (includes undeleted zips).
2 jaren geleden
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 jaren geleden
Glenn Jocher
f735458987
Use `tqdm.auto` ( #7311 )
2 jaren geleden
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 jaren geleden
Glenn Jocher
c759bbdf19
Add `check_requirements(('pycocotools>=2.0',))` ( #7295 )
Add `check_requirements(('pycocotools>=2.0',))`
2 jaren geleden
Glenn Jocher
b1300f3e0b
Add dataset sizes (zipped) ( #7293 )
2 jaren geleden
Glenn Jocher
5f97001ed4
Context manager `open(file) as f` fixes ( #7289 )
* Flask context manager `open()` fix
* Additional read context manager fixes
2 jaren geleden
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 jaren geleden
Glenn Jocher
cea994b3f6
Update hyp.VOC.yaml ( #6772 )
2 jaren geleden
Glenn Jocher
2692e67c5f
Updated VOC hyperparameters ( #6732 )
* Update hyps
* Update hyp.VOC.yaml
* Update pathlib
* Update hyps
* Update hyps
* Update hyps
* Update hyps
2 jaren geleden
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 jaren geleden
Glenn Jocher
26f0415287
Add dataset source citations ( #6032 )
2 jaren geleden
Glenn Jocher
abbdd4802e
train -> val comment fix ( #6024 )
2 jaren geleden
Glenn Jocher
628817dfae
Recommend `jar xf file.zip` for zips ( #5993 )
2 jaren geleden
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 jaren geleden
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 jaren geleden
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 jaren geleden
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 jaren geleden
Glenn Jocher
0000334a6f
Update Objects365.yaml val count ( #5212 )
3 jaren geleden
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 jaren geleden
Glenn Jocher
1c9f710b46
Add hyp.scratch-med.yaml ( #5196 )
* Add hyp.scratch-med.yaml
* Update hyp.scratch-med.yaml
3 jaren geleden
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 jaren geleden
Jebastin Nadar
1492632796
Fix zipfile name for coco128-segments ( #4914 )
3 jaren geleden
Glenn Jocher
24bea5e4b7
Standardize headers and docstrings ( #4417 )
* Implement new headers
* Reformat 1
* Reformat 2
* Reformat 3 - math
* Reformat 4 - yaml
3 jaren geleden
Glenn Jocher
3e7c59ad3b
Fix rename `utils.google_utils` to `utils.downloads` ( #4393 )
3 jaren geleden
Glenn Jocher
b60b62e874
PyCharm reformat ( #4209 )
* PyCharm reformat
* YAML reformat
* Markdown reformat
3 jaren geleden
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 jaren geleden
Glenn Jocher
3fef11706c
Update comments header ( #4184 )
3 jaren geleden
Glenn Jocher
0ad6301c96
Update script headers ( #4163 )
* Update download script headers
* cleanup
* bug fix attempt
* bug fix attempt2
* bug fix attempt3
* cleanup
3 jaren geleden
Glenn Jocher
f8e11483df
Update dataset headers ( #4162 )
3 jaren geleden
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 jaren geleden
Glenn Jocher
8e7f285051
Fix datasets for aws and get_coco.sh ( #3788 )
* merge master
* Update get_coco.sh
3 jaren geleden
Glenn Jocher
bd581b330b
Create hyp.scratch-p6.yaml ( #3787 )
3 jaren geleden
Glenn Jocher
f89941711c
NGA xView 2018 Dataset Auto-Download ( #3775 )
* update clip_coords for numpy
* uncomment
* cleanup
* Add autosplits
* fix
* cleanup
3 jaren geleden
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 jaren geleden
Glenn Jocher
fdc22398fa
Create `data/hyps` directory ( #3747 )
3 jaren geleden
Glenn Jocher
ac34834563
Update download_weights.sh comment ( #3662 )
3 jaren geleden
Glenn Jocher
2754adad46
Remove `/weights` directory ( #3659 )
* Remove `/weights` directory
* cleanup
3 jaren geleden
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 jaren geleden
Glenn Jocher
d6a71ae5f2
Update GlobalWheat2020.yaml test: # 1276 images ( #3101 )
3 jaren geleden
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 jaren geleden
Glenn Jocher
8cab907f60
download() ThreadPool update ( #3027 )
* download() ThreadPool update
* update train image count
* cid + 1
3 jaren geleden
Glenn Jocher
5189b3addb
Objects365 Update
3 jaren geleden
Glenn Jocher
37eaffec7d
Curl update ( #3004 )
* Curl update
* Curl update
3 jaren geleden
Glenn Jocher
f55730ee9c
Objects365 bug fix
3 jaren geleden
Glenn Jocher
54652fe3ff
Objects365 update
3 jaren geleden
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 jaren geleden
Glenn Jocher
33712d6dd0
Global Wheat Detection 2020 Dataset Auto-Download ( #2968 )
* Create GlobalWheat2020.yaml
* Update and rename visdrone.yaml to VisDrone.yaml
* Update GlobalWheat2020.yaml
3 jaren geleden
Glenn Jocher
de19d396e7
Update visdrone.yaml ( #2921 )
3 jaren geleden