Parcourir la source

Fix `www.youtube.com` hostname (#7242)

* Fix `www.youtube.com` hostname

* Update datasets.py
modifyDataloader
Glenn Jocher GitHub il y a 2 ans
Parent
révision
37675e110f
Aucune clé connue n'a été trouvée dans la base pour cette signature ID de la clé GPG: 4AEE18F83AFDEB23
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. +1
    -1
      utils/datasets.py

+ 1
- 1
utils/datasets.py Voir le fichier

@@ -316,7 +316,7 @@ class LoadStreams:
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

Chargement…
Annuler
Enregistrer