Skip to content

pil_to_tensor returns non-contiguous tensor #4199

Closed
@aaronzs

Description

@aaronzs

🐛 Bug torchvision.transforms.functional.pil_to_tensor returns non-contiguous tensor

To Reproduce

from PIL import Image
from torchvision.transforms.functional import pil_to_tensor

pic = Image.open("3_channel_pic.png")
pil_to_tensor(pic).is_contiguous()
# False

Expected behavior

.contiguous() is recommended to be used after .permute(), so that

  1. consistent behavior with to_tensor();
  2. increase computational efficiency

cc @vfdev-5

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions