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をサポートしていないと思うので、このエラーが表示されます。