Pytorch：ImageFolderを使用して画像を読み込めません。

Question

"ImageFolder"を使用して画像を読み込もうとしています。

data_dir = './train_dog'   # directory structure is  
train_dog/image 
dset = datasets.ImageFolder(data_dir, transform) 
train_loader = torch.utils.data.DataLoader(dset, batch_size=128, shuffle=True) 

ただし、動作していないようです。そこで、以下のように保存データを確認しました。

print dset[0][0] 

次に3テンソル（サイズ64x64）しか表示されません。

[torch.FloatTensor of size 3x64x64] 

フォルダには10,000以上の画像があります。どうしてすべてのデータを保存できないのですか？

Answer 1

あなたはこれを試してみてください：画像ファイルの数別名データセットの大きさを表し

print len(dset) 

を。

dset[0]は、dset[0][0]に入力画像のテンソルが含まれ、dset[0][1]に対応するラベルまたはターゲットが含まれているデータセットの（シャッフル）最初のインデックスを意味します。