From c1fcfe8cd9030939dd1635b158984fb066279b22 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 9 Nov 2022 17:20:26 +0100 Subject: [PATCH] YouTube Usage example comments (#10106) * YouTube Usage example comments Signed-off-by: Glenn Jocher * Update dataloaders.py Signed-off-by: Glenn Jocher Signed-off-by: Glenn Jocher --- utils/dataloaders.py | 1 + 1 file changed, 1 insertion(+) diff --git a/utils/dataloaders.py b/utils/dataloaders.py index b33a24a46f9c..621c03cd2db1 100644 --- a/utils/dataloaders.py +++ b/utils/dataloaders.py @@ -353,6 +353,7 @@ def __init__(self, sources='streams.txt', img_size=640, stride=32, auto=True, tr # Start thread to read frames from video stream st = f'{i + 1}/{n}: {s}... ' if urlparse(s).hostname in ('www.youtube.com', 'youtube.com', 'youtu.be'): # if source is YouTube video + # YouTube format i.e. 'https://www.youtube.com/watch?v=Zgi9g1ksQHc' or 'https://youtu.be/Zgi9g1ksQHc' check_requirements(('pafy', 'youtube_dl==2020.12.2')) import pafy s = pafy.new(s).getbest(preftype="mp4").url # YouTube URL