Glenn Jocher
6f5d6fcdaa
Robust objectness loss balancing ( #2256 )
3 years ago
Glenn Jocher
c09964c27c
Update inference default to multi_label=False ( #2252 )
* Update inference default to multi_label=False
* bug fix
* Update plots.py
* Update plots.py
3 years ago
Glenn Jocher
47faf95079
reset head
3 years ago
Glenn Jocher
d2e754b67b
Add isdocker() ( #2232 )
* Add isdocker()
* Update general.py
* Update general.py
3 years ago
Glenn Jocher
26c2e54c8f
Add check_imshow() ( #2231 )
* Add check_imshow()
* Update general.py
* Update general.py
3 years ago
Glenn Jocher
7b833e37bf
LoadStreams() frame loss bug fix ( #2222 )
3 years ago
Glenn Jocher
bdd88e1ed7
YOLOv5 Segmentation Dataloader Updates ( #2188 )
* 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
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* GitHub API rate limit fix
* update
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* ComputeLoss
* astuple
* epochs
* update
* update
* ComputeLoss()
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* merge
* merge
* merge
* merge
* update
* update
* update
* update
* commit=tag == tags[-1]
* Update cudnn.benchmark
* update
* update
* update
* updates
* updates
* updates
* updates
* updates
* updates
* updates
* update
* update
* update
* update
* update
* mosaic9
* update
* update
* update
* update
* update
* update
* institute cache versioning
* only display on existing cache
* reverse cache exists booleans
3 years ago
NanoCode012
a5d5f9262d
Changed socket port and added timeout ( #2176 )
3 years ago
Transigent
c9bda112ae
Update utils/datasets.py to support .webp files ( #2174 )
Simply added 'webp' as an image format to the img_formats array so that webp image files can be used as training data.
3 years ago
Glenn Jocher
86897e3663
Update train.py test batch_size ( #2148 )
* Update train.py
* Update loss.py
3 years ago
train255
e9b3de463a
Unique *.cache filenames fix ( #2134 )
* fix #2121
* Update test.py
* Update train.py
* Update autoanchor.py
* Update datasets.py
* Update log_dataset.py
* Update datasets.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
4bdc5a397e
LoadImages() pathlib update ( #2140 )
3 years ago
Ayush Chaurasia
73a0669930
Start setup for improved W&B integration ( #1948 )
* Add helper functions for wandb and artifacts
* cleanup
* Reorganize files
* Update wandb_utils.py
* Update log_dataset.py
We can remove this code, as the giou hyp has been deprecated for a while now.
* Reorganize and update dataloader call
* yaml.SafeLoader
* PEP8 reformat
* remove redundant checks
* Add helper functions for wandb and artifacts
* cleanup
* Reorganize files
* Update wandb_utils.py
* Update log_dataset.py
We can remove this code, as the giou hyp has been deprecated for a while now.
* Reorganize and update dataloader call
* yaml.SafeLoader
* PEP8 reformat
* remove redundant checks
* Update util files
* Update wandb_utils.py
* Remove word size
* Change path of labels.zip
* remove unused imports
* remove --rect
* log_dataset.py cleanup
* log_dataset.py cleanup2
* wandb_utils.py cleanup
* remove redundant id_count
* wandb_utils.py cleanup2
* rename cls
* use pathlib for zip
* rename dataloader to dataset
* Change import order
* Remove redundant code
* remove unused import
* remove unused imports
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
9646ca438a
Update plot_study() ( #2112 )
3 years ago
Glenn Jocher
be9edffded
Update test.py --task speed and study ( #2099 )
* Add --speed benchmark
* test range 256 - 1536
* update
* update
* update
* update
3 years ago
Glenn Jocher
a18efc3a73
Add variable-stride inference support ( #2091 )
3 years ago
Glenn Jocher
630ec06af6
Update to colors.TABLEAU_COLORS ( #2069 )
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
f59f80114c
Add 'exclude' tuple to check_requirements() ( #2041 )
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
08d3119e09
Add histogram equalization fcn ( #2049 )
3 years ago
Glenn Jocher
8fcbe43423
Check im.format during dataset caching ( #2042 )
* Check im.format during dataset caching
* Update datasets.py
3 years ago
Glenn Jocher
3b7feeafdf
Update general.py check_git_status() fix ( #2020 )
3 years ago
Glenn Jocher
aac33f87ad
Update inference multiple-counting ( #2019 )
* Update inference multiple-counting
* update github check
3 years ago
Glenn Jocher
77fd83225f
Update google_utils.py ( #2017 )
3 years ago
Glenn Jocher
1445ab2b27
check_git_status() Windows fix ( #2015 )
* check_git_status() Windows fix
* Update general.py
* Update general.py
* Update general.py
* Update general.py
* Update general.py
* Update general.py
3 years ago
Glenn Jocher
d9212140b3
Add xywhn2xyxy() ( #1983 )
3 years ago
Glenn Jocher
b1cf25dd9a
check_git_status() asserts ( #1977 )
3 years ago
Abhiram V
17751b9891
Update autoanchor.py with yaml.SafeLoader ( #1971 )
3 years ago
Abhiram V
35400dc7b7
Update plots.py with yaml.SafeLoader ( #1968 )
3 years ago
Glenn Jocher
b26a2f6242
check_git_status() when not exist /workspace ( #1966 )
3 years ago
Glenn Jocher
e8a41e8916
prevent check_git_status() in docker images ( #1951 )
* prevent check_git_status() running docker images
* Update general.py
3 years ago
Glenn Jocher
ca9babb8e6
Add ComputeLoss() class ( #1950 )
3 years ago
Glenn Jocher
b75c432ea0
GitHub API rate limit fallback ( #1930 )
3 years ago
Glenn Jocher
051e9e80de
check_git_status() bug fix ( #1925 )
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
ffef77124e
Update torch_utils.py ( #1895 )
3 years ago
Glenn Jocher
0f11aaf551
GitHub API rate limit newline ( #1894 )
3 years ago
Glenn Jocher
d06ad3bc3d
GitHub API rate limit fix ( #1894 )
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
fea9c9b80c
nn.SiLU() citation correction ( #1713 )
3 years ago
Glenn Jocher
e77c77f580
Add check_requirements() ( #1853 )
* Add check_requirements()
* add import
* parameterize filename
* add to detect, test
3 years ago
Tommy in Tongji
135ec5c5ce
W&B ID reset on training completion ( #1852 )
* Update train.py
Fix the bug of always the same W&B ID and continue overwrite with the old logging.
BUG report
https://github.com/ultralytics/yolov5/issues/1851
* Fix the bug of duplicate W&B ID
fix the bug of https://github.com/ultralytics/yolov5/issues/1851
If we had trained on yolov5s.pt, the program will generate a new unique W&B ID.
If we hadn't, the program will keep the old code, we can still use --resume aug.
* Update general.py
* revert train.py changes
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
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
9f5a18bb80
Torch CUDA synchronize update ( #1826 )
* torch.cuda.synchronize() update
* torch.cuda.synchronize() update
* torch.cuda.synchronize() update
* newline
3 years ago
Glenn Jocher
c0ffcdf998
Display correct CUDA devices ( #1776 )
* Display correct CUDA devices
* cleanup
3 years ago
Glenn Jocher
3004fb5bc1
Automatic m.half() profile on x.half()
3 years ago