yolov5/utils
ketan-b 9d86b54eb3
Add multi-stream saving feature (#3864)
* Added the recording feature for multiple streams

Thanks for the very cool repo!!
I was trying to record multiple feeds at the same time, but the current version of the detector only had one video writer and one vid_path!
So the streams were not being saved and only were initialized with one frame and this process didn't record the whole thing.

Fix:
I made a list of `vid_writer` and `vid_path` and the `i` from the loop over the `pred` took care of the writer which need to work!

I hope this helps, Thanks!

* Cleanup list lengths

* batch size variable

* Update datasets.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-07-04 12:55:57 +02:00
..
aws Fix datasets for aws and get_coco.sh (#3788) 2021-06-26 15:29:16 +02:00
flask_rest_api Update JSON response (#3139) 2021-05-12 18:23:50 +02:00
google_app_engine Bump pip from 18.1 to 19.2 in /utils/google_app_engine (#3561) 2021-06-09 21:14:09 +02:00
wandb_logging W&B: Update Tables API and comply with new dataset_check (#3772) 2021-06-25 20:45:33 +02:00
__init__.py initial commit 2020-05-29 17:04:54 -07:00
activations.py ACON Activation batch-size 1 bug patch (#2901) 2021-04-25 22:54:07 +02:00
autoanchor.py Update check_requirements() exclude list (#2974) 2021-04-29 21:16:23 +02:00
datasets.py Add multi-stream saving feature (#3864) 2021-07-04 12:55:57 +02:00
general.py Move IoU functions to metrics.py (#3820) 2021-06-29 13:18:13 +02:00
google_utils.py Parse URL authentication (#3424) 2021-06-03 18:44:58 +02:00
loss.py Move IoU functions to metrics.py (#3820) 2021-06-29 13:18:13 +02:00
metrics.py Copy-Paste augmentation for YOLOv5 (#3845) 2021-07-01 00:35:04 +02:00
plots.py Update `feature_visualization()` (#3807) 2021-06-28 13:48:14 +02:00
torch_utils.py Update DDP for `torch.distributed.run` with `gloo` backend (#3680) 2021-06-19 16:30:25 +02:00