
    9i                         S SK JrJr  S SKJr  S SKJ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  \" 5       r\R&                  " \R(                  \R(                  S	9 " S
 S\5      5       rg)    )AnyDict)	Pipelines)
inpainting)
OutputKeys)InputPipeline)	PIPELINES)Tasks)
get_logger)module_namec                      ^  \ rS rSrS\4U 4S jjrS\S\\\4   4S jr	S\\\4   S\\\4   4S jr
S\\\4   S\\\4   4S	 jrS
rU =r$ )VideoInpaintingPipeline   modelc                 T   > [         TU ]  " SSU0UD6  [        R                  S5        g)zm
use `model` to create video inpainting pipeline for prediction
Args:
    model: model id on modelscope hub.
r   zload model doneN )super__init__loggerinfo)selfr   kwargs	__class__s      q/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/pipelines/cv/video_inpainting_pipeline.pyr    VideoInpaintingPipeline.__init__   s'     	/u//%&    inputreturnc                     U$ Nr   )r   r   s     r   
preprocess"VideoInpaintingPipeline.preprocess   s    r   c           	          [         R                  " US   5      u  p#pEUb  [        R                  S0$ [         R                  " U R
                  US   US   US   UXE5        [        R                  S0$ )Nvideo_input_pathdecode_error	mask_pathvideo_output_pathDone)r   video_processr   OUTPUTinpainting_by_model_balancer   )r   r   r&   fpswhs         r   forwardVideoInpaintingPipeline.forward    s    ",":":$%#'1 #%%~66..tzz/45G/H/4[/A/45H/I3/0		5 !!6**r   inputsc                     U$ r!   r   )r   r2   s     r   postprocess#VideoInpaintingPipeline.postprocess/   s    r   r   )__name__
__module____qualname____firstlineno__strr   r   r   r   r"   r0   r4   __static_attributes____classcell__)r   s   @r   r   r      su    'c ' $sCx. +T#s(^ +S#X +$sCx. T#s(^  r   r   N)typingr   r   modelscope.metainfor   %modelscope.models.cv.video_inpaintingr   modelscope.outputsr   modelscope.pipelines.baser   r	   modelscope.pipelines.builderr
   modelscope.utils.constantr   modelscope.utils.loggerr   r   register_modulevideo_inpaintingr   r   r   r   <module>rG      s]     ) < ) 5 2 + .	 		(B(BDh Dr   