cGANによるデータ水増しおよびPytorchのDataLoader

Question

実現したいこと

1.学習済みのconditional GANの生成器を使って、データを増やしたい。
2.conditional GANの学習に使ったDataLoaderと1番で増やしたデータを合わせて、新たにDataLoaderを作成したい

現状

下記リンクを参考に、MNISTデータを使ってconditional GANのネットワークを学習させました。
https://gurutaka-log.com/pytorch-cgan

Answer 1

水増しデータを一旦Numpy配列で準備し、元のデータとデータ数次元の方向にconcatしました。
これで新たにpytorchのdatasetを作りました。