^jBddlmZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddl mZddlmZmZmZddlmZddlZddlZddlmZddlmZdd lmZeeZeGd d Z Gd d Z!ddZ" d ddZ# d ddZ$ddddddd ddZ%GddZ&y)) annotationsN)deque)Callable Generator) dataclass)EmptyFullQueue)Any)tqdm) _get_loggercbeZdZUdZded<ded<ded<dZded <ed d Zed d Z y) VideoInfoa A class to store video information, including width, height, fps and total number of frames. Attributes: width: width of the video in pixels height: height of the video in pixels fps: frames per second of the video as a float. Common values include 23.976, 24.0, 25.0, 29.97, 30.0, 59.94, and 60.0. total_frames: total number of frames in the video, default is None Examples: ```python import supervision as sv video_info = sv.VideoInfo.from_video_path(video_path="") video_info # VideoInfo(width=3840, height=2160, fps=25.0, total_frames=538) video_info.resolution_wh # (3840, 2160) ``` intwidthheightfloatfpsN int | None total_framesctj|}|jstd|t |j tj }t |j tj}t|j tj}t |j tj}|jt||||S)NCould not open video at ) cv2 VideoCaptureisOpened ExceptionrgetCAP_PROP_FRAME_WIDTHCAP_PROP_FRAME_HEIGHTr CAP_PROP_FPSCAP_PROP_FRAME_COUNTreleaser)cls video_pathvideorrrrs b/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/supervision/utils/video.pyfrom_video_pathzVideoInfo.from_video_path:s  ,~~6zlCD DEIIc6678UYYs889:EIIc../0599S%=%=>?  \::c2|j|jfSN)rrselfs r& resolution_whzVideoInfo.resolution_whGszz4;;&&r()r$strreturnr)r/ztuple[int, int]) __name__ __module__ __qualname____doc____annotations__r classmethodr'propertyr-r(r&rrsH4 J K J#L*# ; ;''r(rcBeZdZdZdddZd dZd dZ d dZy) VideoSinka Context manager that saves video frames to a file using OpenCV. Attributes: target_path: The path to the output file where the video will be saved. video_info: Information about the video resolution, fps, and total frame count. codec: FOURCC code for video format Example: ```python import supervision as sv video_info = sv.VideoInfo.from_video_path("") frames_generator = sv.get_video_frames_generator("") with sv.VideoSink(target_path="", video_info=video_info) as sink: for frame in frames_generator: sink.write_frame(frame=frame) ``` c<||_||_||_d|_yr*) target_path video_info_VideoSink__codec_VideoSink__writer)r,r;r<codecs r&__init__zVideoSink.__init__cs&$  r(c tj|j|_tj|j|j|jj|jj|_ |S#t$r@}t j dt|tjd|_Yd}~d}~wwxYw)Nz%s. Defaulting to mp4v...mp4v)rVideoWriter_fourccr=_VideoSink__fourcc TypeErrorloggerwarningr. VideoWriterr;r<rr-r>)r,es r& __enter__zVideoSink.__enter__is <22DLLADM    MM OO   OO ) )     < NN6A ?22F;DMM > C6CCcT|j|jj|yy)z Writes a single video frame to the target video file. Args: frame: The video frame to be written to the file. The frame must be in BGR color format. N)r>write)r,frames r& write_framezVideoSink.write_framews% == $ MM   & %r(cR|j|jjyyr*)r>r")r,exc_type exc_value exc_tracebacks r&__exit__zVideoSink.__exit__s# == $ MM ! ! # %r(N)rB)r;r.r<rr?r.)r/r9)rMznpt.NDArray[np.uint8]r/None)rPztype[BaseException] | NonerQzBaseException | NonerRr r/rT)r0r1r2r3r@rJrNrSr7r(r&r9r9LsA,  '$,$($ $  $r(r9ctjd}|tjdyd} t j t jj|dt jjt jj|\}}t j|tj|dddd d |d |d d d d ddddd|gtjtjd}|j dk7r|j"j%dj'}tjd|j |rd|nd |6t jj)|rt j*|yyyt j,|||6t jj)|rt j*|yyy#t.$r }tjd|Yd}~]d}~wwxYw#|6t jj)|rt j*|wwwxYw)zMux audio from `source_path` into `video_path` in-place using ffmpeg. Args: source_path: Path to the original video file containing the audio stream. video_path: Path to the video-only file to be updated with audio. ffmpegNzcffmpeg not found on PATH. Audio will not be preserved. Install ffmpeg to enable audio preservation.)suffixdirz-yz -loglevelerrorz-nostatsz-iz-c:vcopyz-c:az-mapz0:v:0z1:a:0?z -shortesti,)stdoutstderrtimeoutrreplace)errorszTffmpeg failed to mux audio (return code %d)%s. The output video will not have audio.z: z:Audio muxing failed: %s. Output video will not have audio.)shutilwhichrFrGtempfilemkstempospathsplitextdirnameabspathclose subprocessrunDEVNULLPIPE returncoder]decodestripexistsremover_r) source_pathr$ ffmpeg_pathtmp_pathtmp_fdresult stderr_msgexcs r& _mux_audior|s,,x(K ;  H0 #++77##J/2 ;<  ' *%%??1 4    !--Y-?EEGJ NN8!!%/"ZL!R     BGGNN8$< IIh %=  8Z(  BGGNN8$< IIh %=    H#    BGGNN8$< IIh %= s0D2G,G,, H5H HHH:IFctj|}|jstd|t |j tj }|||kDr tdt|d}| t||n|}|r$|dkDrD|j}|sn1|dz}|dkDrn%|dkDr |jtj||||fS)NrzRequested frames are outboundrrW) rrrrrrr!maxmingrabsetCAP_PROP_POS_FRAMES)rustartenditerative_seekr%rsuccesss r&_validate_and_setup_videors   [ )E >> 2;-@AAuyy!9!9:;L 3-788 qME$'O#c< CaijjlG QJE ai  #))51 % r(c#Kt||||\}}}|} |j\}}|r||k\rn3||t|dz D]} |j}|rn||z }N|j yw)a Get a generator that yields the frames of the video. Args: source_path: The path of the video file. stride: Indicates the interval at which frames are returned, skipping stride - 1 frames between each. start: Indicates the starting position from which video should generate frames end: Indicates the ending position at which video should stop generating frames. If None, video will be read to the end. iterative_seek: If True, the generator will seek to the `start` frame by grabbing each frame, which is much slower. This is a workaround for videos that don't open at all when you set the `start` value. Returns: A generator that yields the frames of the video. Examples: ```python import supervision as sv for frame in sv.get_video_frames_generator(source_path=""): ... ``` NrW)rreadrangerr") rustriderrrr%frame_positionrrM_s r&get_video_frames_generatorrsD2UCE5#N .C/   Kvz" AjjlG  &   MMOs AA7A7 zProcessing video) max_framesprefetch writer_buffer show_progressprogress_messagepreserve_audioctj} t| jxsdn| jxsd} t |t |dfd } dfd } t j | d} t|| 5}t j | |fd }| j|jt| | | }d}d } j}|d}n2|\}} |||}j||jd G jdd |s jd }|n | j#d|j#d|j%|| ddd|r4j!rt&j)dyt+|yy#t$r }|}Yd}~d}~wwxYw#t$rYwxYw#t$r| j!sYYwxYw# jdd n#t$rYnwxYw|s: jd }|n#n!#t$r| j!sYnY5wxYw9| j#d|j#d|j%||wxYw#1swY&xYw)uh Process video frames asynchronously using a threaded pipeline. This function orchestrates a three-stage pipeline to optimize video processing throughput: 1. Reader thread: Continuously reads frames from the source video file and enqueues them into a bounded queue (`frame_read_queue`). The queue size is limited by the `prefetch` parameter to control memory usage. 2. Main thread (Processor): Dequeues frames from `frame_read_queue`, applies the user-defined `callback` function to process each frame, then enqueues the processed frames into another bounded queue (`frame_write_queue`) for writing. The processing happens in the main thread, simplifying use of stateful objects without synchronization. 3. Writer thread: Dequeues processed frames from `frame_write_queue` and writes them sequentially to the output video file. Args: source_path: Path to the input video file. target_path: Path where the processed video will be saved. callback: Function called for each frame, accepting the frame as a numpy array and its zero-based index, returning the processed frame. max_frames: Optional maximum number of frames to process. If None, the entire video is processed (default). prefetch: Maximum number of frames buffered by the reader thread. Controls memory use; default is 32. writer_buffer: Maximum number of frames buffered before writing. Controls output buffer size; default is 32. show_progress: Whether to display a tqdm progress bar during processing. Default is False. progress_message: Description shown in the progress bar. preserve_audio: If True, copy the audio stream from `source_path` into `target_path` after frame processing. Requires `ffmpeg` on PATH (e.g. `apt install ffmpeg`, `brew install ffmpeg`). If ffmpeg is not found or the mux step fails, a warning is logged and the output video is saved without audio — no exception is raised. Audio is truncated to match the processed video duration. Default is False. Returns: None Example: ```python import supervision as sv from rfdetr import RFDETRMedium model = RFDETRMedium() def callback(frame, frame_index): return model.predict(frame) sv.process_video( source_path="source.mp4", target_path="target.mp4", callback=callback, preserve_audio=True, ) ``` )r$Nr)maxsizect}t|D]\}}j||fjdy)N)rur)r enumerateput)frame_generator frame_indexrMframe_read_queuerrus r& reader_threadz$process_video..reader_threadqsO4# #,O"< 7 K  +u!5 6 7T"r(cR j}|y|j|&)N)rM)rrN) video_sinkrMframe_write_queues r& writer_threadz$process_video..writer_threadzs2%))+E}  " " " / r(T)targetdaemon)r;r<)rargsr)totaldisabledescFrW)r^ zdWriter thread did not finish in time; skipping audio mux to avoid reading an incomplete output file.)rur$r/rT)rr9r/rT)rr'rrr threadingThreadr9rr rrupdaterr ris_alivejoinrkrFrGr|)rur;callbackrrrrrrr<rrr reader_workerr writer_worker progress_barexception_in_worker read_finished read_itemrrMprocessed_framer{rrs` ` @@r& process_videors?P**k*BJ  ! J # # (q*5  $ $ ) INI>C>#0$$M$GM {z B=*j!((  %%! 15 * *,002 $$(M%." U&.uk&BO%))/: ''* !%%dA%6 ! !$4$8$8$8$C $,!-    r  *   r  *    ".))/y=*~  ! ! # NN>  ;; GA!*-'    !!,557! !# !%%dA%6    ! !$4$8$8$8$C $,!- !!,557! !   r  *   r  *    ".))/y=*=*sA J3H;+G&H(G&;J3?G5;J3 G#GHG##H& G2/J31G22J35HJ3HJ3J0H,+J0, H8 5J07H8 8J0?IJ0I3 .J02I3 3=J00J33J=c<eZdZdZdddZed dZd dZd dZy) FPSMonitorzN A class for monitoring frames per second (FPS) to benchmark latency. c&t||_y)a Args: sample_size: The maximum number of observations for latency benchmarking. Examples: ```python import supervision as sv frames_generator = sv.get_video_frames_generator( source_path="") fps_monitor = sv.FPSMonitor() for frame in frames_generator: # your processing code here fps_monitor.tick() fps = fps_monitor.fps ``` )maxlenN)rall_timestamps)r, sample_sizes r&r@zFPSMonitor.__init__s(-2,Er(c|jsy|jd|jdz }|dk7rt|j|z SdS)z Computes and returns the average FPS based on the stored time stamps. Returns: The average FPS. Returns 0.0 if no time stamps are stored. gr)rlen)r, taken_times r&rzFPSMonitor.fpssQ""((,t/B/B1/EE :D/D''(J6RsRr(c^|jjtjy)zI Adds a new time stamp to the deque for FPS calculation. N)rappendtime monotonicr+s r&tickzFPSMonitor.ticks ""4>>#34r(c8|jjy)z< Clears all the time stamps from the deque. N)rclearr+s r&resetzFPSMonitor.resets !!#r(N))rr)r/rr) r0r1r2r3r@r6rrrr7r(r&rrs-F, S S5 $r(r)rur.r$r.r/rT)F) rur.rrrrrboolr/z!tuple[cv2.VideoCapture, int, int])rWrNF) rur.rrrrrrrrr/z,Generator[npt.NDArray[np.uint8], None, None])rur.r;r.rz=Callable[[npt.NDArray[np.uint8], int], npt.NDArray[np.uint8]]rrrrrrrrrr.rrr/rT)' __future__rrfrbrlrdrr collectionsrcollections.abcrr dataclassesrqueuerr r typingr rnumpynp numpy.typingnpt tqdm.autor supervision.utils.loggerr r0rFrr9r|rrrrr7r(r&rs"  /!$$ 0 X  /'/' /'d=$=$@@ HKP '1CG&4 11 1 1  1  1 2 1r". mHmHmHLmH  mH  mHmHmHmHmH mH`2$2$r(