본문 바로가기

STUDY/ERROR

ValueError: pic should be 2/3 dimensional. Got 4 dimensions.해결법 : torch.from_numpy()

728x90

ValueError: pic should be 2/3 dimensional. Got 4 dimensions.

 

 

 

torchvision 라이브러리에서 지원하는  ToTensor()함수로 ndarr를 tensor로 바꿀 때 아래와 같은 오류 ValueError: pic should be 2/3 dimensional. Got 4 dimensions. 가 발생한다면 from_numpy() 함수를 사용하면 해결할 수 있다.

 

*사용법

import torch

image = torch.from_numpy('numpy img 넣기')

에러가 해결되고! 아래 그림과 같이 img_lr_up가 tensor type으로 바뀐 것을 알 수 있다.

 

728x90