search

DenisTome / Lifting-from-the-Deep-release

Implementation of "Lifting from the Deep: Convolutional 3D Pose Estimation from a Single Image"
https://denistome.github.io/papers/lifting-from-the-deep
GNU General Public License v3.0
450 stars 133 forks source link

ValueError: total size of new array must be unchanged #28

Closed suishii closed 6 years ago

suishii commented 6 years ago

Hello, I try to use the demo.py to get results but for some pictures, it give me the error as show as below. How can I fix it? Expect your answer.

Traceback (most recent call last): File "demo.py", line 89, in sys.exit(main()) File "demo.py", line 54, in main pose_2d, visibility, pose_3d = pose_estimator.estimate(image) File "/home/suiishii/Lifting-from-the-Deep-release/packages/lifting/_pose_estimator.py", line 144, in estimate pose_3d = self.poseLifting.compute_3d(transformed_pose2d, weights) File "/home/suiishii/Lifting-from-the-Deep-release/packages/lifting/utils/prob_model.py", line 256, in compute_3d normpose, = Prob3dPose.normalise_data(reg_joints, weights) File "/home/suiishii/Lifting-from-the-Deep-release/packages/lifting/utils/prob_model.py", line 107, in normalise_data d2 = d2.reshape(d2.shape[0], -1, 2).transpose(0, 2, 1) ValueError: total size of new array must be unchanged

DenisTome commented 6 years ago

The reason of that error is that all the people identified in your images have an uncertainty lower than a certain threshold. In order to solve the problem, you can change such threshold in the file packages/lifting/utils/config.py by setting VISIBLE_PART = 1e-3 to a lower value.

suishii commented 5 years ago

I see! Thank you very much.