tf.gradients возвращает только [Нет]

Я попытался построить свой собственный алгоритм глубокого сна с помощью этого кода, используя Inception Neural Network от Google:

import tensorflow as tf
import matplotlib.pyplot as plt
import numpy as np
#I am using the Kadenze CADL helper function ---> https://github.com/pkmital/CADL/tree/master/session-4/libs
import inception

img = np.random.rand(1,1920,1080,3)

net = inception.get_inception_model()
tf.import_graph_def(net['graph_def'], name='inception')
graph = tf.get_default_graph()
layer = graph.get_tensor_by_name('inception/mixed5b_pool_reduce_pre_relu:0')
gradient = tf.gradients(tf.reduce_mean(layer), img)
sess = tf.Session()
init = tf.global_variables_initializer()
iters = 1440

sess.run(init)
for i in range(iters):
    print(i+1)
    grad = sess.run(gradient[0])[0]
    img += grad
    plt.imshow(img[0])
    plt.savefig('output/'+str(i+1)+'.png')
    plt.close('all')

Но линия tf.gradients(tf.reduce_mean(layer), img) только возвращается [None], Это (конечно) вызывает ошибку. Может кто-нибудь сказать мне, как это исправить?

0 ответов

Другие вопросы по тегам