fix wrong comment

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
     '''