iPERDance / iPERCore

Liquid Warping GAN with Attention: A Unified Framework for Human Image Synthesis
https://iperdance.github.io/work/impersonator-plus-plus.html
Apache License 2.0
2.43k stars 316 forks source link

RuntimeError: dimension mismatch for operand 0: equation 2 tensor 3 #156

Open pilotx-doctor opened 1 year ago

pilotx-doctor commented 1 year ago

When I run the programme:

Traceback (most recent call last): File "C:\Users\28627.conda\envs\i\lib\runpy.py", line 192, in _run_module_as_main return _run_code(code, main_globals, None, File "C:\Users\28627.conda\envs\i\lib\runpy.py", line 85, in _run_code exec(code, run_globals) File "D:\iPERCore-main\iPERCore-main\iPERCore\services\run_imitator.py", line 204, in run_imitator(opt=OPT) File "D:\iPERCore-main\iPERCore-main\iPERCore\services\run_imitator.py", line 193, in run_imitator all_meta_outputs = imitate(opt) File "D:\iPERCore-main\iPERCore-main\iPERCore\services\run_imitator.py", line 130, in imitate imitator.source_setup( File "C:\Users\28627.conda\envs\i\lib\site-packages\torch\autograd\grad_mode.py", line 15, in decorate_context return func(*args, *kwargs) File "D:\iPERCore-main\iPERCore-main\iPERCore\models\imitator.py", line 205, in source_setup src_info = self.body_rec.get_details(src_smpl, offsets, links_ids=links_ids) File "D:\iPERCore-main\iPERCore-main\iPERCore\tools\human_digitalizer\bodynets\base_smpl.py", line 129, in get_details verts, j3d, rs = self.forward(beta=shape, theta=pose, offsets=offsets, links_ids=links_ids, get_skin=True) File "D:\iPERCore-main\iPERCore-main\iPERCore\tools\human_digitalizer\bodynets\batch_smplh.py", line 172, in forward vertices, joints = lbs(beta, full_pose, self.v_template + offsets, File "D:\iPERCore-main\iPERCore-main\iPERCore\tools\human_digitalizer\smplx\lbs.py", line 181, in lbs v_shaped = v_template + blend_shapes(betas, shapedirs) File "D:\iPERCore-main\iPERCore-main\iPERCore\tools\human_digitalizer\smplx\lbs.py", line 270, in blend_shapes blend_shape = torch.einsum('bl,mkl->bmk', [betas, shape_disps]) File "C:\Users\28627.conda\envs\i\lib\site-packages\torch\functional.py", line 325, in einsum return einsum(equation, operands) File "C:\Users\28627.conda\envs\i\lib\site-packages\torch\functional.py", line 327, in einsum return _VF.einsum(equation, operands) RuntimeError: dimension mismatch for operand 0: equation 2 tensor 3

My python version ==3.8.0 CUDA version == 10.1 run on win10 pytorch ==1.6.0 GPU==GTX1650 Thanks for helping me