Jebastin Nadar
1492632796
Fix zipfile name for coco128-segments ( #4914 )
2021-09-25 06:43:24 -07:00
Glenn Jocher
24bea5e4b7
Standardize headers and docstrings ( #4417 )
...
* Implement new headers
* Reformat 1
* Reformat 2
* Reformat 3 - math
* Reformat 4 - yaml
2021-08-14 21:17:51 +02:00
Glenn Jocher
3e7c59ad3b
Fix rename `utils.google_utils` to `utils.downloads` ( #4393 )
2021-08-11 23:40:47 +02:00
Glenn Jocher
b60b62e874
PyCharm reformat ( #4209 )
...
* PyCharm reformat
* YAML reformat
* Markdown reformat
2021-07-28 23:35:14 +02:00
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
2021-07-28 02:04:10 +02:00
Glenn Jocher
3fef11706c
Update comments header ( #4184 )
2021-07-27 23:23:41 +02:00
Glenn Jocher
0ad6301c96
Update script headers ( #4163 )
...
* Update download script headers
* cleanup
* bug fix attempt
* bug fix attempt2
* bug fix attempt3
* cleanup
2021-07-26 15:23:33 +02:00
Glenn Jocher
f8e11483df
Update dataset headers ( #4162 )
2021-07-26 14:23:43 +02:00
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
2021-07-01 00:35:04 +02:00
Glenn Jocher
8e7f285051
Fix datasets for aws and get_coco.sh ( #3788 )
...
* merge master
* Update get_coco.sh
2021-06-26 15:29:16 +02:00
Glenn Jocher
bd581b330b
Create hyp.scratch-p6.yaml ( #3787 )
2021-06-26 15:13:06 +02:00
Glenn Jocher
f89941711c
NGA xView 2018 Dataset Auto-Download ( #3775 )
...
* update clip_coords for numpy
* uncomment
* cleanup
* Add autosplits
* fix
* cleanup
2021-06-26 00:49:05 +02:00
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
2021-06-25 01:25:03 +02:00
Glenn Jocher
fdc22398fa
Create `data/hyps` directory ( #3747 )
2021-06-23 12:49:38 +02:00
Glenn Jocher
ac34834563
Update download_weights.sh comment ( #3662 )
2021-06-17 16:21:10 +02:00
Glenn Jocher
2754adad46
Remove `/weights` directory ( #3659 )
...
* Remove `/weights` directory
* cleanup
2021-06-17 14:44:12 +02:00
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
2021-05-14 17:05:26 +02:00
Glenn Jocher
d6a71ae5f2
Update GlobalWheat2020.yaml test: # 1276 images ( #3101 )
2021-05-10 16:04:10 +02:00
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>
2021-05-06 13:44:14 +02:00
Glenn Jocher
8cab907f60
download() ThreadPool update ( #3027 )
...
* download() ThreadPool update
* update train image count
* cid + 1
2021-05-04 23:37:31 +02:00
Glenn Jocher
5189b3addb
Objects365 Update
2021-05-04 02:25:14 +02:00
Glenn Jocher
37eaffec7d
Curl update ( #3004 )
...
* Curl update
* Curl update
2021-05-01 17:29:51 +02:00
Glenn Jocher
f55730ee9c
Objects365 bug fix
2021-05-01 01:03:09 +02:00
Glenn Jocher
54652fe3ff
Objects365 update
2021-05-01 00:16:58 +02:00
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>
2021-04-29 20:43:33 +02:00
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
2021-04-28 20:11:02 +02:00
Glenn Jocher
de19d396e7
Update visdrone.yaml ( #2921 )
2021-04-24 16:28:27 +02:00
Glenn Jocher
78fd077657
VisDrone2019-DET Dataset Auto-Download ( #2882 )
...
* VisDrone Dataset Auto-Download
* add visdrone.yaml
* cleanup
* add VisDrone2019-DET-test-dev
* cleanup VOC
2021-04-22 12:10:26 +02:00
Glenn Jocher
f7bc685c2c
Implement yaml.safe_load() ( #2876 )
...
* Implement yaml.safe_load()
* yaml.safe_dump()
2021-04-21 14:34:45 +02:00
Glenn Jocher
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).
2021-03-12 22:27:53 -08:00
Glenn Jocher
d5ca8ca34e
CVPR 2021 Argoverse-HD autodownload fix ( #2418 )
2021-03-09 21:07:27 -08:00
Kartikeya Sharma
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>
2021-03-09 19:26:49 -08:00
Glenn Jocher
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
2021-02-11 21:22:45 -08:00
Glenn Jocher
a5359f6c32
Update data-autodownload background tasks ( #2154 )
...
* Update get_coco.sh
* Update get_voc.sh
2021-02-06 22:13:39 -08:00
Glenn Jocher
a41d910c5d
data-autodownload background tasks ( #2034 )
2021-01-24 18:01:58 -08:00
Glenn Jocher
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
2021-01-04 19:54:09 -08:00
Glenn Jocher
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
2020-11-16 13:35:34 +01:00
Glenn Jocher
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
2020-11-08 19:39:05 +01:00
Glenn Jocher
194f16844e
Multi-threaded image caching
2020-11-07 02:18:18 +01:00
Glenn Jocher
4346b13a40
Dataset download bash script updates ( #1132 )
2020-10-13 15:58:03 +02:00
Glenn Jocher
0ada058f63
Generalized regression criterion renaming ( #1120 )
2020-10-11 17:25:17 +02:00
Mingyu Wang
2d2371d2d1
Update get_voc.sh data download URLs ( #1077 )
2020-10-04 19:35:05 +02:00
Glenn Jocher
f1c63e2784
add mosaic and warmup to hyperparameters ( #931 )
2020-09-13 14:03:54 -07:00
Glenn Jocher
23a7f29e2d
Update hyp.finetune.yaml on 306 VOC generations result
2020-09-03 01:20:30 -07:00
NanoCode012
901243c780
Fix relative path for VOC download ( #880 )
...
Cleaned up code
2020-08-31 16:06:10 -07:00
Glenn Jocher
08e97a2f88
Update hyperparameters to add lrf, anchors
2020-08-28 14:58:43 -07:00
Glenn Jocher
fd71fe8451
Update hyp.finetune.yaml
2020-08-22 16:18:53 -07:00
Glenn Jocher
1309400e6b
Update hyp.finetune.yaml on VOC results
2020-08-22 15:05:47 -07:00
Glenn Jocher
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
2020-08-13 14:25:05 -07:00
Glenn Jocher
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
2020-08-09 20:52:57 -07:00