o *ÎjÝã@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é)Zaxisgp?é)rrrr)Ú ori_imagesÚxÚx_rgbÚfpsÚwidthÚheight)rÚappendÚnpÚfliprr r Ú enumerateÚcv2ZcvtColorZ COLOR_RGB2BGRZCOLOR_BGR2GRAYÚresizeZMODELZWIDTHZHEIGHTZreshapeZ expand_dimsZ transposeZ concatenateZastypeZfloat32ÚtorchZ from_numpyZ unsqueeze)ÚinputÚcfgZ video_readerZinputsÚframerÚwÚhZ rgb_imagesZimagesrÚiÚimageZ rgb_imagerr©rúp/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/preprocessors/cv/video_stabilization.pyÚstabilization_preprocessor s@ÿúr) rÚnumpyr rZtorch.nnÚnnZmodelscope.preprocessors.cvrrrrrrÚs