Glenn Jocher
|
08d4918d7f
|
labels.jpg class names (#2454)
* labels.png class names
* fontsize=10
|
2021-03-12 22:15:41 -08:00 |
|
Glenn Jocher
|
f01f3223d5
|
Integer printout (#2450)
* Integer printout
* test.py 'Labels'
* Update train.py
|
2021-03-12 13:35:44 -08:00 |
|
Glenn Jocher
|
e8a2b83268
|
GCP sudo docker userdata.sh (#2393)
* GCP sudo docker
* cleanup
|
2021-03-07 20:21:49 -08:00 |
|
Glenn Jocher
|
7c2c95732c
|
AWS wait && echo "All tasks done." (#2391)
|
2021-03-07 20:18:30 -08:00 |
|
Glenn Jocher
|
ba18528b47
|
bbox_iou() stability and speed improvements (#2385)
|
2021-03-06 13:07:34 -08:00 |
Yonghye Kwon
|
cd8ed3521d
|
image weights compatible faster random index generator v2 for mosaic augmentation (#2383)
image weights compatible faster random index generator v2 for mosaic augmentation
|
2021-03-05 22:58:26 -08:00 |
Ryan Avery
|
2c56ad5436
|
Confusion matrix background axis swap (#2114)
|
2021-03-02 16:09:52 -08:00 |
|
Glenn Jocher
|
fab5085674
|
EMA bug fix 2 (#2330)
* EMA bug fix 2
* update
|
2021-03-01 17:13:34 -08:00 |
Iden Craven
|
a82dce7faa
|
Fix labels being missed when image extension appears twice in filename (#2300)
|
2021-02-25 18:05:38 -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
|
0070995bd5
|
Amazon AWS EC2 startup and re-startup scripts (#2282)
|
2021-02-24 01:43:59 -08:00 |
|
Glenn Jocher
|
d5d275b6e9
|
Amazon AWS EC2 startup and re-startup scripts (#2185)
* Amazon AWS EC2 startup and re-startup scripts
* Create resume.py
* cleanup
|
2021-02-23 23:10:14 -08:00 |
xiaowo1996
|
83dc1b4484
|
FLOPS min stride 32 (#2276)
Signed-off-by: xiaowo1996 <429740343@qq.com>
|
2021-02-23 09:38:56 -08:00 |
|
Glenn Jocher
|
6f5d6fcdaa
|
Robust objectness loss balancing (#2256)
|
2021-02-20 11:19:01 -08:00 |
|
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
|
2021-02-19 15:39:09 -08:00 |
|
Glenn Jocher
|
47faf95079
|
reset head
|
2021-02-19 15:20:41 -08:00 |
|
Glenn Jocher
|
d2e754b67b
|
Add isdocker() (#2232)
* Add isdocker()
* Update general.py
* Update general.py
|
2021-02-16 15:58:07 -08:00 |
|
Glenn Jocher
|
26c2e54c8f
|
Add check_imshow() (#2231)
* Add check_imshow()
* Update general.py
* Update general.py
|
2021-02-16 13:56:47 -08:00 |
|
Glenn Jocher
|
7b833e37bf
|
LoadStreams() frame loss bug fix (#2222)
|
2021-02-15 11:02:20 -08:00 |
|
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
|
2021-02-11 21:22:45 -08:00 |
NanoCode012
|
a5d5f9262d
|
Changed socket port and added timeout (#2176)
|
2021-02-10 13:01:48 -08:00 |
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.
|
2021-02-10 00:16:38 -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 |
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>
|
2021-02-05 11:12:43 -08:00 |
|
Glenn Jocher
|
4bdc5a397e
|
LoadImages() pathlib update (#2140)
|
2021-02-05 09:06:23 -08:00 |
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>
|
2021-02-01 21:38:41 -08:00 |
|
Glenn Jocher
|
9646ca438a
|
Update plot_study() (#2112)
|
2021-02-01 13:51:11 -08:00 |
|
Glenn Jocher
|
be9edffded
|
Update test.py --task speed and study (#2099)
* Add --speed benchmark
* test range 256 - 1536
* update
* update
* update
* update
|
2021-01-31 13:58:49 -08:00 |
|
Glenn Jocher
|
a18efc3a73
|
Add variable-stride inference support (#2091)
|
2021-01-30 13:47:23 -08:00 |
|
Glenn Jocher
|
630ec06af6
|
Update to colors.TABLEAU_COLORS (#2069)
|
2021-01-28 12:37:21 -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
|
f59f80114c
|
Add 'exclude' tuple to check_requirements() (#2041)
|
2021-01-26 21:17:36 -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
|
08d3119e09
|
Add histogram equalization fcn (#2049)
|
2021-01-26 17:30:42 -08:00 |
|
Glenn Jocher
|
8fcbe43423
|
Check im.format during dataset caching (#2042)
* Check im.format during dataset caching
* Update datasets.py
|
2021-01-25 20:55:35 -08:00 |
|
Glenn Jocher
|
3b7feeafdf
|
Update general.py check_git_status() fix (#2020)
|
2021-01-22 15:59:01 -08:00 |
|
Glenn Jocher
|
aac33f87ad
|
Update inference multiple-counting (#2019)
* Update inference multiple-counting
* update github check
|
2021-01-22 15:39:08 -08:00 |
|
Glenn Jocher
|
77fd83225f
|
Update google_utils.py (#2017)
|
2021-01-22 15:08:42 -08:00 |
|
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
|
2021-01-22 12:37:12 -08:00 |
|
Glenn Jocher
|
d9212140b3
|
Add xywhn2xyxy() (#1983)
|
2021-01-19 13:33:52 -08:00 |
|
Glenn Jocher
|
b1cf25dd9a
|
check_git_status() asserts (#1977)
|
2021-01-18 10:49:28 -08:00 |
Abhiram V
|
17751b9891
|
Update autoanchor.py with yaml.SafeLoader (#1971)
|
2021-01-18 10:47:42 -08:00 |
|
Abhiram V
|
35400dc7b7
|
Update plots.py with yaml.SafeLoader (#1968)
|
2021-01-18 10:46:20 -08:00 |
|
Glenn Jocher
|
b26a2f6242
|
check_git_status() when not exist /workspace (#1966)
|
2021-01-17 11:55:25 -08:00 |
|
Glenn Jocher
|
e8a41e8916
|
prevent check_git_status() in docker images (#1951)
* prevent check_git_status() running docker images
* Update general.py
|
2021-01-16 01:15:38 -08:00 |
|
Glenn Jocher
|
ca9babb8e6
|
Add ComputeLoss() class (#1950)
|
2021-01-15 13:50:24 -08:00 |
|
Glenn Jocher
|
b75c432ea0
|
GitHub API rate limit fallback (#1930)
|
2021-01-13 21:09:50 -08:00 |
|
Glenn Jocher
|
051e9e80de
|
check_git_status() bug fix (#1925)
|
2021-01-13 10:25:53 -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 |