Implement ToTensorv2 for multiple images

[loicmagne]

The current behavior is the following:

>>> import numpy as np
>>> import albumentations as A
>>> from albumentations.pytorch import ToTensorV2
>>> 
>>> t = A.Compose([
...     ToTensorV2(),
... ])
>>> x = [np.zeros((224, 224, 3))]
>>> x = t(images=x)['images']
>>> 
>>> print(type(x[0]))
<class 'numpy.ndarray'>

Is that an expected behavior? I added a handler on the ToTensorV2 class to also transform lists of images, let me know if I'm missing something

Summary by Sourcery

New Features:

• Add support for transforming lists of images to tensors using ToTensorV2.

[sourcery-ai]

Reviewer's Guide by Sourcery

This PR extends the ToTensorV2 transform to handle lists of images by adding a new apply_to_images method and updating the targets property to include the 'images' key. The implementation iterates through the list of images and applies the existing single-image transformation to each image.

Updated class diagram for ToTensorV2

classDiagram
    class ToTensorV2 {
        - transpose_mask: bool
        - p: float
        - always_apply: bool
        + apply(img: np.ndarray, **params: Any) torch.Tensor
        + apply_to_images(images: list[np.ndarray], **params: Any) list[torch.Tensor]
        + apply_to_mask(mask: np.ndarray, **params: Any) torch.Tensor
        + targets() dict[str, Any]
    }
    note for ToTensorV2 "Added apply_to_images method and updated targets property to handle lists of images."

Loading

File-Level Changes

Change	Details	Files
Added support for transforming lists of images in ToTensorV2	Added new apply_to_images method that processes a list of numpy arrays into a list of torch tensors Updated targets property to include 'images' key mapping to the new apply_to_images method Maintained existing single image transformation logic in the apply method	albumentations/pytorch/transforms.py

---

Tips and commands

Interacting with Sourcery

• Trigger a new review: Comment @sourcery-ai review on the pull request.
• Continue discussions: Reply directly to Sourcery's review comments.
• Generate a GitHub issue from a review comment: Ask Sourcery to create an
  issue from a review comment by replying to it.
• Generate a pull request title: Write @sourcery-ai anywhere in the pull
  request title to generate a title at any time.
• Generate a pull request summary: Write @sourcery-ai summary anywhere in
  the pull request body to generate a PR summary at any time. You can also use
  this command to specify where the summary should be inserted.

Customizing Your Experience

Access your dashboard to:

• Enable or disable review features such as the Sourcery-generated pull request
  summary, the reviewer's guide, and others.
• Change the review language.
• Add, remove or edit custom review instructions.
• Adjust other review settings.

Getting Help

• Contact our support team for questions or feedback.
• Visit our documentation for detailed guides and information.
• Keep in touch with the Sourcery team by following us on X/Twitter, LinkedIn or GitHub.

[sourcery-ai]

Hey @loicmagne - I've reviewed your changes - here's some feedback:

Overall Comments:

• Please add test coverage for the new apply_to_images functionality to verify it correctly handles lists of images

Here's what I looked at during the review

• 🟡 General issues: 3 issues found
• 🟢 Security: all looks good
• 🟢 Testing: all looks good
• 🟢 Complexity: all looks good
• 🟢 Documentation: all looks good

---

Sourcery is free for open source - if you like our reviews please consider sharing them ✨

• X
• Mastodon
• LinkedIn
• Facebook

_{Help me be more useful! Please click 👍 or 👎 on each comment and I'll use the feedback to improve your reviews.}

[ternaus]

@loicmagne https://github.com/albumentations-team/albumentations/blob/main/CONTRIBUTING.md#using-pre-commit-hooks

[ternaus]

@loicmagne Thanks!

[loicmagne]

Thanks for the quick merge @ternaus!