yolov5/utils
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>
2021-03-14 17:11:27 -07:00
..
aws GCP sudo docker userdata.sh (#2393) 2021-03-07 20:21:49 -08:00
google_app_engine adding the configuration to deploy yolov5 in in app engine (#964) 2020-09-22 11:48:44 -07:00
wandb_logging Start setup for improved W&B integration (#1948) 2021-02-01 21:38:41 -08:00
__init__.py initial commit 2020-05-29 17:04:54 -07:00
activations.py nn.SiLU() citation correction (#1713) 2021-01-06 20:58:41 -08:00
autoanchor.py Update autoanchor.py with yaml.SafeLoader (#1971) 2021-01-18 10:47:42 -08:00
datasets.py Update autosplit() with annotated_only option (#2466) 2021-03-14 17:11:27 -07:00
general.py bbox_iou() stability and speed improvements (#2385) 2021-03-06 13:07:34 -08:00
google_utils.py Update google_utils.py (#2017) 2021-01-22 15:08:42 -08:00
loss.py Robust objectness loss balancing (#2256) 2021-02-20 11:19:01 -08:00
metrics.py Confusion matrix background axis swap (#2114) 2021-03-02 16:09:52 -08:00
plots.py labels.jpg class names (#2454) 2021-03-12 22:15:41 -08:00
torch_utils.py Integer printout (#2450) 2021-03-12 13:35:44 -08:00