Skip to content

Commit

Permalink
Reorder operators to ensure Neg operator is not used (#1584)
Browse files Browse the repository at this point in the history
  • Loading branch information
BloodAxe committed Oct 29, 2023
1 parent 4d20655 commit 776c635
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/super_gradients/training/utils/bbox_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ def batch_distance2bbox(points: Tensor, distance: Tensor, max_shapes: Optional[T
"""
lt, rb = torch.split(distance, 2, dim=-1)
# while tensor add parameters, parameters should be better placed on the second place
x1y1 = -lt + points
x1y1 = points - lt
x2y2 = rb + points
out_bbox = torch.cat([x1y1, x2y2], dim=-1)
if max_shapes is not None:
Expand Down

0 comments on commit 776c635

Please sign in to comment.