【Pytorch】Dataset/DataLoaderについてメモ - Qiita

概要 Pytorchでデータセット作成時に使用するDataset/DataLoaderあたりをメモ 参考： Pytorch TORCH.UTILS.DATA PyTorch transforms/Dataset/DataLoaderの基本動作を確認する データの前処理 データの前処理についてはtorchvision.transformsまたはalbumentationsあたりのライブラリがある。 どちらも基本的な動作は同じ。 前処理のクラスインスタンスをリストに詰め込んで、Compose()の引数にしてインスタンス作成。 Composeは、__call__(self, img)メソッドを持つので、作成したインスタンスの引数に画像を入れれば前処理される。 TORCHVISION.TRANSOFORMS albumentations(GitHub) 画像データ拡張ライブラリ ~ albumen