o ॵiÝã@s<ddlZddlZddlZddlmZddlmZdd„ZdS)éN)Ú VideoReaderc CsZt|ƒ}g}|D] }| tj|dd¡q|j}|j}|j}g}g} g} t|ƒD]W\} }t  |tj ¡}t  |tj ¡} | d} t  | |j j|j jf¡} |  |  dd|j j|j j¡¡t  ||j j|j jf¡} | t t | d¡d¡¡|  t t |d¡d¡¡q(t | d¡ tj¡}t |¡ d¡}t |d¡ tj¡}t |¡ d¡}| |||||dœS)Né)Úaxisgp?é)rrrr)Ú 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_imagerr©r-úc/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/preprocessors/cv/video_stabilization.pyÚstabilization_preprocessor s@ÿúr/) rÚnumpyr rÚtorch.nnÚnnÚmodelscope.preprocessors.cvrr/r-r-r-r.Ús