a Q^+gd @sddlZddlZejdejejejeddlmZddl m Z m Z GdddZ e dkrddlZddZeZe ejd Zeejejd dS) N)render)seed_everythingtiming_decoratorc@s,eZdZdZdddZeddd d ZdS) GifRendererz5 render frame(s) of mesh using pytorch3d cuda:0cCs ||_dS)Ndevice)selfrr ]/apdcephfs_cq5/share_300600172/huiwenshi/3d_services/repos/hunyuan3d-lite/infer/gif_render.py__init__!szGifRenderer.__init__z gif renderrxTc Cs t||||||||j|d dS)N)elevazim resolution gif_dst_pathn_viewsfpsrrgb)rr) r obj_filenamerrrrrrrr r r __call__$s zGifRenderer.__call__N)r)rrr rrrT)__name__ __module__ __qualname____doc__r rrr r r r rs r__main__cCs@t}|jdtdd|jdtdd|jddtd|S)Nz --mesh_pathT)typerequiredz--output_gif_pathz--devicer)defaultr)argparseArgumentParser add_argumentstr parse_args)parserr r r get_args?s r(r)r)ossyspathinsertdirnameabspath__file__Zsvrm.ldm.vis_utilr infer.utilsrrrrr"r(argsrZ gif_rendererZ mesh_pathoutput_gif_pathr r r r s(