kerasで複数のモデルと重みを保存してテストしたい
解決済
回答 1
投稿
- 評価
- クリップ 1
- VIEW 927
kerasでpix2pixのモデルを実装しようと試みている初学者です。
あるサイトに掲載されているサンプルコードを参考にしています。
その中で、以下の3つのモデルが定義されコンパイルされています。
generator_model.compile(loss='mae', optimizer=opt_discriminator)
DCGAN_model.compile(loss=loss, loss_weights=loss_weights, optimizer=opt_dcgan)
discriminator_model.compile(loss='binary_crossentropy', optimizer=opt_discriminator)
これらを含んだ「pix2pixモデル」を学習させた後、重みを含めたモデルの保存を行い、テストしたいと考えています。
その際、例えば下記のように1つ1つ別々にモデルを保存することはできますが、
generator_model.save('generator.h5')
DCGAN_model.save('DCGAN.h5')
discriminator_model.save('discriminator.h5')
学習後に、保存されたこれらのモデルを全て使ってテストすることは可能でしょうか?
もしくは、3つのモデルを1つにまとめてから学習させ、保存し、テストした方がよいのでしょうか。
こうした複数モデルの保存とテストの方法を教えて頂けますと非常に助かります。
何卒よろしくお願い致します。
該当のソースコード
def train():
# load data
rawImage, procImage, rawImage_val, procImage_val = load_data(datasetpath)
img_shape = rawImage.shape[-3:]
patch_num = (img_shape[0] // patch_size) * (img_shape[1] // patch_size)
disc_img_shape = (patch_size, patch_size, procImage.shape[-1])
# train
opt_dcgan = Adam(lr=1E-3, beta_1=0.9, beta_2=0.999, epsilon=1e-08)
opt_discriminator = Adam(lr=1E-3, beta_1=0.9, beta_2=0.999, epsilon=1e-08)
# load generator model
generator_model = load_generator(img_shape, disc_img_shape)
# load discriminator model
discriminator_model = load_DCGAN_discriminator(img_shape, disc_img_shape, patch_num)
generator_model.compile(loss='mae', optimizer=opt_discriminator)
discriminator_model.trainable = False
DCGAN_model = load_DCGAN(generator_model, discriminator_model, img_shape, patch_size)
loss = [l1_loss, 'binary_crossentropy']
loss_weights = [1E1, 1]
DCGAN_model.compile(loss=loss, loss_weights=loss_weights, optimizer=opt_dcgan)
discriminator_model.trainable = True
discriminator_model.compile(loss='binary_crossentropy', optimizer=opt_discriminator)
補足情報(FW/ツールのバージョンなど)
これらモデルは「functional API」を使用して定義されております。
-
気になる質問をクリップする
クリップした質問は、後からいつでもマイページで確認できます。
またクリップした質問に回答があった際、通知やメールを受け取ることができます。
クリップを取り消します
-
良い質問の評価を上げる
以下のような質問は評価を上げましょう
- 質問内容が明確
- 自分も答えを知りたい
- 質問者以外のユーザにも役立つ
評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。
質問の評価を上げたことを取り消します
-
評価を下げられる数の上限に達しました
評価を下げることができません
- 1日5回まで評価を下げられます
- 1日に1ユーザに対して2回まで評価を下げられます
質問の評価を下げる
teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。
- プログラミングに関係のない質問
- やってほしいことだけを記載した丸投げの質問
- 問題・課題が含まれていない質問
- 意図的に内容が抹消された質問
- 過去に投稿した質問と同じ内容の質問
- 広告と受け取られるような投稿
評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。
質問の評価を下げたことを取り消します
この機能は開放されていません
評価を下げる条件を満たしてません
質問の評価を下げる機能の利用条件
この機能を利用するためには、以下の事項を行う必要があります。
- 質問回答など一定の行動
-
メールアドレスの認証
メールアドレスの認証
-
質問評価に関するヘルプページの閲覧
質問評価に関するヘルプページの閲覧
check解決した方法
0
何となく自己解決いたしました。
そもそも画像生成のテストなので、
学習済みの「generatorモデルだけ」をロードしてテストすれば良かっただけでした。
投稿
-
回答の評価を上げる
以下のような回答は評価を上げましょう
- 正しい回答
- わかりやすい回答
- ためになる回答
評価が高い回答ほどページの上位に表示されます。
-
回答の評価を下げる
下記のような回答は推奨されていません。
- 間違っている回答
- 質問の回答になっていない投稿
- スパムや攻撃的な表現を用いた投稿
評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。
15分調べてもわからないことは、teratailで質問しよう!
- ただいまの回答率 88.34%
- 質問をまとめることで、思考を整理して素早く解決
- テンプレート機能で、簡単に質問をまとめられる