1492632796
Fix zipfile name for coco128-segments ( #4914 )
3 лет назад
24bea5e4b7
Standardize headers and docstrings ( #4417 )
* Implement new headers
* Reformat 1
* Reformat 2
* Reformat 3 - math
* Reformat 4 - yaml
3 лет назад
3e7c59ad3b
Fix rename `utils.google_utils` to `utils.downloads` ( #4393 )
3 лет назад
b60b62e874
PyCharm reformat ( #4209 )
* PyCharm reformat
* YAML reformat
* Markdown reformat
3 лет назад
5d66e48723
Train from `--data path/to/dataset.zip` feature ( #4185 )
* Train from `--data path/to/dataset.zip` feature
* Update dataset_stats()
* cleanup
* cleanup2
3 лет назад
3fef11706c
Update comments header ( #4184 )
3 лет назад
0ad6301c96
Update script headers ( #4163 )
* Update download script headers
* cleanup
* bug fix attempt
* bug fix attempt2
* bug fix attempt3
* cleanup
3 лет назад
f8e11483df
Update dataset headers ( #4162 )
3 лет назад
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 лет назад
8e7f285051
Fix datasets for aws and get_coco.sh ( #3788 )
* merge master
* Update get_coco.sh
3 лет назад
bd581b330b
Create hyp.scratch-p6.yaml ( #3787 )
3 лет назад
f89941711c
NGA xView 2018 Dataset Auto-Download ( #3775 )
* update clip_coords for numpy
* uncomment
* cleanup
* Add autosplits
* fix
* cleanup
3 лет назад
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 лет назад
fdc22398fa
Create `data/hyps` directory ( #3747 )
3 лет назад
ac34834563
Update download_weights.sh comment ( #3662 )
3 лет назад
2754adad46
Remove `/weights` directory ( #3659 )
* Remove `/weights` directory
* cleanup
3 лет назад
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 лет назад
d6a71ae5f2
Update GlobalWheat2020.yaml test: # 1276 images ( #3101 )
3 лет назад
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 лет назад
8cab907f60
download() ThreadPool update ( #3027 )
* download() ThreadPool update
* update train image count
* cid + 1
3 лет назад
5189b3addb
Objects365 Update
3 лет назад
37eaffec7d
Curl update ( #3004 )
* Curl update
* Curl update
3 лет назад
f55730ee9c
Objects365 bug fix
3 лет назад
54652fe3ff
Objects365 update
3 лет назад
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 лет назад
33712d6dd0
Global Wheat Detection 2020 Dataset Auto-Download ( #2968 )
* Create GlobalWheat2020.yaml
* Update and rename visdrone.yaml to VisDrone.yaml
* Update GlobalWheat2020.yaml
3 лет назад
de19d396e7
Update visdrone.yaml ( #2921 )
3 лет назад
78fd077657
VisDrone2019-DET Dataset Auto-Download ( #2882 )
* VisDrone Dataset Auto-Download
* add visdrone.yaml
* cleanup
* add VisDrone2019-DET-test-dev
* cleanup VOC
3 лет назад
f7bc685c2c
Implement yaml.safe_load() ( #2876 )
* Implement yaml.safe_load()
* yaml.safe_dump()
3 лет назад
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 лет назад
d5ca8ca34e
CVPR 2021 Argoverse-HD autodownload fix ( #2418 )
3 лет назад
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 лет назад
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 лет назад
a5359f6c32
Update data-autodownload background tasks ( #2154 )
* Update get_coco.sh
* Update get_voc.sh
3 лет назад
a41d910c5d
data-autodownload background tasks ( #2034 )
3 лет назад
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 лет назад
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 лет назад
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 лет назад
194f16844e
Multi-threaded image caching
4 лет назад
4346b13a40
Dataset download bash script updates ( #1132 )
4 лет назад
0ada058f63
Generalized regression criterion renaming ( #1120 )
4 лет назад
2d2371d2d1
Update get_voc.sh data download URLs ( #1077 )
4 лет назад
f1c63e2784
add mosaic and warmup to hyperparameters ( #931 )
4 лет назад
23a7f29e2d
Update hyp.finetune.yaml on 306 VOC generations result
4 лет назад
901243c780
Fix relative path for VOC download ( #880 )
Cleaned up code
4 лет назад
08e97a2f88
Update hyperparameters to add lrf, anchors
4 лет назад
fd71fe8451
Update hyp.finetune.yaml
4 лет назад
1309400e6b
Update hyp.finetune.yaml on VOC results
4 лет назад
916d4aad9a
v3.0 Release ( #725 )
* initial commit
* remove yolov3-spp from test.py study
* update study --img range
* update mAP
* cleanup and speed updates
* update README plot
4 лет назад
41523e2c91
Dataset autodownload feature addition ( #685 )
* initial commit
* move download scripts into data/scripts
* new check_dataset() function in general.py
* move check_dataset() out of with context
* Update general.py
* DDP update
* Update general.py
4 лет назад
Jebastin Nadar
Glenn Jocher
Jiacong Fang
Ferdinand Loesch
Kartikeya Sharma
Mingyu Wang
NanoCode012