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 anos atrás
0000334a6f
Update Objects365.yaml val count ( #5212 )
3 anos atrás
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 anos atrás
1c9f710b46
Add hyp.scratch-med.yaml ( #5196 )
* Add hyp.scratch-med.yaml
* Update hyp.scratch-med.yaml
3 anos atrás
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 anos atrás
1492632796
Fix zipfile name for coco128-segments ( #4914 )
3 anos atrás
24bea5e4b7
Standardize headers and docstrings ( #4417 )
* Implement new headers
* Reformat 1
* Reformat 2
* Reformat 3 - math
* Reformat 4 - yaml
3 anos atrás
3e7c59ad3b
Fix rename `utils.google_utils` to `utils.downloads` ( #4393 )
3 anos atrás
b60b62e874
PyCharm reformat ( #4209 )
* PyCharm reformat
* YAML reformat
* Markdown reformat
3 anos atrás
5d66e48723
Train from `--data path/to/dataset.zip` feature ( #4185 )
* Train from `--data path/to/dataset.zip` feature
* Update dataset_stats()
* cleanup
* cleanup2
3 anos atrás
3fef11706c
Update comments header ( #4184 )
3 anos atrás
0ad6301c96
Update script headers ( #4163 )
* Update download script headers
* cleanup
* bug fix attempt
* bug fix attempt2
* bug fix attempt3
* cleanup
3 anos atrás
f8e11483df
Update dataset headers ( #4162 )
3 anos atrás
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 anos atrás
8e7f285051
Fix datasets for aws and get_coco.sh ( #3788 )
* merge master
* Update get_coco.sh
3 anos atrás
bd581b330b
Create hyp.scratch-p6.yaml ( #3787 )
3 anos atrás
f89941711c
NGA xView 2018 Dataset Auto-Download ( #3775 )
* update clip_coords for numpy
* uncomment
* cleanup
* Add autosplits
* fix
* cleanup
3 anos atrás
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 anos atrás
fdc22398fa
Create `data/hyps` directory ( #3747 )
3 anos atrás
ac34834563
Update download_weights.sh comment ( #3662 )
3 anos atrás
2754adad46
Remove `/weights` directory ( #3659 )
* Remove `/weights` directory
* cleanup
3 anos atrás
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 anos atrás
d6a71ae5f2
Update GlobalWheat2020.yaml test: # 1276 images ( #3101 )
3 anos atrás
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 anos atrás
8cab907f60
download() ThreadPool update ( #3027 )
* download() ThreadPool update
* update train image count
* cid + 1
3 anos atrás
5189b3addb
Objects365 Update
3 anos atrás
37eaffec7d
Curl update ( #3004 )
* Curl update
* Curl update
3 anos atrás
f55730ee9c
Objects365 bug fix
3 anos atrás
54652fe3ff
Objects365 update
3 anos atrás
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 anos atrás
33712d6dd0
Global Wheat Detection 2020 Dataset Auto-Download ( #2968 )
* Create GlobalWheat2020.yaml
* Update and rename visdrone.yaml to VisDrone.yaml
* Update GlobalWheat2020.yaml
3 anos atrás
de19d396e7
Update visdrone.yaml ( #2921 )
3 anos atrás
78fd077657
VisDrone2019-DET Dataset Auto-Download ( #2882 )
* VisDrone Dataset Auto-Download
* add visdrone.yaml
* cleanup
* add VisDrone2019-DET-test-dev
* cleanup VOC
3 anos atrás
f7bc685c2c
Implement yaml.safe_load() ( #2876 )
* Implement yaml.safe_load()
* yaml.safe_dump()
3 anos atrás
747c2653ee
CVPR 2021 Argoverse-HD autodownload curl ( #2455 )
curl preferred over wget for slightly better cross platform compatibility (i.e. out of the box macos compatible).
3 anos atrás
d5ca8ca34e
CVPR 2021 Argoverse-HD autodownload fix ( #2418 )
3 anos atrás
c51dfec8ea
CVPR 2021 Argoverse-HD dataset autodownload support ( #2400 )
* added argoverse-download ability
* bugfix
* add support for Argoverse dataset
* Refactored code
* renamed to argoverse-HD
* unzip -q and YOLOv5
small cleanup items
* add image counts
Co-authored-by: Kartikeya Sharma <kartikes@trinity.vision.cs.cmu.edu>
Co-authored-by: Kartikeya Sharma <kartikes@trinity-0-32.eth>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 anos atrás
bdd88e1ed7
YOLOv5 Segmentation Dataloader Updates ( #2188 )
* Update C3 module
* Update C3 module
* Update C3 module
* Update C3 module
* update
* update
* update
* update
* update
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update datasets
* update
* update
* update
* update attempt_downlaod()
* merge
* merge
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* parameterize eps
* comments
* gs-multiple
* update
* max_nms implemented
* Create one_cycle() function
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* GitHub API rate limit fix
* update
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* astuple
* epochs
* update
* update
* ComputeLoss()
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* merge
* merge
* merge
* merge
* update
* update
* update
* update
* commit=tag == tags[-1]
* Update cudnn.benchmark
* update
* update
* update
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* update
* mosaic9
* update
* update
* update
* update
* update
* update
* institute cache versioning
* only display on existing cache
* reverse cache exists booleans
3 anos atrás
a5359f6c32
Update data-autodownload background tasks ( #2154 )
* Update get_coco.sh
* Update get_voc.sh
3 anos atrás
a41d910c5d
data-autodownload background tasks ( #2034 )
3 anos atrás
69be8e738f
YOLOv5 v4.0 Release ( #1837 )
* Update C3 module
* Update C3 module
* Update C3 module
* Update C3 module
* update
* update
* update
* update
* update
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update datasets
* update
* update
* update
* update attempt_downlaod()
* merge
* merge
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* parameterize eps
* comments
* gs-multiple
* update
* max_nms implemented
* Create one_cycle() function
* update
* update
* update
* update
* update
* update
* update
* update study.png
* update study.png
* Update datasets.py
3 anos atrás
9c91aeae10
W&B logging add hyperparameters ( #1399 )
* W&B logging add hyperparameters
* hyp bug fix and image logging updates
* if plots and wandb:
* cleanup
* wandb/ gitignore add
* cleanup 2
* cleanup 3
* move wandb import to top of file
* wandb evolve
* update import
* wandb.run.finish()
* default anchors: 3
4 anos atrás
4821d076e2
Increment train, test, detect runs/ ( #1322 )
* Increment train, test, detect runs/
* Update ci-testing.yml
* inference/images to data/images
* move images
* runs/exp to runs/train/exp
* update 'results saved to %s' str
4 anos atrás
194f16844e
Multi-threaded image caching
4 anos atrás
4346b13a40
Dataset download bash script updates ( #1132 )
4 anos atrás
0ada058f63
Generalized regression criterion renaming ( #1120 )
4 anos atrás
2d2371d2d1
Update get_voc.sh data download URLs ( #1077 )
4 anos atrás
f1c63e2784
add mosaic and warmup to hyperparameters ( #931 )
4 anos atrás
23a7f29e2d
Update hyp.finetune.yaml on 306 VOC generations result
4 anos atrás
901243c780
Fix relative path for VOC download ( #880 )
Cleaned up code
4 anos atrás
Glenn Jocher
Farley Lai
Jebastin Nadar
Jiacong Fang
Ferdinand Loesch
Kartikeya Sharma
Mingyu Wang
NanoCode012