Mugichoko's blog

Mugichoko’s blog

しがない研究者のプログラミングを中心としたメモ書き.

今更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クラスを適用する

短めですが,今日はここまで.