distance2obb 解码调试出来的结果 为 [N,4] 和注释里面的[N,2]对不上

[vansin]

mmrotate/mmrotate/structures/bbox/transforms.py

Lines 83 to 97 in 4922856

	def distance2obb(points: torch.Tensor,
	distance: torch.Tensor,
	angle_version: str = 'oc'):
	"""Convert distance angle to rotated boxes.
	Args:
	points (Tensor): Shape (B, N, 2) or (N, 2).
	distance (Tensor): Distance from the given point to 4
	boundaries and angle (left, top, right, bottom, angle).
	Shape (B, N, 5) or (N, 5)
	angle_version: angle representations.
	Returns:
	dict[str, torch.Tensor]: Gaussian distributions.
	"""
	distance, angle = distance.split([4, 1], dim=-1)

distance2obb 解码 调试出来的结果 为 [N,4] 和注释里面的[N,2]对不上。