Yukun Xia
35fe031461
Fix: add P2 layer 21 to yolov5-p2.yaml `Detect()` inputs ( #4608 )
Layer 21 includes the information of xsmall objects
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
4e8c81a368
Add `yolov5s-ghost.yaml` ( #4412 )
* Add yolov5s-ghost.yaml
* Finish C3Ghost
* Add C3Ghost to list
* Add C3Ghost to number of repeats if statement
* Fixes
* Cleanup
3 лет назад
Glenn Jocher
b60b62e874
PyCharm reformat ( #4209 )
* PyCharm reformat
* YAML reformat
* Markdown reformat
3 лет назад
Glenn Jocher
e016b15555
Update yolov5-bifpn.yaml ( #4208 )
3 лет назад
Glenn Jocher
1f31b7c503
Create yolov5-bifpn.yaml ( #4195 )
3 лет назад
Glenn Jocher
81b31824f5
Models `*.yaml` reformat ( #3875 )
3 лет назад
Ding Yiwei
1148e2ea63
Add TransformerLayer, TransformerBlock, C3TR modules ( #2333 )
* yolotr
* transformer block
* Remove bias in Transformer
* Remove C3T
* Remove a deprecated class
* put the 2nd LayerNorm into the 2nd residual block
* move example model to models/hub, rename to -transformer
* Add module comments and TODOs
* Remove LN in Transformer
* Add comments for Transformer
* Solve the problem of MA with DDP
* cleanup
* cleanup find_unused_parameters
* PEP8 reformat
Co-authored-by: DingYiwei <846414640@qq.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 лет назад
Glenn Jocher
9646ca438a
Update plot_study() ( #2112 )
3 лет назад
Glenn Jocher
6e6f77be47
Add YOLOv5-P6 models ( #2083 )
3 лет назад
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
3 лет назад
Glenn Jocher
0b6266f5e0
Update yolov3-tiny.yaml ( #1823 )
3 лет назад
Glenn Jocher
791dadb51c
Pycocotools best.pt after COCO train ( #1616 )
* Pycocotools best.pt after COCO train
* cleanup
3 лет назад
Glenn Jocher
97c6a59dd1
PANet anchor order correction ( #1420 )
4 лет назад
Liu Changyu
adaf1d8ac1
Update models/hub *.yaml files for v2.0 ( #546 )
* fix #543 issue
* Update yolov3-spp.yaml
* Update yolov5-fpn.yaml
* Update yolov5-panet.yaml
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 лет назад
Glenn Jocher
0825cb7fd8
create models/hub
4 лет назад