mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity
1,764 Commits 10 Branches 10 Tags
Commit Graph

84 Commits

Author SHA1 Message Date
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.
 2022-02-03 12:15:13 +01:00
Glenn Jocher
26f0415287
Add dataset source citations (#6032) 2021-12-19 15:19:04 +01:00
Glenn Jocher
abbdd4802e
train -> val comment fix (#6024) 2021-12-17 16:43:00 +01:00
Glenn Jocher
628817dfae
Recommend jar xf file.zip for zips (#5993) 2021-12-15 17:19:19 +01:00
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>
 2021-11-09 16:45:02 +01:00
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>
 2021-10-28 18:35:01 +02:00
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>
 2021-10-23 14:20:26 +02:00
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
 2021-10-18 14:24:48 +02:00
Glenn Jocher
0000334a6f
Update Objects365.yaml val count (#5212) 2021-10-15 12:40:40 -07:00
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>
 2021-10-14 20:48:47 -07:00
Glenn Jocher
1c9f710b46
Add hyp.scratch-med.yaml (#5196) 
* Add hyp.scratch-med.yaml

* Update hyp.scratch-med.yaml
 2021-10-14 20:40:30 -07:00
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
 2021-10-11 23:47:18 -07:00
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