Tensorflow + Matplotlib 动画
Tensorflow + Matplotlib animation
我正在尝试使用 tensorflow 实现生命游戏,并使用 matplotlib.animation 来描绘动画。尽管显示了图像,但由于某种原因它没有动画。下面是我使用的代码:
参考:http://learningtensorflow.com/lesson8/
IDE:Pycharm 社区版
我在 FuncAnimation 中传递了 blit=False,因为它在 Mac 和 blit=True
上不起作用
shape = (50, 50)
initial_board = tf.random_uniform(shape, minval=0, maxval=2, dtype=tf.int32)
board = tf.placeholder(tf.int32, shape=shape, name='board')
def update_board(X):
# Check out the details at: https://jakevdp.github.io/blog/2013/08/07/conways-game-of-life/
# Compute number of neighbours,
N = convolve2d(X, np.ones((3, 3)), mode='same', boundary='wrap') - X
# Apply rules of the game
X = (N == 3) | (X & (N == 2))
return X
board_update = tf.py_func(update_board, [board], [tf.int32])
fig = plt.figure()
if __name__ == '__main__':
with tf.Session() as sess:
initial_board_values = sess.run(initial_board)
X = sess.run(board_update, feed_dict={board: initial_board_values})[0]
def game_of_life(*args):
A = sess.run(board_update, feed_dict={board: X})[0]
plot.set_array(A)
return plot,
ani = animation.FuncAnimation(fig, game_of_life, interval=100, blit=False)
plot = plt.imshow(X, cmap='Greys', interpolation='nearest')
plt.show()
当您显示图像时,您显示的是静态图像而不是动画。您需要删除这一行,如教程中所述:
plot = plt.imshow(X, cmap='Greys', interpolation='nearest')
Hint: you will need to remove the plt.show() from the earlier code to make this run!
希望对您有所帮助!
我正在尝试使用 tensorflow 实现生命游戏,并使用 matplotlib.animation 来描绘动画。尽管显示了图像,但由于某种原因它没有动画。下面是我使用的代码:
参考:http://learningtensorflow.com/lesson8/
IDE:Pycharm 社区版
我在 FuncAnimation 中传递了 blit=False,因为它在 Mac 和 blit=True
上不起作用shape = (50, 50)
initial_board = tf.random_uniform(shape, minval=0, maxval=2, dtype=tf.int32)
board = tf.placeholder(tf.int32, shape=shape, name='board')
def update_board(X):
# Check out the details at: https://jakevdp.github.io/blog/2013/08/07/conways-game-of-life/
# Compute number of neighbours,
N = convolve2d(X, np.ones((3, 3)), mode='same', boundary='wrap') - X
# Apply rules of the game
X = (N == 3) | (X & (N == 2))
return X
board_update = tf.py_func(update_board, [board], [tf.int32])
fig = plt.figure()
if __name__ == '__main__':
with tf.Session() as sess:
initial_board_values = sess.run(initial_board)
X = sess.run(board_update, feed_dict={board: initial_board_values})[0]
def game_of_life(*args):
A = sess.run(board_update, feed_dict={board: X})[0]
plot.set_array(A)
return plot,
ani = animation.FuncAnimation(fig, game_of_life, interval=100, blit=False)
plot = plt.imshow(X, cmap='Greys', interpolation='nearest')
plt.show()
当您显示图像时,您显示的是静态图像而不是动画。您需要删除这一行,如教程中所述:
plot = plt.imshow(X, cmap='Greys', interpolation='nearest')
Hint: you will need to remove the plt.show() from the earlier code to make this run!
希望对您有所帮助!