Skip to content

Commit

Permalink
changed if condition of rt_mat to be consistent with code
Browse files Browse the repository at this point in the history
  • Loading branch information
gopi-erabati committed Jul 6, 2021
1 parent dda857b commit 7c68a15
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions mmdet3d/core/bbox/structures/coord_3d_mode.py
Original file line number Diff line number Diff line change
Expand Up @@ -243,9 +243,8 @@ def convert_point(point, src, dst, rt_mat=None):
f'Conversion from Coord3DMode {src} to {dst} '
'is not supported yet')

if isinstance(rt_mat, np.ndarray):
if not isinstance(rt_mat, torch.Tensor):
rt_mat = arr.new_tensor(rt_mat)

if rt_mat.size(1) == 4:
extended_xyz = torch.cat(
[arr[:, :3], arr.new_ones(arr.size(0), 1)], dim=-1)
Expand Down

0 comments on commit 7c68a15

Please sign in to comment.