From 30b428ba74bd7743a17c19fa20f6bfd39b1de057 Mon Sep 17 00:00:00 2001 From: Shunsuke Date: Mon, 22 Jun 2020 09:24:34 +0900 Subject: [PATCH] fix wrong comment --- lib/geometry.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lib/geometry.py b/lib/geometry.py index 821e1bdc..5e88b386 100755 --- a/lib/geometry.py +++ b/lib/geometry.py @@ -5,7 +5,7 @@ def index(feat, uv): ''' :param feat: [B, C, H, W] image features - :param uv: [B, 2, N] uv coordinates in the image plane, range [0, 1] + :param uv: [B, 2, N] uv coordinates in the image plane, range [-1, 1] :return: [B, C, N] image features at the uv coordinates ''' uv = uv.transpose(1, 2) # [B, N, 2] @@ -20,7 +20,7 @@ def orthogonal(points, calibrations, transforms=None): ''' Compute the orthogonal projections of 3D points into the image plane by given projection matrix :param points: [B, 3, N] Tensor of 3D points - :param calibrations: [B, 3, 4] Tensor of projection matrix + :param calibrations: [B, 4, 4] Tensor of projection matrix :param transforms: [B, 2, 3] Tensor of image transform matrix :return: xyz: [B, 3, N] Tensor of xyz coordinates in the image plane ''' @@ -38,7 +38,7 @@ def perspective(points, calibrations, transforms=None): ''' Compute the perspective projections of 3D points into the image plane by given projection matrix :param points: [Bx3xN] Tensor of 3D points - :param calibrations: [Bx3x4] Tensor of projection matrix + :param calibrations: [Bx4x4] Tensor of projection matrix :param transforms: [Bx2x3] Tensor of image transform matrix :return: xy: [Bx2xN] Tensor of xy coordinates in the image plane '''