Spaces:
Runtime error
Runtime error
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 |
|