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

Pin pytubefix==6.3.4 for YouTube fix (#14571)

Browse source
This commit is contained in:
Glenn Jocher 2024-07-21 13:45:47 +02:00 • committed by GitHub
parent 47ff2b4a76
commit 68f1f326f5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/data/loaders.py
View file

@ -545,7 +545,7 @@ def get_best_youtube_url(url, method="pytube"):
""" """
if method == "pytube": if method == "pytube":
# Switched from pytube to pytubefix to resolve https://github.com/pytube/pytube/issues/1954 # Switched from pytube to pytubefix to resolve https://github.com/pytube/pytube/issues/1954
check_requirements("pytubefix") check_requirements("pytubefix==6.3.4") # bug in 6.4.2 https://github.com/JuanBindez/pytubefix/issues/123
from pytubefix import YouTube from pytubefix import YouTube
streams = YouTube(url).streams.filter(file_extension="mp4", only_video=True) streams = YouTube(url).streams.filter(file_extension="mp4", only_video=True)