 Ryan Avery
|
2c56ad5436
|
Confusion matrix background axis swap (#2114)
|
3 years ago |
 Glenn Jocher
|
fab5085674
|
EMA bug fix 2 (#2330)
* EMA bug fix 2
* update
|
3 years ago |
 Iden Craven
|
a82dce7faa
|
Fix labels being missed when image extension appears twice in filename (#2300)
|
3 years ago |
|
Glenn Jocher
|
ec1d8496ba
|
Improved model+EMA checkpointing (#2292)
* Enhanced model+EMA checkpointing
* update
* bug fix
* bug fix 2
* always save optimizer
* ema half
* remove model.float()
* model half
* carry ema/model in fp32
* rm model.float()
* both to float always
* cleanup
* cleanup
|
3 years ago |
|
Glenn Jocher
|
0070995bd5
|
Amazon AWS EC2 startup and re-startup scripts (#2282)
|
3 years ago |
|
Glenn Jocher
|
d5d275b6e9
|
Amazon AWS EC2 startup and re-startup scripts (#2185)
* Amazon AWS EC2 startup and re-startup scripts
* Create resume.py
* cleanup
|
3 years ago |
 xiaowo1996
|
83dc1b4484
|
FLOPS min stride 32 (#2276)
Signed-off-by: xiaowo1996 <429740343@qq.com>
|
3 years ago |
|
Glenn Jocher
|
6f5d6fcdaa
|
Robust objectness loss balancing (#2256)
|
3 years ago |
|
Glenn Jocher
|
c09964c27c
|
Update inference default to multi_label=False (#2252)
* Update inference default to multi_label=False
* bug fix
* Update plots.py
* Update plots.py
|
3 years ago |
|
Glenn Jocher
|
47faf95079
|
reset head
|
3 years ago |
|
Glenn Jocher
|
d2e754b67b
|
Add isdocker() (#2232)
* Add isdocker()
* Update general.py
* Update general.py
|
3 years ago |
|
Glenn Jocher
|
26c2e54c8f
|
Add check_imshow() (#2231)
* Add check_imshow()
* Update general.py
* Update general.py
|
3 years ago |
|
Glenn Jocher
|
7b833e37bf
|
LoadStreams() frame loss bug fix (#2222)
|
3 years ago |
|
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
|
3 years ago |
 NanoCode012
|
a5d5f9262d
|
Changed socket port and added timeout (#2176)
|
3 years ago |
 Transigent
|
c9bda112ae
|
Update utils/datasets.py to support .webp files (#2174)
Simply added 'webp' as an image format to the img_formats array so that webp image files can be used as training data.
|
3 years ago |
|
Glenn Jocher
|
86897e3663
|
Update train.py test batch_size (#2148)
* Update train.py
* Update loss.py
|
3 years ago |
 train255
|
e9b3de463a
|
Unique *.cache filenames fix (#2134)
* fix #2121
* Update test.py
* Update train.py
* Update autoanchor.py
* Update datasets.py
* Update log_dataset.py
* Update datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
3 years ago |
|
Glenn Jocher
|
4bdc5a397e
|
LoadImages() pathlib update (#2140)
|
3 years ago |
 Ayush Chaurasia
|
73a0669930
|
Start setup for improved W&B integration (#1948)
* Add helper functions for wandb and artifacts
* cleanup
* Reorganize files
* Update wandb_utils.py
* Update log_dataset.py
We can remove this code, as the giou hyp has been deprecated for a while now.
* Reorganize and update dataloader call
* yaml.SafeLoader
* PEP8 reformat
* remove redundant checks
* Add helper functions for wandb and artifacts
* cleanup
* Reorganize files
* Update wandb_utils.py
* Update log_dataset.py
We can remove this code, as the giou hyp has been deprecated for a while now.
* Reorganize and update dataloader call
* yaml.SafeLoader
* PEP8 reformat
* remove redundant checks
* Update util files
* Update wandb_utils.py
* Remove word size
* Change path of labels.zip
* remove unused imports
* remove --rect
* log_dataset.py cleanup
* log_dataset.py cleanup2
* wandb_utils.py cleanup
* remove redundant id_count
* wandb_utils.py cleanup2
* rename cls
* use pathlib for zip
* rename dataloader to dataset
* Change import order
* Remove redundant code
* remove unused import
* remove unused imports
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
3 years ago |
|
Glenn Jocher
|
9646ca438a
|
Update plot_study() (#2112)
|
3 years ago |
|
Glenn Jocher
|
be9edffded
|
Update test.py --task speed and study (#2099)
* Add --speed benchmark
* test range 256 - 1536
* update
* update
* update
* update
|
3 years ago |
|
Glenn Jocher
|
a18efc3a73
|
Add variable-stride inference support (#2091)
|
3 years ago |
|
Glenn Jocher
|
630ec06af6
|
Update to colors.TABLEAU_COLORS (#2069)
|
3 years ago |
|
Glenn Jocher
|
f639e14e4d
|
Metric-Confidence plots feature addition (#2057)
* Metric-Confidence plots feature addition
* cleanup
* Metric-Confidence plots feature addition
* cleanup
* Update run-once lines
* cleanup
* save all 4 curves to wandb
|
3 years ago |
|
Glenn Jocher
|
f59f80114c
|
Add 'exclude' tuple to check_requirements() (#2041)
|
3 years ago |
|
Glenn Jocher
|
59c21c7bcb
|
W&B log epoch (#1946)
* W&B log epoch
* capitalize
* W&B log epoch
* capitalize
* Update train.py
New try using https://docs.wandb.ai/library/log#incremental-logging
* Update train.py
* Update test.py
* Update train.py
* Update plots.py
* Update train.py
* Update train.py
* label plot step -1
* update
* update
* update
* update
* update
* update
* Update train.py
* Update train.py
|
3 years ago |
|
Glenn Jocher
|
08d3119e09
|
Add histogram equalization fcn (#2049)
|
3 years ago |
|
Glenn Jocher
|
8fcbe43423
|
Check im.format during dataset caching (#2042)
* Check im.format during dataset caching
* Update datasets.py
|
3 years ago |
|
Glenn Jocher
|
3b7feeafdf
|
Update general.py check_git_status() fix (#2020)
|
3 years ago |
|
Glenn Jocher
|
aac33f87ad
|
Update inference multiple-counting (#2019)
* Update inference multiple-counting
* update github check
|
3 years ago |
|
Glenn Jocher
|
77fd83225f
|
Update google_utils.py (#2017)
|
3 years ago |
|
Glenn Jocher
|
1445ab2b27
|
check_git_status() Windows fix (#2015)
* check_git_status() Windows fix
* Update general.py
* Update general.py
* Update general.py
* Update general.py
* Update general.py
* Update general.py
|
3 years ago |
|
Glenn Jocher
|
d9212140b3
|
Add xywhn2xyxy() (#1983)
|
3 years ago |
|
Glenn Jocher
|
b1cf25dd9a
|
check_git_status() asserts (#1977)
|
3 years ago |
 Abhiram V
|
17751b9891
|
Update autoanchor.py with yaml.SafeLoader (#1971)
|
3 years ago |
|
Abhiram V
|
35400dc7b7
|
Update plots.py with yaml.SafeLoader (#1968)
|
3 years ago |
|
Glenn Jocher
|
b26a2f6242
|
check_git_status() when not exist /workspace (#1966)
|
3 years ago |
|
Glenn Jocher
|
e8a41e8916
|
prevent check_git_status() in docker images (#1951)
* prevent check_git_status() running docker images
* Update general.py
|
3 years ago |
|
Glenn Jocher
|
ca9babb8e6
|
Add ComputeLoss() class (#1950)
|
3 years ago |
|
Glenn Jocher
|
b75c432ea0
|
GitHub API rate limit fallback (#1930)
|
3 years ago |
|
Glenn Jocher
|
051e9e80de
|
check_git_status() bug fix (#1925)
|
3 years ago |
|
Glenn Jocher
|
509dd51aca
|
check_git_status() improvements (#1916)
* check_online()
* Update general.py
* update check_git_status()
* reverse rev-parse order
* fetch
* improved responsiveness
* comment
* comment
* remove hyp['giou'] compat warning
|
3 years ago |
|
Glenn Jocher
|
dd03b20ba5
|
colorstr() updates (#1909)
* W&B ImportError message fix
* colorstr() updates
* colorstr() updates
* colorstr() default to 'blue', 'bold'
* train: magenta
* train: blue
|
3 years ago |
|
Glenn Jocher
|
1d1c0567a4
|
PyTorch Hub results.render() (#1897)
|
3 years ago |
|
Glenn Jocher
|
ffef77124e
|
Update torch_utils.py (#1895)
|
3 years ago |
|
Glenn Jocher
|
0f11aaf551
|
GitHub API rate limit newline (#1894)
|
3 years ago |
|
Glenn Jocher
|
d06ad3bc3d
|
GitHub API rate limit fix (#1894)
|
3 years ago |
|
Glenn Jocher
|
6ab589583c
|
Add colorstr() (#1887)
* Add colorful()
* update
* newline fix
* add git description
* --always
* update loss scaling
* update loss scaling 2
* rename to colorstr()
|
3 years ago |
|
Glenn Jocher
|
fea9c9b80c
|
nn.SiLU() citation correction (#1713)
|
3 years ago |
