search

facebookresearch / pifuhd

High-Resolution 3D Human Digitization from A Single Image.
Other
9.5k stars 1.44k forks source link

Excustion Issue #104

Closed sonyxo2 closed 3 years ago

sonyxo2 commented 3 years ago

net = PoseEstimationWithMobileNet() checkpoint = torch.load('checkpoint_iter_370000.pth', map_location='cpu') load_state(net, checkpoint)

get_rect(net.cuda(), [image_path], 512)

TypeError Traceback (most recent call last)

in () 3 load_state(net, checkpoint) 4 ----> 5 get_rect(net.cuda(), [image_path], 512) in get_rect(net, images, height_size) 28 total_keypoints_num += extract_keypoints(heatmaps[:, :, kpt_idx], all_keypoints_by_type, total_keypoints_num) 29 ---> 30 pose_entries, all_keypoints = group_keypoints(all_keypoints_by_type, pafs, demo=True) 31 for kpt_id in range(all_keypoints.shape[0]): 32 all_keypoints[kpt_id, 0] = (all_keypoints[kpt_id, 0] * stride / upsample_ratio - pad[1]) / scale TypeError: group_keypoints() got an unexpected keyword argument 'demo'
vj-abishek commented 3 years ago

https://github.com/facebookresearch/pifuhd/issues/103#issue-772654610