HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

ultralytics 8.2.54 resolve YouTube bug with switch to pytubefix (#14354)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-07-11 09:32:30 +02:00 • committed by GitHub
parent 95736975fc
commit 5b330ad310
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 12 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

5
ultralytics/data/loaders.py
View file

@ -544,8 +544,9 @@ def get_best_youtube_url(url, method="pytube"):
(str): The URL of the best quality MP4 video stream, or None if no suitable stream is found.
"""
if method == "pytube":
check_requirements("pytube")
from pytube import YouTube
# Switched from pytube to pytubefix to resolve https://github.com/pytube/pytube/issues/1954
check_requirements("pytubefix")
from pytubefix import YouTube
streams = YouTube(url).streams.filter(file_extension="mp4", only_video=True)
streams = sorted(streams, key=lambda s: s.resolution, reverse=True) # sort streams by resolution