Glenn Jocher
2d41e70e82
Scipy kmeans-robust autoanchor update ( #2470 )
Fix for https://github.com/ultralytics/yolov5/issues/2394
hace 3 años
Yann Defretin
38ff499b26
Update autosplit() with annotated_only option ( #2466 )
* Be able to create dataset from annotated images only
Add the ability to create a dataset/splits only with images that have an annotation file, i.e a .txt file, associated to it. As we talked about this, the absence of a txt file could mean two things:
* either the image wasn't yet labelled by someone,
* either there is no object to detect.
When it's easy to create small datasets, when you have to create datasets with thousands of images (and more coming), it's hard to track where you at and you don't want to wait to have all of them annotated before starting to train. Which means some images would lack txt files and annotations, resulting in label inconsistency as you say in #2313 . By adding the annotated_only argument to the function, people could create, if they want to, datasets/splits only with images that were labelled, for sure.
* Cleanup and update print()
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
hace 3 años
Glenn Jocher
08d4918d7f
labels.jpg class names ( #2454 )
* labels.png class names
* fontsize=10
hace 3 años
Glenn Jocher
f01f3223d5
Integer printout ( #2450 )
* Integer printout
* test.py 'Labels'
* Update train.py
hace 3 años
Glenn Jocher
e8a2b83268
GCP sudo docker userdata.sh ( #2393 )
* GCP sudo docker
* cleanup
hace 3 años
Glenn Jocher
7c2c95732c
AWS wait && echo "All tasks done." ( #2391 )
hace 3 años
Glenn Jocher
ba18528b47
bbox_iou() stability and speed improvements ( #2385 )
hace 3 años
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
hace 3 años
Ryan Avery
2c56ad5436
Confusion matrix background axis swap ( #2114 )
hace 3 años
Glenn Jocher
fab5085674
EMA bug fix 2 ( #2330 )
* EMA bug fix 2
* update
hace 3 años
Iden Craven
a82dce7faa
Fix labels being missed when image extension appears twice in filename ( #2300 )
hace 3 años
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
hace 3 años
Glenn Jocher
0070995bd5
Amazon AWS EC2 startup and re-startup scripts ( #2282 )
hace 3 años
Glenn Jocher
d5d275b6e9
Amazon AWS EC2 startup and re-startup scripts ( #2185 )
* Amazon AWS EC2 startup and re-startup scripts
* Create resume.py
* cleanup
hace 3 años
xiaowo1996
83dc1b4484
FLOPS min stride 32 ( #2276 )
Signed-off-by: xiaowo1996 <429740343@qq.com>
hace 3 años
Glenn Jocher
6f5d6fcdaa
Robust objectness loss balancing ( #2256 )
hace 3 años
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
hace 3 años
Glenn Jocher
47faf95079
reset head
hace 3 años
Glenn Jocher
d2e754b67b
Add isdocker() ( #2232 )
* Add isdocker()
* Update general.py
* Update general.py
hace 3 años
Glenn Jocher
26c2e54c8f
Add check_imshow() ( #2231 )
* Add check_imshow()
* Update general.py
* Update general.py
hace 3 años
Glenn Jocher
7b833e37bf
LoadStreams() frame loss bug fix ( #2222 )
hace 3 años
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
hace 3 años
NanoCode012
a5d5f9262d
Changed socket port and added timeout ( #2176 )
hace 3 años
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.
hace 3 años
Glenn Jocher
86897e3663
Update train.py test batch_size ( #2148 )
* Update train.py
* Update loss.py
hace 3 años
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>
hace 3 años
Glenn Jocher
4bdc5a397e
LoadImages() pathlib update ( #2140 )
hace 3 años
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>
hace 3 años
Glenn Jocher
9646ca438a
Update plot_study() ( #2112 )
hace 3 años
Glenn Jocher
be9edffded
Update test.py --task speed and study ( #2099 )
* Add --speed benchmark
* test range 256 - 1536
* update
* update
* update
* update
hace 3 años
Glenn Jocher
a18efc3a73
Add variable-stride inference support ( #2091 )
hace 3 años
Glenn Jocher
630ec06af6
Update to colors.TABLEAU_COLORS ( #2069 )
hace 3 años
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
hace 3 años
Glenn Jocher
f59f80114c
Add 'exclude' tuple to check_requirements() ( #2041 )
hace 3 años
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
hace 3 años
Glenn Jocher
08d3119e09
Add histogram equalization fcn ( #2049 )
hace 3 años
Glenn Jocher
8fcbe43423
Check im.format during dataset caching ( #2042 )
* Check im.format during dataset caching
* Update datasets.py
hace 3 años
Glenn Jocher
3b7feeafdf
Update general.py check_git_status() fix ( #2020 )
hace 3 años
Glenn Jocher
aac33f87ad
Update inference multiple-counting ( #2019 )
* Update inference multiple-counting
* update github check
hace 3 años
Glenn Jocher
77fd83225f
Update google_utils.py ( #2017 )
hace 3 años
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
hace 3 años
Glenn Jocher
d9212140b3
Add xywhn2xyxy() ( #1983 )
hace 3 años
Glenn Jocher
b1cf25dd9a
check_git_status() asserts ( #1977 )
hace 3 años
Abhiram V
17751b9891
Update autoanchor.py with yaml.SafeLoader ( #1971 )
hace 3 años
Abhiram V
35400dc7b7
Update plots.py with yaml.SafeLoader ( #1968 )
hace 3 años
Glenn Jocher
b26a2f6242
check_git_status() when not exist /workspace ( #1966 )
hace 3 años
Glenn Jocher
e8a41e8916
prevent check_git_status() in docker images ( #1951 )
* prevent check_git_status() running docker images
* Update general.py
hace 3 años
Glenn Jocher
ca9babb8e6
Add ComputeLoss() class ( #1950 )
hace 3 años
Glenn Jocher
b75c432ea0
GitHub API rate limit fallback ( #1930 )
hace 3 años
Glenn Jocher
051e9e80de
check_git_status() bug fix ( #1925 )
hace 3 años