Commit 39eb4ec6 authored by liyining's avatar liyining Committed by zhouzaida
Browse files

Fix typos in processing.py

parent a1fb0022
......@@ -1086,12 +1086,12 @@ class RandomFlip(BaseTransform):
else:
raise ValueError(
f"Flipping direction must be 'horizontal', 'vertical', \
or 'diagnal', but got '{direction}'")
or 'diagonal', but got '{direction}'")
return flipped
def flip_keypoints(self, keypoints: np.ndarray, img_shape: Tuple[int, int],
direction: str) -> np.ndarray:
"""Flip keypoints horizontally, vertically or diagnally.
"""Flip keypoints horizontally, vertically or diagonally.
Args:
keypoints (numpy.ndarray): Keypoints, shape (..., 2)
......@@ -1116,7 +1116,7 @@ class RandomFlip(BaseTransform):
else:
raise ValueError(
f"Flipping direction must be 'horizontal', 'vertical', \
or 'diagnal', but got '{direction}'")
or 'diagonal', but got '{direction}'")
flipped = np.concatenate([keypoints, meta_info], axis=-1)
return flipped
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment