rlawjdghek commited on
Commit
2dbd7c6
β€’
1 Parent(s): 107d5d9

Update preprocess/detectron2/detectron2/modeling/box_regression.py

Browse files
preprocess/detectron2/detectron2/modeling/box_regression.py CHANGED
@@ -180,7 +180,7 @@ class Box2BoxTransformRotated:
180
  ), "Input boxes to Box2BoxTransformRotated are not valid!"
181
  return deltas
182
 
183
- def apply_deltas(self, deltas, boxes):
184
  """
185
  Apply transformation `deltas` (dx, dy, dw, dh, da) to `boxes`.
186
 
 
180
  ), "Input boxes to Box2BoxTransformRotated are not valid!"
181
  return deltas
182
 
183
+ def apply_deltas(self, deltas: torch.Tensor, boxes: torch.Tensor):
184
  """
185
  Apply transformation `deltas` (dx, dy, dw, dh, da) to `boxes`.
186