CHUNYUWANG / imu-human-pose-pytorch

This is an official Pytorch implementation of "Fusing Wearable IMUs with Multi-View Images for Human Pose Estimation: A Geometric Approach, CVPR 2020".
MIT License
95 stars 17 forks source link

RuntimeError: expected scalar type Double but found Float #8

Closed Oliver-ny closed 2 years ago

Oliver-ny commented 2 years ago

Hi,I‘m sorry to disturb you.I meet a problem when running your program.The details are as follows: File "/root/Source/imu-human-pose/run/pose2d/../../lib/models/orn.py", line 86, in batch_uv_to_global_with_multi_depth coords_global = torch.bmm(inv_cam_extri_R, coords_img_frame_all_depth) + inv_cam_extri_T RuntimeError: expected scalar type Double but found Float I try to change the data type as inv_cam_extri_R.to(torch.float64).It doesn't work.So I want to know if there is a solution.