Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/lmdeploy.git" did not exist on "db3b986be00c035f8ef585093371e312972dc414"
Unverified Commit 98f1a409 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Replace `getbands()` with `get_image_num_channels()` (#6941)

parent ffd5a567
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torchvision/transforms/functional.py
View file @ 98f1a409
......@@ -167,7 +167,7 @@ def to_tensor(pic) -> Tensor:
if pic.mode == "1":
img = 255 * img
img = img.view(pic.size[1], pic.size[0], len(pic.getbands()))
img = img.view(pic.size[1], pic.size[0], F_pil.get_image_num_channels(pic))
# put it from HWC to CHW format
img = img.permute((2, 0, 1)).contiguous()
if isinstance(img, torch.ByteTensor):
......@@ -205,7 +205,7 @@ def pil_to_tensor(pic: Any) -> Tensor:
# handle PIL Image
img = torch.as_tensor(np.array(pic, copy=True))
img = img.view(pic.size[1], pic.size[0], len(pic.getbands()))
img = img.view(pic.size[1], pic.size[0], F_pil.get_image_num_channels(pic))
# put it from HWC to CHW format
img = img.permute((2, 0, 1))
return img
......
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