Skip to content

GitLab

"fmoe/git@developer.sourcefind.cn:OpenDAS/fastmoe.git" did not exist on "0f091a1d84ab97a869e40870e9608380e7ca49c1"
Commit bb5af1d7 authored by Philip Meier's avatar Philip Meier Committed by Francisco Massa
Browse files

generalize number of bands calculation in to_tensor (#1781)

parent 70dee777
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 8 deletions
torchvision/transforms/functional.py
View file @ bb5af1d7
...@@ -85,14 +85,8 @@ def to_tensor(pic): ...@@ -85,14 +85,8 @@ def to_tensor(pic):
img = 255 * torch.from_numpy(np.array(pic, np.uint8, copy=False)) img = 255 * torch.from_numpy(np.array(pic, np.uint8, copy=False))
else: else:
img = torch.ByteTensor(torch.ByteStorage.from_buffer(pic.tobytes())) img = torch.ByteTensor(torch.ByteStorage.from_buffer(pic.tobytes()))
# PIL image mode: L, LA, P, I, F, RGB, YCbCr, RGBA, CMYK
if pic.mode == 'YCbCr': img = img.view(pic.size[1], pic.size[0], len(pic.getbands()))
nchannel = 3
elif pic.mode == 'I;16':
nchannel = 1
else:
nchannel = len(pic.mode)
img = img.view(pic.size[1], pic.size[0], nchannel)
# put it from HWC to CHW format # put it from HWC to CHW format
# yikes, this transpose takes 80% of the loading time/CPU # yikes, this transpose takes 80% of the loading time/CPU
img = img.transpose(0, 1).transpose(0, 2).contiguous() img = img.transpose(0, 1).transpose(0, 2).contiguous()
......
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