
    =pj                         S SK JrJrJrJrJr  S SKrSSKJ	r	  SSK
JrJr  SSKJr  SSKJr  S	S
KJr  \R&                  \	" S5       " S S\5      5       5       rg)    )AnyDictListOptionalUnionN   )pipeline_requires_extra   )	HPIConfigPaddlePredictorOption)DetVideoResult)	benchmark   )BasePipelinevideoc                     ^  \ rS rSrSrSrSSSSSSS.S\S\\   S	\\   S
\\\\	4      S\\
   S\S\\\\\	4   \4      SS4U 4S jjjr  SS\\\\   \R"                  \\R"                     4   S\S\S\4S jjrSrU =r$ )VideoDetectionPipeline   zVideo detection Pipelinevideo_detectionNFdeviceengineengine_config	pp_optionuse_hpip
hpi_configconfigr   r   r   r   r   r   returnc          
         > [         TU ]  " SUUUUUUS.UD6  US   S   n	0 n
SU	;   a  U	S   U
S'   SU	;   a  U	S   U
S'   U R                  " U	40 U
D6U l        g)a  Initializes the video detection pipeline.

Args:
    config (Dict): Configuration dictionary containing model and other parameters.
    device (Optional[str], optional): The device to use for prediction. Defaults to `None`.
    engine (Optional[str], optional): Inference engine. Defaults to `None`.
    engine_config (Optional[Dict[str, Any]], optional): Engine-specific config. Defaults to `None`.
    pp_option (Optional[PaddlePredictorOption], optional): Paddle predictor options.
        Defaults to `None`.
    use_hpip (bool, optional): Whether to use HPIP. Defaults to `False`.
    hpi_config (Optional[Union[Dict[str, Any], HPIConfig]], optional):
        HPIP configuration. Defaults to `None`.
r   
SubModulesVideoDetection
nms_threshscore_threshN )super__init__create_modelvideo_detection_model)selfr   r   r   r   r   r   r   kwargsvideo_detection_model_configmodel_kwargs	__class__s              p/var/www/html/pdf-tiff/venv/lib/python3.13/site-packages/paddlex/inference/pipelines/video_detection/pipeline.pyr&   VideoDetectionPipeline.__init__!   s    2 	 	
'!	
 	
 (.l';<L'M$77)El)SL&99+G+WL(%)%6%6(&
,8&
"    inputr"   r#   c              +   @   #    U R                  XUS9 Sh  vN   g N7f)a?  Predicts video detection results for the given input.

Args:
    input (Union[str, list[str], np.ndarray, list[np.ndarray]]): The input image(s) or path(s) to the images.
    **kwargs: Additional keyword arguments that can be passed to the function.

Returns:
    DetVideoResult: The predicted video detection results.
)r"   r#   Nr(   )r)   r1   r"   r#   r*   s        r.   predictVideoDetectionPipeline.predictN   s)     " --| . 
 	
 	
s   r3   )g      ?g?)__name__
__module____qualname____firstlineno____doc__entitiesr   r   strr   r   boolr   r   r&   r   npndarrayfloatr   r4   __static_attributes____classcell__)r-   s   @r.   r   r      s    # H !% $2659AE+
+
 	+

 +
  S#X/+
 12+
 +
 U4S>9#<=>+
 
+
 +
`  !	
S$s)RZZbjj1AAB
 
 	
 

 
r0   r   )typingr   r   r   r   r   numpyr>   
utils.depsr	   modelsr   r   models.video_detection.resultr   utils.benchmarkr   baser   time_methodsr   r$   r0   r.   <module>rK      sP    4 3  2 6 ; (  !E
\ E
 " E
r0   