Yolov5 detection produces no output for video streams (list of urls)

ocean-data-factory-sweden / kso

Notebooks to upload/download marine footage, connect to a citizen science project, train machine learning models and publish marine biological observations.

GNU General Public License v3.0

5 stars 12 forks source link

Yolov5 detection produces no output for video streams (list of urls) #207

Closed jannesgg closed 1 year ago

jannesgg commented 1 year ago

List of videos (urls from S3) treated as streams (e.g. webcam) and do not result in output video.

Line: https://github.com/ultralytics/yolov5/blob/7398d2d77cbac9f66259926d49c26bfa3c257a9b/detect.py#LL82C5-L82C58

Proposed quick-fix for local runs:

jannesgg commented 1 year ago

[x] Customise detect.py to allow output when desired

jannesgg commented 1 year ago

@pilarnavarro This quick fix has been implemented in 2779a0abc5ff745a6ea073f9fffaa362b0626f49. See Tutorial 6 for the updated argument. By default, the output should now be created when using detect_yolov5.