При запуске модели POSA: ошибка выполнения в smplx/body_models.py, функция reset_params
Я просматривал репозиторий POSA, чтобы найти точки соприкосновения тела и сцены SMPL. Ссылка на репозиторий POSA GitHub
Traceback (most recent call last):
File "src/affordance.py", line 105, in <module>
vertices_org, vertices_can, faces_arr, body_model, R_can, pelvis, torch_param, vertices_clothed = data_utils.pkl_to_canonical(
File "<path>/POSA/src/data_utils.py", line 83, in pkl_to_canonical
body_model.reset_params(**torch_param)
File "<path>/lib/python3.8/site-packages/torch/autograd/grad_mode.py", line 28, in decorate_context
return func(*args, **kwargs)
File “<path>/lib/python3.8/site-packages/smplx-0.1.28-py3.8.egg/smplx/body_models.py", line 289, in reset_params
RuntimeError: The expanded size of the tensor (16) must match the existing size (10) at non-singleton dimension 1. Target sizes: [1, 16]. Tensor sizes: [10]
Вышеупомянутая ошибка времени выполнения возникает при запуске примеров команд, таких как:
python src/affordance.py --config cfg_files/contact_semantics.yaml --checkpoint_path $POSA_dir/trained_models/contact_semantics.pt --pkl_file_path $POSA_dir/POSA_rp_poses/rp_aaron_posed_001_0_0.pkl --scene_name MPH16 --render 1 --viz 1 --use_cuda False
python src/gen_rand_samples.py --config cfg_files/contact.yaml --checkpoint_path $POSA_dir/trained_models/contact.pt --pkl_file_path $POSA_dir/POSA_rp_poses/rp_aaron_posed_001_0_0.pkl --render 1 --viz 1 --num_rand_samples 3
Я попытался напечатать аргументы, которые передавались вreset_paramsфункция которая есть. Посмотрев на формы тензоров вtorch_paramЯ попытался усечь тензоры до 10, чтобы устранить эту ошибку, но она начала выдавать новую ошибку, и после просмотра кода я понял, что такое усечение - неправильная вещь.