INNER CODE UNIT · Python
driven_pose_feat
Tencent-Hunyuan/HunyuanPortrait · gradio_app.py:255
driven_pose_feat = motion_headpose_model(motion_pose_image * 2 + 1)
driven_pose_feat_embed = torch.cat([driven_pose_feat['rotation'], driven_pose_feat['translation'] * 0], dim=-1)
driven_feat = torch.cat([motion_feature_embed, driven_pose_feat_embed.unsqueeze(1).repeat(1, motion_feature_embed.shape[1], 1)], dim=-1)
driven_feat = driven_feat.unsqueeze(0)
uncond_driven_feat = torch.zeros_like(driven_feat)
pose_cond_tensor = pose_cond_tensor.unsqueeze(0)
pose_cond_tensor = rearrange(pose_cond_tensor, 'b f c h w -> b c f h w')
pose_cond_tensor_all.append(pose_cond_tensor)
driven_feat_all.append(driven_feat)
uncond_driven_feat_all.append(uncond_driven_feat)
driven_video_all.append(driven_video)
num_frames_all += num_frames
driven_video_all = torch.cat(driven_video_all, dim=0)
pose_cond_tensor_all = torch.cat(pose_cond_tensor_all, dim=2)