問題タブ [torchvision]

このチュートリアルでは、次のコードを使用してデータセットを読み取ります。

そのコードでは、png データセットが異なるフォルダーに分けられています。

各フォルダの名前は、フォルダ内の png ファイルのカテゴリを意味します。

私は次のような状況にいます：

• すべての png ファイルは 1 つのフォルダーにあります。
• 各 png ファイルのカテゴリは train.csv ファイルに記録されます。

上記の関数を使用してコードを実装するにはどうすればよいdatasets.ImageFolderですか?

torch.Size([32, 3, 64, 64]) のような torch tensor があります。

私はテンソルをこれらのアサーションを渡すことができるものに変換しようとしています:

私は現在、テンソルを変換するためにこれを行っています:

そして、私はこのエラーを受け取ります:

タイプ（画像）がリストでタイプ（画像[0]がnp.ndarray）になるようにテンソルを正しく変換するにはどうすればよいですか？どんな助けでも大歓迎です。前もって感謝します。