ultralytics 8.2.54 resolve YouTube bug with switch to pytubefix (#14354)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
95736975fc
commit
5b330ad310
4 changed files with 12 additions and 16 deletions
|
|
@ -1,6 +1,6 @@
|
|||
# Ultralytics YOLO 🚀, AGPL-3.0 license
|
||||
|
||||
__version__ = "8.2.53"
|
||||
__version__ = "8.2.54"
|
||||
|
||||
import os
|
||||
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -17,7 +17,7 @@ from ultralytics.utils import LOGGER, TQDM, checks, clean_url, emojis, is_online
|
|||
# Define Ultralytics GitHub assets maintained at https://github.com/ultralytics/assets
|
||||
GITHUB_ASSETS_REPO = "ultralytics/assets"
|
||||
GITHUB_ASSETS_NAMES = (
|
||||
[f"yolov8{k}{suffix}.pt" for k in "nsmlx" for suffix in ("", "-cls", "-seg", "-pose", "-obb")]
|
||||
[f"yolov8{k}{suffix}.pt" for k in "nsmlx" for suffix in ("", "-cls", "-seg", "-pose", "-obb", "-oiv7")]
|
||||
+ [f"yolov5{k}{resolution}u.pt" for k in "nsmlx" for resolution in ("", "6")]
|
||||
+ [f"yolov3{k}u.pt" for k in ("", "-spp", "-tiny")]
|
||||
+ [f"yolov8{k}-world.pt" for k in "smlx"]
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue