Spaces:
Running
on
Zero
Running
on
Zero
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 |
|