
    9i`                         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  SSK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   )TopkVideoResult)	benchmark)	HPIConfig)PaddlePredictorOption   )BasePipelinevideoc                      ^  \ rS rSrSrSr    SS\S\S\S\	S	\
\\\\4   \4      S
S4U 4S jjjr SS\\\\   \R"                  \\R"                     4   S\\S4   S
\4S jjrSrU =r$ )VideoClassificationPipeline   zVideo Classification Pipelinevideo_classificationNconfigdevice	pp_optionuse_hpip
hpi_configreturnc                 ^   > [         TU ]  X#XES9  US   S   nU R                  U5      U l        g)ah  
Initializes the class with given configurations and options.

Args:
    config (Dict): Configuration dictionary containing model and other parameters.
    device (str): The device to run the prediction on. Default is None.
    pp_option (PaddlePredictorOption): Options for PaddlePaddle predictor. Default is None.
    use_hpip (bool, optional): Whether to use the high-performance
        inference plugin (HPIP) by default. Defaults to False.
    hpi_config (Optional[Union[Dict[str, Any], HPIConfig]], optional):
        The default high-performance inference configuration dictionary.
        Defaults to None.
)r   r   r   r   
SubModulesVideoClassificationN)super__init__create_modelvideo_classification_model)selfr   r   r   r   r   !video_classification_model_config	__class__s          y/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/paddlex/inference/pipelines/video_classification/pipeline.pyr    $VideoClassificationPipeline.__init__"   sE    * 	 	 	
 -3<,@AV,W)*.*;*;-+
'    inputtopkc              +   >   #    U R                  XS9 Sh  vN   g N7f)a  Predicts video classification 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.
    topk: Union[int, None]: The number of top predictions to return. Defaults to 1.
    **kwargs: Additional keyword arguments that can be passed to the function.

Returns:
    TopkVideoResult: The predicted top k results.
)r*   Nr"   )r#   r)   r*   kwargss       r&   predict#VideoClassificationPipeline.predict@   s     " 2252DDDs   r,   )NNFN)   )__name__
__module____qualname____firstlineno____doc__entitiesr   strr   boolr   r   r   r   r    r   npndarrayintr   r.   __static_attributes____classcell__)r%   s   @r&   r   r      s     (%H
 +/AE

 
 )	

 
 U4S>9#<=>
 

 
B "#ES$s)RZZbjj1AABE CIE
 
E Er(   r   )typingr   r   r   r   r   numpyr9   
utils.depsr	   "models.video_classification.resultr   utils.benchmarkr   	utils.hpir   utils.pp_optionr   baser   time_methodsr    r(   r&   <module>rH      sS    4 3  2 A ( " 4  !4E, 4E " 4Er(   