o *Îj ã@s,ddlZdd„Zdd„Zdd„Zdd „ZdS) éNcCsF|jddks Jdƒ‚t |¡}|ddd…f |ddd…f<||S)Néÿÿÿÿéz"q must be a tensor of shape (*, 4).é)ÚshapeÚtorchZ ones_like)ÚqÚmask©r úp/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/utils/cv/motion_utils/motion_process.pyÚqinvs r cCsÌ|jddks J‚|jddksJ‚|jdd…|jdd…ks"J‚t|jƒ}| ¡ dd¡}| ¡ dd¡}|dd…dd…f}tj||dd}tj||dd}|d|dd…dd…f|| |¡S)zð Rotate vector(s) v about the rotation described by quaternion(s) q. Expects a tensor of shape (*, 4) for q and a tensor of shape (*, 3) for v, where * denotes any number of dimensions. Returns a tensor of shape (*, 3). rréNr©Údimé)rÚlistÚ contiguousÚviewrÚcross)rÚvZoriginal_shapeZqvecZuvZuuvr r r Úqrots  *rcCsü|d}t |¡ |j¡} |ddd…f|ddd…f<tj|dd}t |jdd…d¡ |j¡}t |¡|d<t |¡|d<t |jdd…d¡ |j¡}|ddd…dd …f|ddd…d d gf< t t |ƒ|ƒ}tj|d d}|d |d<||fS)N©.r.rrr ©r©.r)r r rréþÿÿÿ).r ).r) rZ zeros_likeÚtoZdeviceZcumsumZzerosrÚcosÚsinrr )ÚdataZrot_velZ r_rot_angÚ r_rot_quatÚr_posr r r Úrecover_root_rot_pos$s  ( r cCsÊt|ƒ\}}|dd|ddd…f}| |jdd…d¡} tt|dddd…fƒ |jdd…d¡|ƒ} |d|dd d…f7<|d |dd d…f7< tj| d ¡|gd d }|S)N.rrr r)rr rrrrrrr ) r rrrr ÚexpandrÚcatZ unsqueeze)rZ joints_numrrZ positionsr r r Úrecover_from_ric:s (þr#)rr rr r#r r r r Ús