a yf^@s.ddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl Z ddl ZddlZddlZddlmZddlmZmZmZddlmZmZmZmZdd lmZeGd d d ZGd d d ZGdddZ GdddZ!GdddZ"GdddZ#ddZ$dddZ%ee"e!e fZ&dS)N) dataclass)Path)Thread)urlparse)Image)FORMATS_HELP_MSG IMG_FORMATS VID_FORMATS)IS_COLAB IS_KAGGLELOGGERops)check_requirementsc@sBeZdZUdZdZeed<dZeed<dZeed<dZ eed<dS) SourceTypeszBClass to represent various types of input sources for predictions.Fstream screenshotfrom_imgtensorN) __name__ __module__ __qualname____doc__rbool__annotations__rrrrrT/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/data/loaders.pyrs    rc@sBeZdZdZdddZddZd d Zd d Zd dZddZ dS) LoadStreamsaQ Stream Loader for various types of video streams, Supports RTSP, RTMP, HTTP, and TCP streams. Attributes: sources (str): The source input paths or URLs for the video streams. vid_stride (int): Video frame-rate stride, defaults to 1. buffer (bool): Whether to buffer input streams, defaults to False. running (bool): Flag to indicate if the streaming thread is running. mode (str): Set to 'stream' indicating real-time capture. imgs (list): List of image frames for each stream. fps (list): List of FPS for each stream. frames (list): List of total frames for each stream. threads (list): List of threads for each stream. shape (list): List of shapes for each stream. caps (list): List of cv2.VideoCapture objects for each stream. bs (int): Batch size for processing. Methods: __init__: Initialize the stream loader. update: Read stream frames in daemon thread. close: Close stream loader and release resources. __iter__: Returns an iterator object for the class. __next__: Returns source paths, transformed, and original images for processing. __len__: Return the length of the sources object. Example: ```bash yolo predict source='rtsp://example.com/media.mp4' ``` file.streamsFc Csdtjj_||_d|_d|_||_tj |r>t | n|g}t|}||_dg||_dg||_dg||_dg||_ddt|D|_ddt|D|_dd|D|_t|D]\}}|d d |d |d }t|jd vrt|}|rt|n|}|dkr0ts(tr0t dt!"||j|<|j|#sbt$|d|t%|j|&t!j'}t%|j|&t!j(} |j|&t!j)} t*t%|j|&t!j+dpt,d|j|<t*t-.| r| ndddpd|j|<|j|/\} } | r| dur*t$|d||j|0| | j|j|<t1|j2||j||gdd|j|<t34|d|j|d|d| d|j|dd |j|5qt34ddS)zKInitialize instance variables and check for consistent input stream shapes.TrrNcSsg|]}gqSrr.0_rrr Pz(LoadStreams.__init__..cSsg|]}gqSrrrrrrr"Qr#cSsg|]}t|qSr)r Z clean_strr xrrrr"Rr#r/: z... >zwww.youtube.comz youtube.comzyoutu.bezm'source=0' webcam not supported in Colab and Kaggle notebooks. Try running 'source=0' in a local environment.zFailed to open infdzFailed to read images from )targetargsdaemonu Success ✅ (z frames of shape r%z at z.2fz FPS))6torchbackendsZcudnnZ benchmarkbufferrunningmode vid_strideospathisfiler read_textrsplitlenbsfpsframesthreadscapsrangeimgsshapesources enumeraterhostnameget_best_youtube_url isnumericevalr r NotImplementedErrorcv2 VideoCaptureisOpenedConnectionErrorintgetZCAP_PROP_FRAME_WIDTHZCAP_PROP_FRAME_HEIGHT CAP_PROP_FPSmaxCAP_PROP_FRAME_COUNTfloatmathisfinitereadappendrupdater infostart) selfrCr4r1nisstwhr<successimrrr__init__AsV "      * "6zLoadStreams.__init__cCsd|j|}}|jr|r||dkrt|j|dkr|d7}|||jdkr|\}}|stj |j |tj d}t d|||jr|j||q|g|j|<qtdqdS)z(Read stream `i` frames in daemon thread.rrr*dtypeuQWARNING ⚠️ Video stream unresponsive, please check your IP camera connection.g{Gz?N)r=r2rLr:rAgrabr4retrievenpzerosrBuint8r warningopenr1rWtimesleep)r[r]caprr\frbrcrrrrXts   zLoadStreams.updatec Csd|_|jD]}|r |jddq |jD]F}z |Wq,typ}ztd|WYd}~q,d}~00q,t dS)z*Close stream loader and release resources.F)timeoutu6WARNING ⚠️ Could not release VideoCapture object: N) r2r>is_alivejoinr?release Exceptionr rlrJZdestroyAllWindows)r[threadrperrrcloses   (zLoadStreams.closecCs d|_|S)zCIterates through YOLO image feed and re-opens unresponsive streams.countr[rrr__iter__szLoadStreams.__iter__cCs|jd7_g}t|jD]\}}|s|j|rHtdtdkrT|t t dt |j |j|}|s$td|q$|jr||dq||r|dntj|j|tjd|q|j|dg|jfS)zEReturns source paths, transformed and original images for processing.rqu"WARNING ⚠️ Waiting for stream rr{rer.)r}rDrAr>rtrJZwaitKeyordrz StopIterationrnrominr<r rlr1rWpoprirjrBrkclearrCr;)r[imagesr]r%rrr__next__s   * zLoadStreams.__next__cCs|jS)z(Return the length of the sources object.r;r~rrr__len__szLoadStreams.__len__N)rrF) rrrrrdrXrzrrrrrrrr!s 3 rc@s(eZdZdZddZddZddZdS) LoadScreenshotsa YOLOv8 screenshot dataloader. This class manages the loading of screenshot images for processing with YOLOv8. Suitable for use with `yolo predict source=screen`. Attributes: source (str): The source input indicating which screen to capture. screen (int): The screen number to capture. left (int): The left coordinate for screen capture area. top (int): The top coordinate for screen capture area. width (int): The width of the screen capture area. height (int): The height of the screen capture area. mode (str): Set to 'stream' indicating real-time capture. frame (int): Counter for captured frames. sct (mss.mss): Screen capture object from `mss` library. bs (int): Batch size, set to 1. monitor (dict): Monitor configuration details. Methods: __iter__: Returns an iterator object. __next__: Captures the next screenshot and returns it. c Cs:tdddl}|^}}d\|_}}}}t|dkrHt|d|_nJt|dkrldd|D\}}}}n&t|d krd d|D\|_}}}}d |_d|_||_d|_ d |_ |jj |j}|dur|d n |d ||_ |dur|dn |d||_ |p |d|_|p|d|_|j |j |j|jd|_dS)z8Source = [screen_number left top width height] (pixels).mssrN)rNNNNrcss|]}t|VqdSNrNr$rrr r#z+LoadScreenshots.__init__..rrcss|]}t|VqdSrrr$rrrrr#rr*topleftwidthheight)rrrr)rrsplitscreenr:rNr3framesctr;r<Zmonitorsrrrrmonitor) r[sourcerparamsrrrrrrrrrds*     zLoadScreenshots.__init__cCs|S)z"Returns an iterator of the object.rr~rrrrszLoadScreenshots.__iter__c Cs|t|j|jddddddf}d|jd|jd|jd|jd|j d }|j d7_ t |jg|g|gfS)zHScreen capture with 'mss' to get raw pixels from the screen as np array.Nzscreen z (LTWH): ,r'r) riasarrayrrgrrrrrrrstr)r[im0r^rrrrs*.zLoadScreenshots.__next__N)rrrrrdrrrrrrrsrc@s:eZdZdZdddZddZddZd d Zd d Zd S)LoadImagesAndVideosa YOLOv8 image/video dataloader. This class manages the loading and pre-processing of image and video data for YOLOv8. It supports loading from various formats, including single image files, video files, and lists of image and video paths. Attributes: files (list): List of image and video file paths. nf (int): Total number of files (images and videos). video_flag (list): Flags indicating whether a file is a video (True) or an image (False). mode (str): Current mode, 'image' or 'video'. vid_stride (int): Stride for video frame-rate, defaults to 1. bs (int): Batch size, set to 1 for this class. cap (cv2.VideoCapture): Video capture object for OpenCV. frame (int): Frame counter for video. frames (int): Total number of frames in the video. count (int): Counter for iteration, initialized at 0 during `__iter__()`. Methods: _new_video(path): Create a new cv2.VideoCapture object for a given video path. rc Csd}t|tr6t|jdkr6t|j}t|}g}t|ttfrPt |n|gD]}tt| }d|vr| t t j |ddqVt j|r| t t t j|dqVt j|r||qV|r||r|t|| qVt|dqVgg}} |D]B} | dd } | tvrB|| n| tvr| | qt|t| } } || |_| | |_| |_d g| dg| |_d |_||_||_ t!| r|"| d nd|_#|jd krtd |dt$dS)zHInitialize the Dataloader and raise FileNotFoundError if file not found.Nz.txt*T) recursivez*.*z does not exist.r{FimagerzNo images or videos found in z. )% isinstancerrsuffixparentr8 splitlineslisttuplesortedabsoluteextendglobr5r6isdirrur7rWis_fileFileNotFoundErrorrlowerrr r:filesnfni video_flagr3r4r;any _new_videorpr)r[r6batchr4rrparZvideosrqrrnvrrrrdsH              zLoadImagesAndVideos.__init__cCs d|_|S)z:Returns an iterator object for VideoStream or ImageFolder.rr|r~rrrr>szLoadImagesAndVideos.__iter__c Csggg}}}t||jkr|j|jkr>|r:|||fSt|j|j}|j|jrvd|_|jrn|j sx| |t |j D]}|j }|sqq|r4|j\}}|rt|jd7_|||||d|jdd|jd|jd|jd|d |j|jkrt|jd7_|jn@|jd7_|jrT|j|j|jkr| |j|jqd|_t|}|d urtd |n:|||||d |jdd|jd |d|jd7_|j|jkrqq|||fS) zUReturns the next batch of images or video frames along with their paths and metadata.Zvideorzvideo r&z (frame z) r'rNu WARNING ⚠️ Image Read Error zimage  )r:r;r}rrrrr3rprLrr@r4rgrhrrWr=rvrJZimreadr rlr)r[pathsrArYr6r!rbrrrrrCsR       6      & zLoadImagesAndVideos.__next__cCs\d|_t||_t|jtj|_|js>t d|t|jtj |j |_ dS)z6Creates a new video capture object for the given path.rzFailed to open video N) rrJrKrprNrOrPr<rLrrRr4r=)r[r6rrrrxs   zLoadImagesAndVideos._new_videocCst|j|jS)z,Returns the number of batches in the object.)rTceilrr;r~rrrrszLoadImagesAndVideos.__len__N)rr) rrrrrdrrrrrrrrrs  ,5 rc@s<eZdZdZddZeddZddZdd Zd d Z d S) LoadPilAndNumpya Load images from PIL and Numpy arrays for batch processing. This class is designed to manage loading and pre-processing of image data from both PIL and Numpy formats. It performs basic validation and format conversion to ensure that the images are in the required format for downstream processing. Attributes: paths (list): List of image paths or autogenerated filenames. im0 (list): List of images stored as Numpy arrays. mode (str): Type of data being processed, defaults to 'image'. bs (int): Batch size, equivalent to the length of `im0`. Methods: _single_check(im): Validate and format a single image to a Numpy array. csNt|ts|g}ddt|D_fdd|D_d_tj_dS)z$Initialize PIL and Numpy Dataloader.cSs$g|]\}}t|dd|dqSfilenamerz.jpggetattrr r]rcrrrr"r#z,LoadPilAndNumpy.__init__..csg|]}|qSr) _single_check)r rcr~rrr"r#rN)rrrDrrr3r:r;r[rrr~rrds  zLoadPilAndNumpy.__init__cCstt|tjtjfs$Jdt|t|tjrp|jdkrD|d}t|dddddddf}t|}|S)z,Validate and format an image to numpy array.z,Expected PIL/np.ndarray image type, but got RGBNr{) rrrindarraytyper3convertrZascontiguousarray)rcrrrrs$   " zLoadPilAndNumpy._single_checkcCs t|jS)z*Returns the length of the 'im0' attribute.)r:rr~rrrrszLoadPilAndNumpy.__len__cCs2|jdkrt|jd7_|j|jdg|jfS)z8Returns batch paths, images, processed images, None, ''.rr.r}rrrr;r~rrrrs zLoadPilAndNumpy.__next__cCs d|_|S)z,Enables iteration for class LoadPilAndNumpy.rr|r~rrrrszLoadPilAndNumpy.__iter__N) rrrrrd staticmethodrrrrrrrrrs  rc@sDeZdZdZddddZedddZd d Zd d Zd dZ dS) LoadTensoraR Load images from torch.Tensor data. This class manages the loading and pre-processing of image data from PyTorch tensors for further processing. Attributes: im0 (torch.Tensor): The input tensor containing the image(s). bs (int): Batch size, inferred from the shape of `im0`. mode (str): Current mode, set to 'image'. paths (list): List of image paths or filenames. count (int): Counter for iteration, initialized at 0 during `__iter__()`. Methods: _single_check(im, stride): Validate and possibly modify the input tensor. N)returncCs8|||_|jjd|_d|_ddt|D|_dS)zInitialize Tensor Dataloader.rrcSs$g|]\}}t|dd|dqSrrrrrrr"r#z'LoadTensor.__init__..N)rrrBr;r3rDrrrrrrds zLoadTensor.__init__ cCsd|dt|jd}t|jdkrPt|jdkr=6.5.2r)YouTubeZmp4T)Zfile_extensionZ only_videocSs|jSr) resolution)r^rrr(r#z&get_best_youtube_url..)keyreverseNr{i8pafy)rzyoutube_dl==2020.12.2)Zpreftypezyt-dlpquietF)downloadformatsrirZvcodecnoneZacodecexturl)rZ pytubefixrstreamsfilterrrrNrrnewZ getbestvideoyt_dlpZ YoutubeDLZ extract_inforeversedrO) rmethodrrrrrZydlZ info_dictrqZ good_sizerrrrFs*  ,*0rF)r)'rrTr5rn dataclassesrpathlibr threadingr urllib.parserrJnumpyrirr/ZPILrZultralytics.data.utilsrrr Zultralytics.utilsr r r r Zultralytics.utils.checksrrrrrrrrrFZLOADERSrrrrs6       A 7@ 3