enghock1 / Real-Time-2D-and-3D-Hand-Pose-Estimation

Real-Time 2D and 3D Hand Pose Estimation from RGB Image
GNU General Public License v2.0
48 stars 15 forks source link

runtime error when testing with webcam #11

Open zy1996829 opened 1 year ago

zy1996829 commented 1 year ago

Traceback (most recent call last): File "/home/PycharmProjects/Real-Time-2D-and-3D-Hand-Pose-Estimation-master/7.real_time_3D_handposeestimation.py", line 71, in , est_pose_uv, est_pose_cam_xyz = model(frame_device, K, pose_scale) File "/home/anaconda3/envs/gesture_recognition/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl return forward_call(*input, kwargs) File "/home/PycharmProjects/Real-Time-2D-and-3D-Hand-Pose-Estimation-master/hand_shape_pose/model/pose_mlp_network.py", line 87, in forward est_hm_list, encoding = self.net_hm(images) File "/home/anaconda3/envs/gesture_recognition/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl return forward_call(*input, *kwargs) File "/home/PycharmProjects/Real-Time-2D-and-3D-Hand-Pose-Estimation-master/hand_shape_pose/model/net_hg.py", line 112, in forward x = self.r1(x) File "/home/anaconda3/envs/gesture_recognition/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl return forward_call(input, kwargs) File "/home/PycharmProjects/Real-Time-2D-and-3D-Hand-Pose-Estimation-master/hand_shape_pose/util/net_util.py", line 60, in forward out = self.bn(x) File "/home/anaconda3/envs/gesture_recognition/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl return forward_call(*input, **kwargs) File "/home/anaconda3/envs/gesture_recognition/lib/python3.9/site-packages/torch/nn/modules/batchnorm.py", line 168, in forward return F.batch_norm( File "/home/anaconda3/envs/gesture_recognition/lib/python3.9/site-packages/torch/nn/functional.py", line 2282, in batch_norm return torch.batch_norm(

RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 7.92 GiB total capacity; 6.26 GiB already allocated; 44.38 MiB free; 6.26 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

Process finished with exit code 1