Glenn Jocher
e931b9da33
Resume with custom anchors fix ( #2361 )
...
* Resume with custom anchors fix
* Update train.py
2021-03-03 21:06:36 -08:00
Glenn Jocher
a3ecf0fd64
Anchor override ( #2350 )
2021-03-02 23:08:21 -08:00
Glenn Jocher
fab5085674
EMA bug fix 2 ( #2330 )
...
* EMA bug fix 2
* update
2021-03-01 17:13:34 -08:00
Glenn Jocher
dfeec198cb
final_epoch EMA bug fix ( #2317 )
2021-02-27 12:51:33 -08:00
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>
2021-02-25 19:18:19 -08:00
Glenn Jocher
71dd2768f2
Improved model+EMA checkpointing 2 ( #2295 )
2021-02-24 21:03:21 -08:00
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
2021-02-24 18:26:46 -08:00
Glenn Jocher
ca5b10b759
Update train.py ( #2290 )
...
* Update train.py
* Update train.py
* Update train.py
* Update train.py
* Create train.py
2021-02-24 13:31:20 -08:00
Glenn Jocher
e27ca0d845
Update minimum stride to 32 ( #2266 )
2021-02-21 21:46:42 -08:00
Glenn Jocher
6b634c6b87
Linear LR scheduler option ( #2150 )
...
* Linear LR scheduler option
* Update train.py
2021-02-06 11:26:54 -08:00
Glenn Jocher
ad839eda38
Update train.py ( #2149 )
2021-02-06 11:21:04 -08:00
Glenn Jocher
86897e3663
Update train.py test batch_size ( #2148 )
...
* Update train.py
* Update loss.py
2021-02-06 10:29:32 -08:00
Glenn Jocher
2acbe9699a
W&B epoch logging update ( #2073 )
2021-01-28 18:57:24 -08:00
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
2021-01-27 17:10:53 -08:00
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
2021-01-26 21:16:01 -08:00
Glenn Jocher
046c37e465
verbose on final_epoch ( #1997 )
2021-01-20 18:27:38 -08:00
Abhiram V
e9941d50fa
Update train.py with yaml.SafeLoader ( #1972 )
2021-01-18 10:49:08 -08:00
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>
2021-01-17 14:10:16 -08:00
Glenn Jocher
ca9babb8e6
Add ComputeLoss() class ( #1950 )
2021-01-15 13:50:24 -08:00
Glenn Jocher
f4a78e1b40
W&B mosaic log bug fix ( #1949 )
2021-01-15 12:09:53 -08:00
Glenn Jocher
777032efbc
Daemon thread mosaic plots fix ( #1943 )
2021-01-14 13:16:48 -08:00
NanoCode012
3a56cac414
Fix batch-size on resume for multi-gpu ( #1942 )
2021-01-14 09:53:13 -08:00
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
2021-01-12 21:51:49 -08:00
Glenn Jocher
dd03b20ba5
colorstr() updates ( #1909 )
...
* W&B ImportError message fix
* colorstr() updates
* colorstr() updates
* colorstr() default to 'blue', 'bold'
* train: magenta
* train: blue
2021-01-12 10:33:15 -08:00
Glenn Jocher
1d1c0567a4
PyTorch Hub results.render() ( #1897 )
2021-01-10 20:06:11 -08:00
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()
2021-01-09 15:24:18 -08:00
Glenn Jocher
e77c77f580
Add check_requirements() ( #1853 )
...
* Add check_requirements()
* add import
* parameterize filename
* add to detect, test
2021-01-06 16:35:40 -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
0e341c5660
Create one_cycle() function ( #1836 )
2021-01-04 15:49:08 -08:00
Glenn Jocher
7180b22e2d
DDP Multi-GPU --resume bug fix ( #1810 )
2020-12-30 12:40:45 -08:00
Glenn Jocher
685d601308
Increase plot_labels() speed ( #1736 )
2020-12-18 18:05:38 -08:00
Glenn Jocher
7e161d9774
Single class train update ( #1719 )
2020-12-17 12:02:03 -08:00
Glenn Jocher
c923fbff90
W&B artifacts feature addition ( #1712 )
...
* Log artifacts
* cleanup
2020-12-16 17:52:12 -08:00
Glenn Jocher
8bc0027afc
Update loss criteria constructor ( #1711 )
2020-12-16 08:39:35 -08:00
Glenn Jocher
7947c86b57
Update COCO train postprocessing ( #1702 )
2020-12-15 21:50:28 -08:00
Glenn Jocher
791dadb51c
Pycocotools best.pt after COCO train ( #1616 )
...
* Pycocotools best.pt after COCO train
* cleanup
2020-12-06 14:58:33 +01:00
Glenn Jocher
b6ed1104a6
Daemon thread plotting ( #1561 )
...
* Daemon thread plotting
* remove process_batch
* plot after print
2020-11-30 16:44:14 +01:00
Glenn Jocher
9728e2b8ae
--image_weights bug fix ( #1524 )
2020-11-26 11:49:01 +01:00
Glenn Jocher
bde5d9aaaa
Update caching ( #1496 )
2020-11-24 16:23:00 +01:00
Glenn Jocher
0822cda781
Update caching ( #1496 )
2020-11-24 16:22:02 +01:00
Glenn Jocher
0a3ff71ae0
Confusion matrix ( #1474 )
...
* initial commit
* add plotting
* matrix to cpu
* bug fix
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* seaborn pandas to requirements.txt
* seaborn pandas to requirements.txt
* update wandb plotting
* remove pandas
* if plots
* if plots
* if plots
* if plots
* if plots
* initial commit
* add plotting
* matrix to cpu
* bug fix
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* seaborn pandas to requirements.txt
* seaborn pandas to requirements.txt
* update wandb plotting
* remove pandas
* if plots
* if plots
* if plots
* if plots
* if plots
* Cat apriori to autolabels
* cleanup
2020-11-23 15:27:14 +01:00
asim266
9ef4760b74
Rename precision_recall_curve.png bug fix ( #1431 )
...
* fix error in train.py line 400, changed precision-recall_curve.png to precision_recall_curve.png and fix error in test.py of {wandb.init() before wandb.log()} by importing wand and initializing it before wand.log()
* Update test.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-11-17 23:47:27 +01: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
0c26c4e831
Increase pycocotools robustness ( #1396 )
2020-11-14 13:53:58 +01:00
Glenn Jocher
fe341fa44d
Utils reorganization ( #1392 )
...
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
2020-11-14 11:50:32 +01:00
Glenn Jocher
379396e896
Yaml constructor posixpath --resume bug fix ( #1390 )
...
* resume fix for yaml constructor posixpath error
* fix update
* remove weights/ dir backup
2020-11-13 19:03:02 +01:00
Glenn Jocher
c4addd7761
Unified '/project/name' results saving ( #1377 )
...
* Project/name update
* Update ci-testing.yml
* address project with path separator failure mode
* Project/name update
* address project with path separator failure mode
* Update ci-testing.yml
* detect.py default --name bug fix
* missing rstrip PR
* train/exp0 to train/exp
2020-11-12 23:37:46 +01:00
Glenn Jocher
ce3872f86c
W&B evolution bug fix ( #1373 )
2020-11-12 12:55:52 +01:00
Khiem Doan
1c8464e199
Use pathlib instead of low-level module ( #1329 )
...
* Use pathlib instead of low-level module
* Use pathlib instead of low-level module
* Update detect.py
* Update test.py
* reformat
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-11-10 13:01:19 +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
