今更PyTorchマスターへの道 (4) - データロード
今日もPython Engineerのチュートリアルをこなしていきます.09から10をやりました.
Tutorial 09
「Dataset And Dataloader - PyTorch Beginner 09」では,Dataset,DataLoaderクラスの使い方が分かる.
Datasetを継承したクラスを作り以下の3つの関数を記述する__init__関数__getitem__関数__len__関数
DataLoaderに作ったDatasetクラスを渡してデータを取り出す- 画像処理の例は特に出ていないが,
Datasetを継承したクラスとその関数のルールを守っておけば問題なさそう
Tutorial 10
「Dataset Transforms - PyTorch Beginner 10」では,Data augmentation (transform) の実装の基礎が分かる.
- 大体の操作はPyTorchに既に実装されている
- 独自の操作を行いたい場合はCallableクラスを定義する
__call__関数を含めて,そこに操作を書き込む- あとはDatasetクラスの
__getitem__関数でデータを返す際にこのCallableクラスを適用する
短めですが,今日はここまで.