 Phat Tran
|
9c803f2f7e
|
Add --label-smoothing eps argument to train.py (default 0.0) (#2344)
* Add label smoothing option
* Correct data type
* add_log
* Remove log
* Add log
* Update loss.py
remove comment (too versbose)
Co-authored-by: phattran <phat.tranhoang@cyberlogitec.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
3 years ago |
 Ayush Chaurasia
|
518c09578e
|
W&B resume ddp from run link fix (#2579)
* W&B resume ddp from run link fix
* Native DDP W&B support for training, resuming
|
3 years ago |
|
Ayush Chaurasia
|
dc51e80b00
|
Fix: evolve with wandb (#2634)
|
3 years ago |
 Glenn Jocher
|
9f98201dd9
|
W&B DDP fix 2 (#2587)
Revert unintentional change to test batch sizes caused by PR https://github.com/ultralytics/yolov5/pull/2125
|
3 years ago |
|
Glenn Jocher
|
e5b0200cd2
|
Update tensorboard>=2.4.1 (#2576)
* Update tensorboard>=2.4.1
Update tensorboard version to attempt to address https://github.com/ultralytics/yolov5/issues/2573 (tensorboard logging fail in Docker image).
* cleanup
|
3 years ago |
|
Ayush Chaurasia
|
1bf9365280
|
W&B DDP fix (#2574)
|
3 years ago |
|
Ayush Chaurasia
|
e8fc97aa38
|
Improved W&B integration (#2125)
* Init Commit
* new wandb integration
* Update
* Use data_dict in test
* Updates
* Update: scope of log_img
* Update: scope of log_img
* Update
* Update: Fix logging conditions
* Add tqdm bar, support for .txt dataset format
* Improve Result table Logger
* Init Commit
* new wandb integration
* Update
* Use data_dict in test
* Updates
* Update: scope of log_img
* Update: scope of log_img
* Update
* Update: Fix logging conditions
* Add tqdm bar, support for .txt dataset format
* Improve Result table Logger
* Add dataset creation in training script
* Change scope: self.wandb_run
* Add wandb-artifact:// natively
you can now use --resume with wandb run links
* Add suuport for logging dataset while training
* Cleanup
* Fix: Merge conflict
* Fix: CI tests
* Automatically use wandb config
* Fix: Resume
* Fix: CI
* Enhance: Using val_table
* More resume enhancement
* FIX : CI
* Add alias
* Get useful opt config data
* train.py cleanup
* Cleanup train.py
* more cleanup
* Cleanup| CI fix
* Reformat using PEP8
* FIX:CI
* rebase
* remove uneccesary changes
* remove uneccesary changes
* remove uneccesary changes
* remove unecessary chage from test.py
* FIX: resume from local checkpoint
* FIX:resume
* FIX:resume
* Reformat
* Performance improvement
* Fix local resume
* Fix local resume
* FIX:CI
* Fix: CI
* Imporve image logging
* (:(:Redo CI tests:):)
* Remember epochs when resuming
* Remember epochs when resuming
* Update DDP location
Potential fix for #2405
* PEP8 reformat
* 0.25 confidence threshold
* reset train.py plots syntax to previous
* reset epochs completed syntax to previous
* reset space to previous
* remove brackets
* reset comment to previous
* Update: is_coco check, remove unused code
* Remove redundant print statement
* Remove wandb imports
* remove dsviz logger from test.py
* Remove redundant change from test.py
* remove redundant changes from train.py
* reformat and improvements
* Fix typo
* Add tqdm tqdm progress when scanning files, naming improvements
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
3 years ago |
|
Glenn Jocher
|
08d4918d7f
|
labels.jpg class names (#2454)
* labels.png class names
* fontsize=10
|
3 years ago |
|
Glenn Jocher
|
f01f3223d5
|
Integer printout (#2450)
* Integer printout
* test.py 'Labels'
* Update train.py
|
3 years ago |
|
Glenn Jocher
|
886f1c03d8
|
DDP after autoanchor reorder (#2421)
|
3 years ago |
|
Glenn Jocher
|
e931b9da33
|
Resume with custom anchors fix (#2361)
* Resume with custom anchors fix
* Update train.py
|
3 years ago |
|
Glenn Jocher
|
a3ecf0fd64
|
Anchor override (#2350)
|
3 years ago |
|
Glenn Jocher
|
fab5085674
|
EMA bug fix 2 (#2330)
* EMA bug fix 2
* update
|
3 years ago |
|
Glenn Jocher
|
dfeec198cb
|
final_epoch EMA bug fix (#2317)
|
3 years ago |
 Jan Hajek
|
efa4946d15
|
W&B entity support (#2298)
* W&B entity support
* shorten wandb_entity to entity
Co-authored-by: Jan Hajek <jan.hajek@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
|
3 years ago |
|
Glenn Jocher
|
71dd2768f2
|
Improved model+EMA checkpointing 2 (#2295)
|
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
|
ca5b10b759
|
Update train.py (#2290)
* Update train.py
* Update train.py
* Update train.py
* Update train.py
* Create train.py
|
3 years ago |
|
Glenn Jocher
|
e27ca0d845
|
Update minimum stride to 32 (#2266)
|
3 years ago |
|
Glenn Jocher
|
6b634c6b87
|
Linear LR scheduler option (#2150)
* Linear LR scheduler option
* Update train.py
|
3 years ago |
|
Glenn Jocher
|
ad839eda38
|
Update train.py (#2149)
|
3 years ago |
|
Glenn Jocher
|
86897e3663
|
Update train.py test batch_size (#2148)
* Update train.py
* Update loss.py
|
3 years ago |
|
Glenn Jocher
|
2acbe9699a
|
W&B epoch logging update (#2073)
|
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
|
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
|
046c37e465
|
verbose on final_epoch (#1997)
|
3 years ago |
 Abhiram V
|
e9941d50fa
|
Update train.py with yaml.SafeLoader (#1972)
|
3 years ago |
 huntr.dev | the place to protect open source
|
b5d851d653
|
Security Fix for Arbitrary Code Execution - huntr.dev (#1962)
Co-authored-by: Anon-Artist <61599526+Anon-Artist@users.noreply.github.com>
Co-authored-by: Jamie Slome <jamie@418sec.com>
|
3 years ago |
|
Glenn Jocher
|
ca9babb8e6
|
Add ComputeLoss() class (#1950)
|
3 years ago |
|
Glenn Jocher
|
f4a78e1b40
|
W&B mosaic log bug fix (#1949)
|
3 years ago |
|
Glenn Jocher
|
777032efbc
|
Daemon thread mosaic plots fix (#1943)
|
3 years ago |
 NanoCode012
|
3a56cac414
|
Fix batch-size on resume for multi-gpu (#1942)
|
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
|
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
|
e77c77f580
|
Add check_requirements() (#1853)
* Add check_requirements()
* add import
* parameterize filename
* add to detect, test
|
3 years ago |
|
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 years ago |
|
Glenn Jocher
|
0e341c5660
|
Create one_cycle() function (#1836)
|
3 years ago |
|
Glenn Jocher
|
7180b22e2d
|
DDP Multi-GPU --resume bug fix (#1810)
|
3 years ago |
|
Glenn Jocher
|
685d601308
|
Increase plot_labels() speed (#1736)
|
3 years ago |
|
Glenn Jocher
|
7e161d9774
|
Single class train update (#1719)
|
3 years ago |
|
Glenn Jocher
|
c923fbff90
|
W&B artifacts feature addition (#1712)
* Log artifacts
* cleanup
|
3 years ago |
|
Glenn Jocher
|
8bc0027afc
|
Update loss criteria constructor (#1711)
|
3 years ago |
|
Glenn Jocher
|
7947c86b57
|
Update COCO train postprocessing (#1702)
|
3 years ago |
|
Glenn Jocher
|
791dadb51c
|
Pycocotools best.pt after COCO train (#1616)
* Pycocotools best.pt after COCO train
* cleanup
|
3 years ago |
|
Glenn Jocher
|
b6ed1104a6
|
Daemon thread plotting (#1561)
* Daemon thread plotting
* remove process_batch
* plot after print
|
4 years ago |
|
Glenn Jocher
|
9728e2b8ae
|
--image_weights bug fix (#1524)
|
4 years ago |
|
Glenn Jocher
|
bde5d9aaaa
|
Update caching (#1496)
|
4 years ago |
|
Glenn Jocher
|
0822cda781
|
Update caching (#1496)
|
4 years ago |
