From 23c28ff2f7abcc57124a6e29c9d8ed21cf8afe53 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 1 Apr 2022 21:38:49 +0200 Subject: [PATCH] Fix `www.youtube.com` hostname (#7242) * Fix `www.youtube.com` hostname * Update datasets.py --- utils/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/datasets.py b/utils/datasets.py index 7e8b423c3174..b2d4fa54ae0d 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -316,7 +316,7 @@ def __init__(self, sources='streams.txt', img_size=640, stride=32, auto=True): for i, s in enumerate(sources): # index, source # Start thread to read frames from video stream st = f'{i + 1}/{n}: {s}... ' - if urlparse(s).hostname in ('youtube.com', 'youtu.be'): # if source is YouTube video + if urlparse(s).hostname in ('www.youtube.com', 'youtube.com', 'youtu.be'): # if source is YouTube video check_requirements(('pafy', 'youtube_dl==2020.12.2')) import pafy s = pafy.new(s).getbest(preftype="mp4").url # YouTube URL