pytorchでのimagefloderの使用

問題の鍵は、写真とカテゴリ間の対応をどのように決定するかです

from torchvision.datasets import ImageFolder

dataset = ImageFolder('F:/my_code/data/0dB/train/')

データの読み込み

print(dataset.imgs[2])

この時点で、dataset.imgsは多くの要素を含むリストであり、2番目の要素
が選択され、出力は次のようになります。

('F:/my_code/data/0dB/train/BFSK\\BFSK100.jpg', 0)

つまり、
イメージフォルダはフォルダ名の順に配置する必要があります。BFSKは0に
対応し、BPSKは1に対応し
ます。