60fe54449d
Update train.py ( #3099 )
преди 3 години
a833ee2a46
Update check_requirements() exclude list ( #2974 )
преди 3 години
f7bc685c2c
Implement yaml.safe_load() ( #2876 )
* Implement yaml.safe_load()
* yaml.safe_dump()
преди 3 години
c949fc86d1
Detection cropping+saving feature addition for detect.py and PyTorch Hub ( #2827 )
* Update detect.py
* Update detect.py
* Update greetings.yml
* Update cropping
* cleanup
* Update increment_path()
* Update common.py
* Update detect.py
* Update detect.py
* Update detect.py
* Update common.py
* cleanup
* Update detect.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
преди 3 години
6dd1083bbb
Tensorboard model visualization bug fix ( #2758 )
This fix should allow for visualizing YOLOv5 model graphs correctly in Tensorboard by uncommenting line 335 in train.py:
```python
if tb_writer:
tb_writer.add_graph(torch.jit.trace(model, imgs, strict=False), []) # add model graph
```
The problem was that the detect() layer checks the input size to adapt the grid if required, and tracing does not seem to like this shape check (even if the shape is fine and no grid recomputation is required). The following will warn:
0cae7576a9/train.py (L335)https://user-images.githubusercontent.com/26833433/114286928-349bd600-9a63-11eb-941f-7139ee6cd602.png ">
преди 3 години
1148e2ea63
Add TransformerLayer, TransformerBlock, C3TR modules ( #2333 )
* yolotr
* transformer block
* Remove bias in Transformer
* Remove C3T
* Remove a deprecated class
* put the 2nd LayerNorm into the 2nd residual block
* move example model to models/hub, rename to -transformer
* Add module comments and TODOs
* Remove LN in Transformer
* Add comments for Transformer
* Solve the problem of MA with DDP
* cleanup
* cleanup find_unused_parameters
* PEP8 reformat
Co-authored-by: DingYiwei <846414640@qq.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
преди 3 години
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 години
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 години
dc51e80b00
Fix: evolve with wandb ( #2634 )
преди 3 години
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 години
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 години
1bf9365280
W&B DDP fix ( #2574 )
преди 3 години
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 години
08d4918d7f
labels.jpg class names ( #2454 )
* labels.png class names
* fontsize=10
преди 3 години
f01f3223d5
Integer printout ( #2450 )
* Integer printout
* test.py 'Labels'
* Update train.py
преди 3 години
886f1c03d8
DDP after autoanchor reorder ( #2421 )
преди 3 години
e931b9da33
Resume with custom anchors fix ( #2361 )
* Resume with custom anchors fix
* Update train.py
преди 3 години
a3ecf0fd64
Anchor override ( #2350 )
преди 3 години
fab5085674
EMA bug fix 2 ( #2330 )
* EMA bug fix 2
* update
преди 3 години
dfeec198cb
final_epoch EMA bug fix ( #2317 )
преди 3 години
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 години
71dd2768f2
Improved model+EMA checkpointing 2 ( #2295 )
преди 3 години
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 години
ca5b10b759
Update train.py ( #2290 )
* Update train.py
* Update train.py
* Update train.py
* Update train.py
* Create train.py
преди 3 години
e27ca0d845
Update minimum stride to 32 ( #2266 )
преди 3 години
6b634c6b87
Linear LR scheduler option ( #2150 )
* Linear LR scheduler option
* Update train.py
преди 3 години
ad839eda38
Update train.py ( #2149 )
преди 3 години
86897e3663
Update train.py test batch_size ( #2148 )
* Update train.py
* Update loss.py
преди 3 години
2acbe9699a
W&B epoch logging update ( #2073 )
преди 3 години
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 години
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 години
046c37e465
verbose on final_epoch ( #1997 )
преди 3 години
e9941d50fa
Update train.py with yaml.SafeLoader ( #1972 )
преди 3 години
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 години
ca9babb8e6
Add ComputeLoss() class ( #1950 )
преди 3 години
f4a78e1b40
W&B mosaic log bug fix ( #1949 )
преди 3 години
777032efbc
Daemon thread mosaic plots fix ( #1943 )
преди 3 години
3a56cac414
Fix batch-size on resume for multi-gpu ( #1942 )
преди 3 години
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 години
dd03b20ba5
colorstr() updates ( #1909 )
* W&B ImportError message fix
* colorstr() updates
* colorstr() updates
* colorstr() default to 'blue', 'bold'
* train: magenta
* train: blue
преди 3 години
1d1c0567a4
PyTorch Hub results.render() ( #1897 )
преди 3 години
6ab589583c
Add colorstr() ( #1887 )
* Add colorful()
* update
* newline fix
* add git description
* --always
* update loss scaling
* update loss scaling 2
* rename to colorstr()
преди 3 години
e77c77f580
Add check_requirements() ( #1853 )
* Add check_requirements()
* add import
* parameterize filename
* add to detect, test
преди 3 години
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 години
0e341c5660
Create one_cycle() function ( #1836 )
преди 3 години
7180b22e2d
DDP Multi-GPU --resume bug fix ( #1810 )
преди 3 години
685d601308
Increase plot_labels() speed ( #1736 )
преди 3 години
7e161d9774
Single class train update ( #1719 )
преди 3 години
c923fbff90
W&B artifacts feature addition ( #1712 )
* Log artifacts
* cleanup
преди 3 години
8bc0027afc
Update loss criteria constructor ( #1711 )
преди 3 години
Glenn Jocher
Burhan
Ding Yiwei
Phat Tran
Ayush Chaurasia
Jan Hajek
Abhiram V
huntr.dev | the place to protect open source
NanoCode012
