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 години