只在一个窗口中显示更新图像

时间:2017-03-13 14:38:55

标签: python image window neural-network updating

我想在窗口中显示jpg,每秒更新多次。 我编写了一个非常紧凑的程序,只有100行代码(一个创建图像的神经网络),并且不想再显示另外100行代码来显示图像。 有什么办法可以解决这个问题吗?

很多thx,jj

1 个答案:

答案 0 :(得分:3)

正如评论中所述,IO不是问题,我们将直接使用matplotlib中使用的可用标准图像绘图工具,因为它是python的事实标准绘图库。虽然不知道源自神经网络的典型图像的尺寸,但需要快速比较调用例如神经网络所需的平均时间。对于不同的图片尺寸,imshowpcolormeshmatshow不会受到影响(pcolor明显较慢,因此省略)。

import matplotlib.pyplot as plt
import numpy as np
import timeit

n = 13
repeats = 20
timetable = np.zeros((4, n-1))
labellist = ['imshow', 'matshow', 'pcolormesh']

for i in range(1, n):
    image = np.random.rand(2**i, 2**i)
    print('image size:', 2**i)
    timetable[0, i - 1] = 2**i

    timetable[1, i - 1] = timeit.timeit("plt.imshow(image)", setup="from __main__ import plt, image", number=repeats)/repeats
    plt.close('all')
    timetable[2, i - 1] = timeit.timeit("plt.matshow(image)", setup="from __main__ import plt, image", number=repeats)/repeats
    plt.close('all')
    timetable[3, i - 1] = timeit.timeit("plt.pcolormesh(image)", setup="from __main__ import plt, image", number=repeats)/repeats
    plt.close('all')

for i in range(1, 4):
    plt.semilogy(timetable[0, :], timetable[i, :], label=labellist[i - 1])
    plt.legend()
    plt.xlabel('image size')
    plt.ylabel('avg. exec. time [s]')
    plt.ylim(1e-3, 1)

plt.show()

enter image description here

所以,imshow就是这样。在matplotlib中更新或动画图的优雅方法是它提供的动画框架。这样就不需要像许多代码那样烦恼了。这是一个简单的例子:

import matplotlib.pyplot as plt
import numpy as np
import time
from matplotlib import animation

data = np.random.rand(128, 128)

fig = plt.figure()
ax = fig.add_subplot(1,1,1)

im = ax.imshow(data, animated=True)

def update_image(i):
    data = np.random.rand(128, 128)
    im.set_array(data)
    # time.sleep(.5)
    # plt.pause(0.5)
ani = animation.FuncAnimation(fig, update_image, interval=0)

plt.show()

在此示例中,将从更新函数中调用神经网络。繁重的计算工作下的更新行为可以由time.sleep模拟。如果您的应用程序是多线程的plt.pause可能会派上用场,让其他线程有时间完成他们的工作。 interval=0基本上使得情节尽可能频繁地更新。

我希望这能指出你的大方向并且有所帮助。如果您不想使用动画,则需要手动处理画布清除和/或blitting。