From 9982279cbcc8e220de6ab015e3e93e27198105cc Mon Sep 17 00:00:00 2001 From: Kai Wang <21052222@hdu.edu.cn> Date: Thu, 2 Nov 2023 19:30:45 +0800 Subject: [PATCH] fix arg passed error in func get_correspondence --- src/models/pano/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/models/pano/utils.py b/src/models/pano/utils.py index 7b9729c..a86d5a6 100644 --- a/src/models/pano/utils.py +++ b/src/models/pano/utils.py @@ -27,7 +27,7 @@ def get_correspondences(R, K, img_h, img_w): R_left@torch.inverse(K_left)) - xyz_l = torch.tensor(get_x_2d(img_h, img_w), + xyz_l = torch.tensor(get_x_2d(img_w, img_h), device=R.device) xyz_l = ( xyz_l.reshape(-1, 3).T)[None].repeat(homo_l.shape[0], 1, 1)