[Pytorch] RuntimeError: Input type (torch.cuda.ByteTensor) and weight type (torch.cuda.FloatTensor) should be the same

파이썬에서 딥러닝 프레임워크를 pytorch로 쓰다가 다음과 같은 에러 메시지를 나왔다.

 

 

RuntimeError: Input type (torch.cuda.ByteTensor) and weight type (torch.cuda.FloatTensor) should be the same

 

상황은 다음과 같다.

 

import torch
import cv2

image = cv2.imread(./sample_image.jpg)
tensor = torch.from_numpy(image).to(device).unsqueeze(0).permute(0,3,1,2)

 

 

이를 해결하기 위해 to(device) 앞에 .float()처리를 해주었다.

 

import torch
import cv2

image = cv2.imread(./sample_image.jpg)
tensor = torch.from_numpy(image).float().to(device).unsqueeze(0).permute(0,3,1,2)

 

 

매우 잘 처리되었다. 

 

 

 

 

-히비스서커스-