Pytorch如何将图像转换为张量

Question

我存储了多张图片，如下所示。

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

我想在以后的CNN培训中使用这些图像。但是，当我将这些数据存储在“torch.utils.data.TensorDataset”中时，如下所示，它显示错误：

“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.” 

所以我检查了图像的数据类型，它是“对象”。

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

我该如何解决这个问题？我完全陷入困境......

Answer 1

我认为Pytorsh不支持ndarray的ndarray，这就是你看到这个错误的原因。