yolov5/utils
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>
2021-03-23 00:44:50 +01:00
..
aws GCP sudo docker userdata.sh (#2393) 2021-03-07 20:21:49 -08:00
google_app_engine adding the configuration to deploy yolov5 in in app engine (#964) 2020-09-22 11:48:44 -07:00
wandb_logging Improved W&B integration (#2125) 2021-03-23 00:44:50 +01:00
__init__.py initial commit 2020-05-29 17:04:54 -07:00
activations.py nn.SiLU() citation correction (#1713) 2021-01-06 20:58:41 -08:00
autoanchor.py Scipy kmeans-robust autoanchor update (#2470) 2021-03-14 21:58:12 -07:00
datasets.py PyTorch Hub models default to CUDA:0 if available (#2472) 2021-03-14 23:16:17 -07:00
general.py PyTorch Hub models default to CUDA:0 if available (#2472) 2021-03-14 23:16:17 -07:00
google_utils.py Update google_utils.py (#2017) 2021-01-22 15:08:42 -08:00
loss.py Robust objectness loss balancing (#2256) 2021-02-20 11:19:01 -08:00
metrics.py Confusion matrix background axis swap (#2114) 2021-03-02 16:09:52 -08:00
plots.py labels.jpg class names (#2454) 2021-03-12 22:15:41 -08:00
torch_utils.py PyTorch Hub models default to CUDA:0 if available (#2472) 2021-03-14 23:16:17 -07:00