1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorch で学ぶ画像向け Deep Learning
Connected

演習

画像を生成する

GAN の設計と学習が完了したので、生成できる画像の品質を評価していきます。まずは目視で確認し、生成結果がポケモンに見えるかどうかを確かめましょう。そのために、ジェネレーターへの入力としてランダムノイズを作成し、モデルに渡して出力をプロットします。

学習済み重みを持つ Deep Convolutional Generator は gen として利用できます。torch と、matplotlib.pyplot は plt としてすでにインポート済みです。

指示

100 XP
  • 生成する画像枚数を表す num_images_to_generate × 入力ノイズサイズ 16 の形状を持つランダムノイズのテンソルを作成し、noise に代入します。
  • ノイズをジェネレーターに渡して画像を生成し、fake に代入します。
  • for ループ内で、fake をスライスして i 番目の画像を取り出し、image_tensor に代入します。
  • image_tensor の次元を (color, height, width) から (height, width, color) に並べ替え、出力を image_tensor_permuted に代入します。