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
'STUDY > ERROR' 카테고리의 다른 글
Cannot handle this data type: (1, 1, 3), <f8 해결방법 (0) | 2023.04.20 |
---|---|
An exception has occurred, use %tb to see the full traceback. 주피터 노트북에서 argparser 사용하기 (0) | 2023.04.11 |
[파이썬 가상환경] python 가상환경 진입 오류 해결 (0) | 2022.12.16 |
downgrade the protobuf package to 3.20.x or lower. (0) | 2022.06.23 |
unicodedecodeerror: 'cp949' codec can't decode byte 0x99 in position 15: illegal multibyte sequence (0) | 2022.06.22 |