046c37e465
verbose on final_epoch ( #1997 )
3 years ago
e9941d50fa
Update train.py with yaml.SafeLoader ( #1972 )
3 years ago
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
ca9babb8e6
Add ComputeLoss() class ( #1950 )
3 years ago
f4a78e1b40
W&B mosaic log bug fix ( #1949 )
3 years ago
777032efbc
Daemon thread mosaic plots fix ( #1943 )
3 years ago
3a56cac414
Fix batch-size on resume for multi-gpu ( #1942 )
3 years ago
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
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
1d1c0567a4
PyTorch Hub results.render() ( #1897 )
3 years ago
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
e77c77f580
Add check_requirements() ( #1853 )
* Add check_requirements()
* add import
* parameterize filename
* add to detect, test
3 years ago
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
0e341c5660
Create one_cycle() function ( #1836 )
3 years ago
7180b22e2d
DDP Multi-GPU --resume bug fix ( #1810 )
3 years ago
685d601308
Increase plot_labels() speed ( #1736 )
3 years ago
7e161d9774
Single class train update ( #1719 )
3 years ago
c923fbff90
W&B artifacts feature addition ( #1712 )
* Log artifacts
* cleanup
3 years ago
8bc0027afc
Update loss criteria constructor ( #1711 )
3 years ago
7947c86b57
Update COCO train postprocessing ( #1702 )
3 years ago
791dadb51c
Pycocotools best.pt after COCO train ( #1616 )
* Pycocotools best.pt after COCO train
* cleanup
3 years ago
b6ed1104a6
Daemon thread plotting ( #1561 )
* Daemon thread plotting
* remove process_batch
* plot after print
4 years ago
9728e2b8ae
--image_weights bug fix ( #1524 )
4 years ago
bde5d9aaaa
Update caching ( #1496 )
4 years ago
0822cda781
Update caching ( #1496 )
4 years ago
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
4 years ago
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>
4 years ago
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
4 years ago
0c26c4e831
Increase pycocotools robustness ( #1396 )
4 years ago
fe341fa44d
Utils reorganization ( #1392 )
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
4 years ago
379396e896
Yaml constructor posixpath --resume bug fix ( #1390 )
* resume fix for yaml constructor posixpath error
* fix update
* remove weights/ dir backup
4 years ago
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
4 years ago
ce3872f86c
W&B evolution bug fix ( #1373 )
4 years ago
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>
4 years ago
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
4 years ago
187f7c2ed1
Change optimizer parameters group method ( #1239 )
* Change optimizer parameters group method
* Add torch nn
* Change isinstance method(torch.Tensor to nn.Parameter)
* parameter freeze fix, PEP8 reformat
* freeze bug fix
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 years ago
96fcde40b8
W&B feature improvements ( #1258 )
* W&B feature improvements
This PR add:
* Class to id labels. Now, the caption of bounding boxes will display the class name and the class confidence score.
* The project name is set to "Yolov5" and the run name will be set to opt.logdir
* cleanup
* remove parenthesis on caption
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 years ago
ca290dca24
Weights & Biases (W&B) Feature Addition ( #1235 )
* Add wandb metric logging and bounding box debugging
* Improve formatting, readability
* Remove mutliple path for init, improve formatting
* Add wandb params
* Remove typecasting in bbox coordinates and reformat
* Cleanup
* add wandb to requirements.txt
* minor updates to test.py
* general reorg
* reduce --log-imgs to 10
* clean wandb import
* reverse wandb import assert
* add except AssertionError to try import
* move wandb init to all global ranks
* replace print() with logger.info()
* replace print() with logger.info()
* move wandb.init() bug fix
* project PosixPath to basename bug fix
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 years ago
453acdec67
Update tensorboard logging
4 years ago
c67e72200e
fix compatibility for hyper config ( #1146 )
* fix/hyper
* Hyp giou check to train.py
* restore general.py
* train.py overwrite fix
* restore general.py and pep8 update
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 years ago
4d3680c81d
Minor import and spelling updates ( #1133 )
4 years ago
00917a6225
update expt name comment and folder parsing for training ( #978 )
* comment
* fix parsing
* fix evolve
* folder
* tqdm
* Update train.py
* Update train.py
* reinstate anchors into meta dict
anchor evolution is working correctly now
* reinstate logger
prefer the single line readout for concise logging, which helps simplify notebook and tutorials etc.
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 years ago
0ada058f63
Generalized regression criterion renaming ( #1120 )
4 years ago
5fac5ad165
Precision-Recall Curve Feature Addition ( #1107 )
* initial commit
* Update general.py
Indent update
* Update general.py
refactor duplicate code
* 200 dpi
4 years ago
66676eb039
init_torch_seeds >> init_seeds bug fix
4 years ago
f1c63e2784
add mosaic and warmup to hyperparameters ( #931 )
4 years ago
a62a45b2dd
prevent testloader caching on --notest
4 years ago
c8e51812a5
hyp evolution force-autoanchor fix
4 years ago
c687d5c129
reorganize train initialization steps
4 years ago
44cdcc7e0b
hyp['anchors'] evolution update
4 years ago
Glenn Jocher
Abhiram V
huntr.dev | the place to protect open source
NanoCode012
asim266
Khiem Doan
Junghoon Kim
Ayush Chaurasia
Jirka Borovec