PyTorchデータ処理、データセットの使用、DataLoaderとそのツール

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

第二に、

 

おすすめ

転載: www.cnblogs.com/xixixing/p/12759849.html