@ -98,7 +98,7 @@ class LoadStreams:
def close ( self ) :
""" Close stream loader and release resources. """
self . running = False # stop flag for Thread
for i , thread in enumerate ( self . threads ) :
for thread in self . threads :
if thread . is_alive ( ) :
thread . join ( timeout = 5 ) # Add timeout
for cap in self . caps : # Iterate through the stored VideoCapture objects
@ -210,7 +210,6 @@ class LoadImages:
self . vid_stride = vid_stride # video frame-rate stride
self . bs = 1
if any ( videos ) :
self . orientation = None # rotation degrees
self . _new_video ( videos [ 0 ] ) # new video
else :
self . cap = None
@ -263,20 +262,6 @@ class LoadImages:
self . frame = 0
self . cap = cv2 . VideoCapture ( path )
self . frames = int ( self . cap . get ( cv2 . CAP_PROP_FRAME_COUNT ) / self . vid_stride )
if hasattr ( cv2 , ' CAP_PROP_ORIENTATION_META ' ) : # cv2<4.6.0 compatibility
self . orientation = int ( self . cap . get ( cv2 . CAP_PROP_ORIENTATION_META ) ) # rotation degrees
# Disable auto-orientation due to known issues in https://github.com/ultralytics/yolov5/issues/8493
# self.cap.set(cv2.CAP_PROP_ORIENTATION_AUTO, 0)
def _cv2_rotate ( self , im ) :
""" Rotate a cv2 video manually. """
if self . orientation == 0 :
return cv2 . rotate ( im , cv2 . ROTATE_90_CLOCKWISE )
elif self . orientation == 180 :
return cv2 . rotate ( im , cv2 . ROTATE_90_COUNTERCLOCKWISE )
elif self . orientation == 90 :
return cv2 . rotate ( im , cv2 . ROTATE_180 )
return im
def __len__ ( self ) :
""" Returns the number of files in the object. """
@ -385,10 +370,10 @@ def autocast_list(source):
return files
LOADERS = [ LoadStreams , LoadPilAndNumpy , LoadImages , LoadScreenshots ]
LOADERS = LoadStreams , LoadPilAndNumpy , LoadImages , LoadScreenshots # tuple
def get_best_youtube_url ( url , use_pafy = Tru e) :
def get_best_youtube_url ( url , use_pafy = Fals e) :
"""
Retrieves the URL of the best quality MP4 video stream from a given YouTube video .
@ -411,9 +396,11 @@ def get_best_youtube_url(url, use_pafy=True):
import yt_dlp
with yt_dlp . YoutubeDL ( { ' quiet ' : True } ) as ydl :
info_dict = ydl . extract_info ( url , download = False ) # extract info
for f in info_dict . get ( ' formats ' , None ) :
if f [ ' vcodec ' ] != ' none ' and f [ ' acodec ' ] == ' none ' and f [ ' ext ' ] == ' mp4 ' and f . get ( ' width ' ) > 1280 :
return f . get ( ' url ' , None )
for f in reversed ( info_dict . get ( ' formats ' , [ ] ) ) : # reversed because best is usually last
# Find a format with video codec, no audio, *.mp4 extension at least 1920x1080 size
good_size = ( f . get ( ' width ' ) or 0 ) > = 1920 or ( f . get ( ' height ' ) or 0 ) > = 1080
if good_size and f [ ' vcodec ' ] != ' none ' and f [ ' acodec ' ] == ' none ' and f [ ' ext ' ] == ' mp4 ' :
return f . get ( ' url ' )
if __name__ == ' __main__ ' :
Glenn Jocher
1 year ago
committed by
GitHub