Numpy 형 이미지 Tensor 변환시

Question

numpy형 이미지를 받아 ToTensor()로 텐서로 변환했는데 픽셀값이 0-1로 바뀌지 않고 0-255로 그대로 입니다 이유가 무엇일까요 ㅠㅠdef eval(model, img): trans &#61; transforms.ToTensor() img &#61; trans(img) print(img.size()) print(img.type()) print(img.min(), img.max())fakeimg &#61; np.random.randint(256, size&#61;(50, 50, 3))eval(net, fakeimg)>torch.Size([3, 50, 50])>torch.LongTensor>tensor(0) tensor(255) # 왜 0-1사이로 바뀌지 않는지 궁금합니다 ㅠㅠ

휴지통비우기 · Answer

https://pytorch.org/vision/stable/generated/torchvision.transforms.ToTensor.html공식문서에는 unit8로 되어 있어야 스케일링된다고 하네요.

Numpy 형 이미지 Tensor 변환시

1개의 답변

Q&A 목록