torchvisionは、多くの画像処理ツールを提供するPyTorchのビジュアルツールキットです。
データセットはImageFolderツールを使用して、カスタマイズされた画像を取得し、カテゴリラベルを自動的に生成します。切り抜き、回転、標準化、正規化など
DataLoaderは、データセットをスクランブルし、それらをバッチに分割し、並行して加速することができます。
1.写真と名前付きタグを取得する
で使用torchvision.datasets ImageFolderのツール、機能:
1.フォルダ名はカテゴリ名です
2.フォルダーのラベルを上から下に自動的に作成します(0、1、2、...)。Class_to_idx、imgs属性を表示できます。
3.各画像のデータとラベルを返す
torchvision.datasets インポートImageFolder のDataSet = ImageFolder(「E:/データ/ dogcat_2 /鉄道/ 」)#取得パス、図データ、ラベルのすべてのリターン。 印刷(dataset.class_to_idx) #のビューのカテゴリ名、および対応するタグ。 印刷(dataset.imgs) #は、すべての画像のパスを表示し、対応するラベル
印刷(データセット[0] [1]) #図1のラベルの データセット[0] [0] #図1のデータ。1
第二に、