Fix issues woth inplace modification of tensors in BaseInstance3DBoxes (

#283)
open-mmlab · Jan 22, 2021 · 0a7fb9c · 0a7fb9c
1 parent f472787
commit 0a7fb9c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mmdet3d/core/bbox/structures/base_box3d.py b/mmdet3d/core/bbox/structures/base_box3d.py
@@ -57,7 +57,7 @@ def __init__(self, tensor, box_dim=7, with_yaw=True, origin=(0.5, 0.5, 0)):
  else:
  self.box_dim = box_dim
  self.with_yaw = with_yaw
- self.tensor = tensor
+ self.tensor = tensor.clone()
 
  if origin != (0.5, 0.5, 0):
  dst = self.tensor.new_tensor((0.5, 0.5, 0))

diff --git a/mmdet3d/core/bbox/structures/cam_box3d.py b/mmdet3d/core/bbox/structures/cam_box3d.py
@@ -63,7 +63,7 @@ def __init__(self,
  else:
  self.box_dim = box_dim
  self.with_yaw = with_yaw
- self.tensor = tensor
+ self.tensor = tensor.clone()
 
  if origin != (0.5, 1.0, 0.5):
  dst = self.tensor.new_tensor((0.5, 1.0, 0.5))