Skip to content

Commit

Permalink
clarified docstring in random_affine as well
Browse files Browse the repository at this point in the history
  • Loading branch information
cansik committed Jun 4, 2024
1 parent 26ac056 commit 9164fe4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/super_gradients/training/transforms/transforms.py
Original file line number Diff line number Diff line change
Expand Up @@ -1483,7 +1483,7 @@ def random_affine(
:param img: Input image of shape [h, w, c]
:param targets: Input target
:param targets_seg: Targets derived from segmentation masks
:param target_size: Desired output shape
:param target_size: Desired output shape tuple formatted as (rows, cols).
:param degrees: Degrees for random rotation, when float the random values are drawn uniformly
from (-degrees, degrees).
:param translate: Translate size (in pixels) for random translation, when float the random values
Expand Down

0 comments on commit 9164fe4

Please sign in to comment.