Как я могу сохранить тензор изображений в формате PNG?

Question

У меня есть tensor, который выглядит как:

Tensor("Identity:0", shape=(10000, 32, 32, 3), dtype=float32)

Я понял, как его перебрать:

for adv_x in tf.unstack(adv):
    asnumpy = tf.Session().run(tf.unstack(adv_x))
    print(asnumpy)

Возвращает:

...
Tensor("unstack:9997", shape=(32, 32, 3), dtype=float32)
Tensor("unstack:9998", shape=(32, 32, 3), dtype=float32)
Tensor("unstack:9999", shape=(32, 32, 3), dtype=float32)

Как я могу получить значения каждого и сохранить в png?

Это с python3.7 и tensorflow1.13

У меня ошибка:

tensorflow.python.framework.errors_impl.FailedPreconditionError: Attempting to use uninitialized value conv2d_1/kernel
         [[{{node conv2d_1/kernel/read}}]]

Answer 1

Преобразуйте его в numy, а затем сохраните его, используя scipy.misc.imsave

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    asnumpy = sess.run(tf.unstack(adv)) # `asnumpy` stores images as numpy arrays

Для хранения:

import scipy.misc

for i, image in enumerate(asnumpy):
    scipy.misc.imsave('image' + str(i) + '.png', image)