@@ -183,9 +183,9 @@ def _cxcywhr_to_xywhr(cxcywhr: torch.Tensor, inplace: bool) -> torch.Tensor:
183183 half_wh = cxcywhr [..., 2 :- 1 ].div (- 2 , rounding_mode = None if cxcywhr .is_floating_point () else "floor" ).abs_ ()
184184 r_rad = cxcywhr [..., 4 ].mul (torch .pi ).div (180.0 )
185185 # (cx - width / 2 * cos - height / 2 * sin) = x1
186- cxcywhr [..., 0 ].sub_ (half_wh [..., 0 ].mul (r_rad .cos ()).add ( half_wh [..., 1 ].mul (r_rad .sin () )).to (cxcywhr .dtype ))
186+ cxcywhr [..., 0 ].sub_ (half_wh [..., 0 ].mul (r_rad .cos ()).to ( cxcywhr . dtype )). sub_ ( half_wh [..., 1 ].mul (r_rad .sin ()).to (cxcywhr .dtype ))
187187 # (cy + width / 2 * sin - height / 2 * cos) = y1
188- cxcywhr [..., 1 ].add_ (half_wh [..., 0 ].mul (r_rad .sin ()).sub ( half_wh [..., 1 ].mul (r_rad .cos () )).to (cxcywhr .dtype ))
188+ cxcywhr [..., 1 ].add_ (half_wh [..., 0 ].mul (r_rad .sin ()).to ( cxcywhr . dtype )). sub_ ( half_wh [..., 1 ].mul (r_rad .cos ()).to (cxcywhr .dtype ))
189189
190190 return cxcywhr
191191
@@ -197,9 +197,9 @@ def _xywhr_to_cxcywhr(xywhr: torch.Tensor, inplace: bool) -> torch.Tensor:
197197 half_wh = xywhr [..., 2 :- 1 ].div (- 2 , rounding_mode = None if xywhr .is_floating_point () else "floor" ).abs_ ()
198198 r_rad = xywhr [..., 4 ].mul (torch .pi ).div (180.0 )
199199 # (x1 + width / 2 * cos + height / 2 * sin) = cx
200- xywhr [..., 0 ].add_ (half_wh [..., 0 ].mul (r_rad .cos ()).add ( half_wh [..., 1 ].mul (r_rad .sin () )).to (xywhr .dtype ))
200+ xywhr [..., 0 ].add_ (half_wh [..., 0 ].mul (r_rad .cos ()).to ( xywhr . dtype )). add_ ( half_wh [..., 1 ].mul (r_rad .sin ()).to (xywhr .dtype ))
201201 # (y1 - width / 2 * sin + height / 2 * cos) = cy
202- xywhr [..., 1 ].add_ (half_wh [..., 1 ].mul (r_rad .cos ()).sub ( half_wh [..., 0 ].mul (r_rad .sin () )).to (xywhr .dtype ))
202+ xywhr [..., 1 ].sub_ (half_wh [..., 0 ].mul (r_rad .sin ()).to ( xywhr . dtype )). add_ ( half_wh [..., 1 ].mul (r_rad .cos ( )).to (xywhr .dtype ))
203203
204204 return xywhr
205205
0 commit comments