PyTorch-データの読み込み

PyTorchには、データセットのロードと準備に使用されるtorchvisionというパッケージが含まれています。これには、データセットの変換とロードに役立つDatasetとDataLoaderという2つの基本的な関数が含まれています。

データセット

データセットは、指定されたデータセットからデータポイントを読み取って変換するために使用されます。実装する基本的な構文は以下のとおりです-

trainset = torchvision.datasets.CIFAR10(root = './data', train = True,
   download = True, transform = transform)

DataLoaderは、データのシャッフルとバッチ処理に使用されます。マルチプロセッシングワーカーと並行してデータをロードするために使用できます。

trainloader = torch.utils.data.DataLoader(trainset, batch_size = 4,
   shuffle = True, num_workers = 2)

例:CSVファイルの読み込み

PythonパッケージPandaを使用してcsvファイルをロードします。元のファイルの形式は次のとおりです(画像名、68個のランドマーク-各ランドマークにはax、y座標があります)。

landmarks_frame = pd.read_csv('faces/face_landmarks.csv')

n = 65
img_name = landmarks_frame.iloc[n, 0]
landmarks = landmarks_frame.iloc[n, 1:].as_matrix()
landmarks = landmarks.astype('float').reshape(-1, 2)