Inflearn Community Q&A
asked
torchvision 전처리 관련 질문있습니다.
Written on
·
179
1
torchvision 전처리 예시에서 넘파이 배열에 tr.Compose로 tr.ToPILImage 적용 후 tr.ToTensor를 적용해주는데, 그렇다면
inputs = torch.FloatTensor(inputs)
inputs = inputs.permute(2,0,1)
labels = torch.FloatTensor(labels)
해당 코드를 안써도 무방한거 아닌가요??
헷갈리네요ㅠㅠ
딥러닝python인공신경망머신러닝 배워볼래요? pytorch
Answer 1
1
dlbro
Instructor
안녕하세요.
좋은 질문 감사합니다.
해당 경우에는 PILImage인 경우에만 가능한 전처리를 사용하기 위해 텐서를 PILImage로 강제로 바꿔 준 예시입니다. 또한 tr.ToPILImage가 텐서 아니면 넘파이 배열만 받기 때문에 텐서로 시작하여 텐서 -> PIL -> 다시 텐서의 데이터 흐름을 보여드린 것입니다! 즉, 하나의 예시라고 보시면 되고 나중에 실제 데이터 다루실 때 주어진 데이터 형태에 맞게 플로우를 만들어 주시면 됩니다 :)
열공하세요!