Pythonいぬ

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

2020-04-11から1日間の記事一覧

PytorchのDataLoaderから任意のバッチを出力

Pytorchで自作Datasetを作って、さらにDataLoaderからのバッチ出力を(x, y)だけではなく(x1, x2, y)とか(x, y, c)などと自由に制御したいときがある。 例えばtriplet lossを使いたいときは、ターゲットクラスyに対して複数の入力(x1, x2)を取ってくる必要が…