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!

希望对您有所帮助!