Почему tf.write_file не записывает файл? - PullRequest
0 голосов
/ 18 апреля 2019

Мой код:

i = 0
fgsm = FastGradientMethod(wrap)
adv = fgsm.generate(x_test_tensor, **fgsm_params)

for adv_x in tf.unstack(adv):
    img = tf.cast(adv_x, dtype=tf.uint8)
    tf_image = tf.image.encode_jpeg(img)
    tf.write_file('adversarial_examples/' + str(i) + '.jpg', tf_image)
    i += 1

Где FastGradientMethod от cleverhans. Это работает без ошибок, но JPG не существует в моей папке. Чего не хватает?

1 Ответ

1 голос
/ 18 апреля 2019

возможно путь к файлу назначения неверен

import os

for adv_x in tf.unstack(adv):
    img = tf.cast(adv_x, dtype=tf.uint8)
    tf_image = tf.image.encode_jpeg(img)
    pth = os.path.join('adversarial_examples', '%s.jpg'%i)
    tf.write_file(pth, tf_image)
    i += 1
...