o *Îj ã@s^ddlZddlZddlmmZddlmZddlm Z ddl m Z m Z dd„Z d d „ZdS) éN)Útqdmé)Úcfgé)ÚHomoCalcÚHomoProjc Csþ|j}t t tjdtjjtjj|dtjdtjj tjj|d¡d¡  ddd¡  d¡  ¡}|t  ||gd¡}g}t|jdƒD]O} t|d|| ƒ} t t tjdtjj|dtjdtjj |d¡d¡  ddd¡  ¡} t| |  ¡ dd¡  dd¡ƒ  dd¡} | |  ¡j| jŽ  ddd¡¡q@t |d¡}|dd…dd…dd…dftjjdd|dd…dd…dd…df<|dd…dd…dd…dftjj dd|dd…dd…dd…df<|  dd dd¡}tj|||fd d d }|  ddd d¡}tj||d d } | S)z¯ @param frame current frame [N, 1, H, W] @param x_motion [N, 1, G_H, G_W] @param y_motion [N, 1, G_H, G_W] @return mesh warping according to given motion r)ÚdevicerréÿÿÿÿNg@gð?éZbilinearT)ÚmodeÚ align_corners)r )rÚtorchÚstackZmeshgridZarangerZMODELZWIDTHZPIXELSZHEIGHTZpermuteZ unsqueezeÚfloatÚcatÚrangeÚshaperrÚ contiguousÚviewÚappendÚFZ interpolateZ grid_sample)ÚframeÚx_motionÚy_motionÚ cap_widthÚ cap_heightZ target_deviceZ src_gridsZ des_gridsZ projectionÚiZhomoZ origin_kpZ projected_kpZgenerated_frame©rúy/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/models/cv/video_stabilization/utils/WarpUtils.pyÚmesh_warp_frame sp ÿÿýúúþüüþþ ÿ ÿ DÿÿÿýrcCszt |d¡}t t |d¡d¡}t t |d¡d¡}t | tj¡¡}t | tj¡¡}t | tj¡¡}t|||||ƒS)zt @param images List(image [1, 1, H, W]) @param x_motion [G_H, G_W, N] @param y_motion [G_H, G_W, N] r)rrrr) ÚnpZ concatenateZ expand_dimsZ transposer Z from_numpyZastypeZfloat32r)ZimagesrrrrÚframesrrrÚ warpListImageEs r")Únumpyr r Ztorch.nn.functionalÚnnZ functionalrrZ DUT.configrZProjectionUtilsrrrr"rrrrÚs   8