torch_concepts.data.utils.transform_images

transform_images(images, transformations, colors=None, degrees=None, scales=None)[source]

Apply a sequence of transformations to a batch of images.

Parameters:
  • images – Tensor of shape [N, H, W] or [N, 3, H, W].

  • transformations – List of transformation names (e.g., [‘colorize’, ‘affine’]).

  • colors – Optional color labels for colorization.

  • degrees – Optional rotation degrees for affine transform.

  • scales – Optional scaling factors for affine transform.

Returns:

Transformed images.

Return type:

Tensor