open-mmlab / mmhuman3d

OpenMMLab 3D Human Parametric Model Toolbox and Benchmark
https://mmhuman3d.readthedocs.io/
Apache License 2.0
1.25k stars 137 forks source link

Smpl and smplx rendering on cpu #353

Open shaheryar1 opened 1 year ago

shaheryar1 commented 1 year ago

Unable to visualize or render smpl/smplx models using when passing --device 'cpu' in arguments.

I predicted the inference_result.npz file using hmr or Expose (for smplx) however in visualize_smpl() I got following error

Traceback (most recent call last): File "demo\estimate_smpl.py", line 594, in main(args) File "demo\estimate_smpl.py", line 492, in main single_person_with_mmdet(args, frames_iter) File "demo\estimate_smpl.py", line 279, in single_person_with_mmdet visualize_smpl_hmr( File "c:\users\shaherya\desktop\atg work\human 3d reconstruction\mmhuman3d\mmhuman3d\core\visualization\visualize_smpl.py", line 1120, in visualize_smpl_hmr return func(Ks=Ks, K=K, T=T, *kwargs) File "c:\users\shaherya\desktop\atg work\human 3d reconstruction\mmhuman3d\mmhuman3d\core\visualization\visualize_smpl.py", line 1004, in render_smpl renderer = SMPLRenderer( File "c:\users\shaherya\desktop\atg work\human 3d reconstruction\mmhuman3d\mmhuman3d\core\renderer\torch3d_renderer\smpl_renderer.py", line 61, in init self.image_renderer = build_renderer( File "c:\users\shaherya\desktop\atg work\human 3d reconstruction\mmhuman3d\mmhuman3d\core\renderer\torch3d_renderer\builder.py", line 48, in build_renderer return RENDERER.build(cfg) File "C:\Users\shaherya.conda\envs\open-mmlab\lib\site-packages\mmcv\utils\registry.py", line 237, in build return self.build_func(args, **kwargs, registry=self) File "C:\Users\shaherya.conda\envs\open-mmlab\lib\site-packages\mmcv\utils\registry.py", line 72, in build_from_cfg raise type(e)(f'{obj_cls.name}: {e}') AssertionError: MeshRenderer: Torch not compiled with CUDA enabled