Pytorch 이미지를 텐서로 변환하는 방법

Question

다음과 같이 여러 이미지를 저장했습니다.

for name in image_names: 
    images.append(cv2.imread("./train_mini/"+name)) 

그리고 나중에 CNN 교육에서 이러한 이미지를 사용하고 싶습니다.

“RuntimeError: can’t convert a given np.ndarray to a tensor - it has an invalid type. The only supported types are: double, float, int64, int32, and uint8.” 

그래서 나는 이미지의 데이터 유형을 체크하고 "개체"이었다 : 나는 아래와 같이 "torch.utils.data.TensorDataset"에서 그 데이터를 저장 그러나, 그것은 오류가 표시됩니다.

train = torch.utils.data.TensorDataset(torch.from_numpy(X_train), torch.from_numpy(Y_train)) 

이 문제를 어떻게 해결할 수 있습니까? 나는 완전히 붙어있어 ...

Answer 1

나는 Pytorsh가 ndarray의 ndarray를 지원하지 않는다고 생각한다. 이것이 왜이 오류를 보게되는지이다.