Pythonいぬ

pythonを使った画像処理に関する記事を書いていきます

2020-03-01から1日間の記事一覧

PytorchでDataLoaderからデータを取り出す

PytorchのDataLoaderって便利だなと思いつつも、forループ以外の方法でデータを取り出すことができなかったので、中身を少し調べてみた。以下のようにすればデータがとれることがわかった。 tmp = testloader.__iter__() x1, y1 = tmp.next() x2, y2 = tmp.n…