Как восстановить генератор GAN из модели Tensorflow?

Я пытаюсь восстановить обученный генератор генеративно-состязательной сети с использованием модели Tensorflow (метаграф и контрольная точка).

Я новичок в tensorflow и python, поэтому не уверен, что то, что я делаю, имеет смысл. уже пытался импортировать метаграф из метафайла и восстанавливать переменные из контрольной точки, но я уверен, что делать дальше. Моя цель — восстановить обученный генератор с последней контрольной точки, а затем использовать его для генерации новых данных из ввода шума.

Вот ссылка на диск с файлами моделей: https://drive.google.com/drive/folders/1MaELMC4aOroSQlMJ32J3_ff3wxiBT_Fq?usp=sharing

До сих пор я пробовал следующее, и, похоже, он загружает график:


# import the graph from the file
imported_graph = tf.train.import_meta_graph("../../models/model-9.meta")

# list all the tensors in the graph
for tensor in tf.get_default_graph().get_operations():
    print (tensor.name)

# run the session
with tf.Session() as sess:
    # restore the saved vairable
    imported_graph.restore(sess, "../../models/model-9")

Однако я не уверен, что делать дальше. Можно ли запустить только обученный генератор, используя эти файлы? Как я могу получить к нему доступ?


python tensorflow neural-network generative-adversarial-network
person Carlos Tejeda    schedule 19.06.2019    source источник

Ответы (1)


arrow_upward
0
arrow_downward

В документе Tensorflow 2 они сохраняют и генератор, и дискриминатор. Однако они не объясняют, как только восстановить генератор.

checkpoint_dir = './training_checkpoints'
checkpoint_prefix = os.path.join(checkpoint_dir, "ckpt")
checkpoint = tf.train.Checkpoint(generator_optimizer=generator_optimizer,
                                 discriminator_optimizer=discriminator_optimizer,
                                 generator=generator,
                                 discriminator=discriminator)

А затем восстановить с помощью

checkpoint.restore(tf.train.latest_checkpoint(checkpoint_dir))

Из https://www.tensorflow.org/tutorials/generative/dcgan#save_checkpoints

person pushStack    schedule 09.11.2019