o *ju1@s4ddlZddlZddlZddlmZddlZddlmm Z d1ddZ ddZ d2d d Z d d ZddZddZdejdejdejfddZdejdejdejfddZd3dejdedejfddZd3dejdedejfddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.ejdejfd/d0ZdS)4NFcCsFt|d }|}Wdn1swYg}g}g}g}g}g} d} |D]} | dddkrHdd| dddD} || | ddd kr| ddd} d d| D}t| t|} ||d | dvrt| dd ddkrd d| D}||d | dvrt| dd d krt| dd ddkrdd| D}| || dd dkr| ddd} dd| D}||| dd dkr| ddd} dd| D}||q)t| tj }| d krt| tj }nt d|j dd kr ||d}n|dddd f|ddd df|d}t|dkrJt| tj }||d<t|dkr^t| tj }||d<t|dkrw| d krst| tj }||d<t| dkr| d krt|  tj } | |d<|rt dt|t dt||S)Nrrzv cS g|] }t|dkrt|qSrlenfloat.0ar m/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/models/cv/human3d_animation/utils.py szread_obj.. f cSs*g|]}t|dkrt|ddqS)r/rintsplitr r r r r *rcS*g|]}t|dkrt|ddqS)rrrrr r r r r%rcSr)rrrrr r r r r*s$zvt cSrrrr r r r r1 zvn cSrrrr r r r r6rznot a triangle face mesh!)verticesfaces)rcolorsruvsnormalsfaces_uv faces_normalznum of verticesz num of faces)open readlinesstriprappendmaxrnparrayastypefloat32Zint32printshape)Zobj_pathZ print_shapefZ bfm_linesrrrZvnsrr Zmax_face_lengthlineZvertexitemsfaceface_uv face_normaluvvnmeshr r r read_obj s    & $      r5c stj|}tjtj|d}d|vrnttj||d|dttj||dd5}| d| d| d| d | d | d | d | d |dWdn1siwYt|d }d|vr| d| d |d|vrt |dD])\}}| d |d|d|d|d|d|d|d|d|dqn|dD]}| d |d|d|dqd|vr|dD]}| d |d|dqd|vr|dD]}| d |d|d|dqd|vrmt |dD]c\} d|vsd|vrRd|vr+|d| nd|vr9|d| nddfdd t t Dd!} nddfd"d t t Dd!} | | qWddSWddS1swYdS)#NrZ texture_mapz.pngz.mtlwznewmtl material_0 zKa 1.000000 0.000000 0.000000 zKd 1.000000 1.000000 1.000000 zKs 0.000000 0.000000 0.000000 z Tr 0.000000 zillum 0 z Ns 0.000000 z map_Kd {} z# Create by ModelScope zmtllib ./{}.mtl rrzv {} {} {} {} {} {} rrz v {} {} {} rz vt {} {} rz vn {} {} {} rrr rrcs&g|]}d|||qS)z{}/{}/{}formatr ir/r1r0r r rszwrite_obj.. csg|] }d|qS)z{}r7r9)r/r r rs)ospathdirnamesplitextbasenamecv2Zimwritejoinr!writer8 enumerateranger) Z save_pathr4save_dirZ save_nameZwfr:vr2r3indrowr r;r write_objcsz            "   "      $rK皙??I@c Cs^t||dddgd||ddgdd|| ||d|| ||ggdgtjS)Nrr)rrrr&r'r(r))xnr,r r r projections *rScCs6tddd|gddd|gddd|ggdgtjS)NrrrrrrrP)rQyzr r r translates "rWcCsJt|t|}}tgdd||dgd| |dggdgtjS)N)rrrrrrTr&sincosr'r(r)r scr r r rotate_x  r^cCsJt|t|}}t|d|dggd| d|dggdgtjS)Nr)rrrrrTrXr[r r r rotate_yr_r`rQrUreturncCstj||dddS)NrOTkeepdim)torchsum)rQrUr r r dotsrfrRcCsdt||||S)Nr)rf)rQrRr r r reflectsrg#B ;epscCsttjt|||dS)N)min)rdsqrtclamprfrQrir r r lengthsrncCs|t||S)N)rnrmr r r safe_normalizesrocCsXt|tjr t|n|}tj|t|jddggdd}t || dS)Nrr)ZaxisN.) isinstancer&ZndarrayrdZ from_numpycudacatZonesr+matmult)mtxposZt_mtxZposwr r r transform_poss&rxc Cst||} tj|| |||gd\} } |r/tj|d| || dd\} }tj|d| |d| d}nt|d| |\} }tj|d| dd}|tj}||ddd fddf}||ddd fddf}||ddd fddf}tt ||||}tj d |j d tj d d dddf d d}t|d| |\}}|d d}t| dddfd d }||d |t|}||d |t|}|||fS)N) resolutionrpall)Zrast_dbZ diff_attrszlinear-mipmap-linear) filter_mode max_mip_levelZlinear)r{rrrrr)ZdtypeZdevicer?.rO)rxdrZ rasterizeZ interpolateZtexturetyperdlongrocrossZaranger+Zint64repeatrrlZ ones_like)ZglctxrvrwZpos_idxr2Zuv_idxtexryZ enable_mipr|Zpos_clipZrast_outZ rast_out_dbZtexcZtexdcolor_Zv0Zv1Zv2Z face_normalsZface_normal_indicesZgb_geometric_normalnormalmaskr r r rendersT         rcCs |dk|dkk}t|| |S)a Return a tensor where each element has the absolute value taken from the, corresponding element of a, with sign taken from the corresponding element of b. This is like the standard copysign floating-point operation, but is not careful about negative 0 and NaN. Args: a: source tensor. b: tensor whose signs will be used, of the same shape as a. Returns: Tensor of the same shape as a with the signs of b. r)rdwhere)r bZ signs_differr r r _copysignsrcCs(t|}|dk}t||||<|S)z[ Returns torch.sqrt(torch.max(0, x)) but with a zero subgradient where x is 0. r)rdZ zeros_likerk)rQretZ positive_maskr r r _sqrt_positive_parts rc Cs|ddks|ddkrtd|jd|d}|d}|d}d td |||}d td |||}d td |||}d td |||}t||d |d }t||d |d} t||d|d} t||| | fdS)z Convert rotations given as rotation matrices to quaternions. Args: matrix: Rotation matrices as tensor of shape (..., 3, 3). Returns: quaternions with real part first, as tensor of shape (..., 4). rOrz Invalid rotation matrix shape f.).rr).rr).rrr}r).rr).rr).rr).rr).rr).rr)size ValueErrorr+rrrdstack) matrixZm00Zm11Zm22Zo0rQrUrVZo1o2Zo3r r r matrix_to_quaternions rcCstj|dddfdddd}t||dddf}d|}d}||k}t|}t||||||<d ||||d ||<|dddf|S) a Convert rotations given as quaternions to axis/angle. Args: quaternions: quaternions with real part first, as tensor of shape (..., 4). Returns: Rotations given as a vector in axis angle form, as a tensor of shape (..., 3), where the magnitude is the angle turned anticlockwise in radians around the vector's direction. .rNrrOT)pdimrcgư>r}0)rdZnormatan2absZ empty_likerY)Z quaternionsZnormsZ half_anglesZanglesriZ small_anglesZsin_half_angles_over_anglesr r r quaternion_to_axis_angle5s  rcCs tt|S)a{ Convert rotations given as rotation matrices to axis/angle. Args: matrix: Rotation matrices as tensor of shape (..., 3, 3). Returns: Rotations given as a vector in axis angle form, as a tensor of shape (..., 3), where the magnitude is the angle turned anticlockwise in radians around the vector's direction. )rr)rr r r matrix_to_axis_angleRs rd6cCs||dddf|dddf}}tj|dd}|||jddd|}tj|dd}tj||dd}tj|||fddS) a Converts 6D rotation representation by Zhou et al. [1] to rotation matrix using Gram--Schmidt orthogonalisation per Section B of [1]. Args: d6: 6D rotation representation, of size (*, 6) Returns: batch of rotation matrices of size (*, 3, 3) [1] Zhou, Y., Barnes, C., Lu, J., Yang, J., & Li, H. On the Continuity of Rotation Representations in Neural Networks. IEEE Conference on Computer Vision and Pattern Recognition, 2019. Retrieved from http://arxiv.org/abs/1812.07035 .NrrO)rTrbr)F normalizererdrr)rZa1Za2b1b2Zb3r r r rotation_6d_to_matrixbs "r)F)rLrMrN)rh)r=rBnumpyr&Znvdiffrast.torchrdr~Ztorch.nn.functionalnnZ functionalrr5rKrSrWr^r`ZTensorrfrgrrnrorxrrrrrrrr r r r s0  W >.