[Deep Learning] numpy.ndarray를 torch.Tensor로 변환하는 방법

OpenCV를 이용하여 이미지 등의 파일을 불러올 경우 

img1 = cv2.imread('your file path')
print(type(img1))

img1의 타입

위의 사진과 같이 numpy.ndarray의 형태로 저장된다.

이미지 데이터 전처리 등의 과정을 거치려면 Tensor형태로 변환을 해 주어야 할 때가 있는데 그 때는

 

1) torch.Tensor()를 이용한 방법
tensor_img1 = torch.Tensor(cv2.resize(img1)

2) IQA_Pytorch를 이용한 방법
tensor_img1 = utils.prepare_image(img1)

위의 코드를 이용하여 저장하면 torch.Tensor의 형태로 이미지 데이터를 저장할 수 있다.