Skip to content

Commit

Permalink
fix flake8 issue
Browse files Browse the repository at this point in the history
Signed-off-by: Yiheng Wang <vennw@nvidia.com>
  • Loading branch information
yiheng-wang-nv committed Sep 7, 2023
1 parent 5c8fb69 commit 3bf169e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions models/spleen_deepedit_annotation/scripts/transforms.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,8 @@ def transform_points(self, point, affine):
"""transform point to the coordinates of the transformed image
point: numpy array [bs, N, 3]
"""
bs, N = point.shape[:2]
point = np.concatenate((point, np.ones((bs, N, 1))), axis=-1)
bs, n = point.shape[:2]
point = np.concatenate((point, np.ones((bs, n, 1))), axis=-1)
point = rearrange(point, "b n d -> d (b n)")
point = affine @ point
point = rearrange(point, "d (b n)-> b n d", b=bs)[:, :, :3]
Expand Down

0 comments on commit 3bf169e

Please sign in to comment.