-
카테고리
-
세부 분야
딥러닝 · 머신러닝
-
해결 여부
미해결
torchvision 전처리 관련 질문있습니다.
22.03.31 22:00 작성 조회수 105
1
torchvision 전처리 예시에서 넘파이 배열에 tr.Compose로 tr.ToPILImage 적용 후 tr.ToTensor를 적용해주는데, 그렇다면
inputs = torch.FloatTensor(inputs)
inputs = inputs.permute(2,0,1)
labels = torch.FloatTensor(labels)
해당 코드를 안써도 무방한거 아닌가요??
헷갈리네요ㅠㅠ
답변을 작성해보세요.
1
딥러닝호형
지식공유자2022.03.31
안녕하세요.
좋은 질문 감사합니다.
해당 경우에는 PILImage인 경우에만 가능한 전처리를 사용하기 위해 텐서를 PILImage로 강제로 바꿔 준 예시입니다. 또한 tr.ToPILImage가 텐서 아니면 넘파이 배열만 받기 때문에 텐서로 시작하여 텐서 -> PIL -> 다시 텐서의 데이터 흐름을 보여드린 것입니다! 즉, 하나의 예시라고 보시면 되고 나중에 실제 데이터 다루실 때 주어진 데이터 형태에 맞게 플로우를 만들어 주시면 됩니다 :)
열공하세요!
답변 1