kornia-examples
kornia-examples copied to clipboard
Value error
I tried depth_estimation.ipynb
.
# warp by depth
i_trans_j = i_pose_w @ torch.inverse(j_pose_w)
k_trans_j = k_pose_w @ torch.inverse(j_pose_w)
dst_trans_src = torch.cat([i_trans_j, k_trans_j], dim=0)
depth_src = depth_j
intrinsics_src = intrinsics_j
image_src = image_j
image_dst = torch.cat([image_i, image_k], dim=0)
image_dst_to_src = kornia.warp_frame_depth(image_dst, depth_src, dst_trans_src, intrinsics_src)
print(image_dst_to_src.shape)
It throws the following error,
ValueError: Input batch size must be the same for both tensors or 1