{-j ddlmZmZmZmZmZddlZddlm Z ddl m Z m Z ddl mZddlmZd d lmZeje d Gd d eZdS))AnyDictListOptionalUnionN)pipeline_requires_extra) HPIConfigPaddlePredictorOption)TopkVideoResult) benchmark) BasePipelinevideoc,eZdZdZdZddddddddedeedeed eeeefd ee d e d ee eeefe fd dffdZ dde eeeejeejfde edfd efdZxZS)VideoClassificationPipelinezVideo Classification Pipelinevideo_classificationNFdeviceengine engine_config pp_optionuse_hpip hpi_configconfigrrrrrrreturnc tjd||||||d||dd} || |_dS)aInitializes the video classification 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 SubModulesVideoClassificationN)super__init__ create_modelvideo_classification_model) selfrrrrrrrkwargs!video_classification_model_config __class__s y/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddlex/inference/pipelines/video_classification/pipeline.pyr#z$VideoClassificationPipeline.__init__!sz2  '!      -3<,@AV,W)*.*;*; -+ + '''inputtopkc+DK|||Ed{VdS)aPredicts 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.r's r*predictz#VideoClassificationPipeline.predictIs9"225t2DDDDDDDDDDDr+)r,)__name__ __module__ __qualname____doc__entitiesrrstrrr boolrr r#rnpndarrayintr r0 __classcell__)r)s@r*rrsS('%H !% $2659AE& & & &  &  &  S#X/ & 12& & U4S>9#<=>&  & & & & & & V"#EES$s)RZbj1AABECIE  EEEEEEEEr+r)typingrrrrrnumpyr8 utils.depsr modelsr r "models.video_classification.resultr utils.benchmarkrbaser time_methodsrr!r+r*rDs4333333333333322222266666666AAAAAA(((((( !!>E>E>E>E>E,>E>E"!>E>E>Er+