
    9i                     :    S SK r S SKrS SKrS SKJr  S SKJr  S rg)    N)VideoReaderc           	         [        U 5      n/ nU H'  nUR                  [        R                  " USS95        M)     UR                  nUR
                  nUR                  n/ n/ n	/ n
[        U5       GH  u  p[        R                  " U[        R                  5      n[        R                  " U[        R                  5      nUS-  n[        R                  " XR                  R                  UR                  R                  45      nU	R                  UR!                  SSUR                  R                  UR                  R                  5      5        [        R                  " XAR                  R                  UR                  R                  45      nUR                  [        R"                  " [        R$                  " US5      S5      5        U
R                  [        R"                  " [        R$                  " US5      S5      5        GM     [        R&                  " U	S5      R)                  [        R*                  5      n[,        R.                  " U5      R1                  S5      n[        R&                  " US5      R)                  [        R*                  5      n[,        R.                  " U5      R1                  S5      nU
UUUUUS.$ )N   )axisgp?   )r   r   r   r   )
ori_imagesxx_rgbfpswidthheight)r   appendnpflipr   r   r   	enumeratecv2cvtColorCOLOR_RGB2BGRCOLOR_BGR2GRAYresizeMODELWIDTHHEIGHTreshapeexpand_dims	transposeconcatenateastypefloat32torch
from_numpy	unsqueeze)inputcfgvideo_readerinputsframer   wh
rgb_imagesimagesr   iimage	rgb_imager	   r
   s                   o/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/preprocessors/cv/video_stabilization.pystabilization_preprocessorr0      s    u%LFbgge!,- 


CAAJFJf%UC$5$56UC$6$67#

599??CII4D4D"EFemmAq#))*:*:CIIOOLMJJuyy		8H8H&IJ	NN2<<	9=qA	C"..eY)GKL & 	vq!((4A%%a(ANN:q)00<EU#--a0E !     )	r   numpyr   r    torch.nnnnmodelscope.preprocessors.cvr   r0    r1   r/   <module>r7      s        3!r1   