为什么即使未调用plt.show(),matplotlib.pyplot仍显示图像?

时间:2020-08-11 08:22:59

标签: python matplotlib

我正在使用scikit-image的分割算法,在某个时候我想将分割后的图像存储到数组对象中,该任务通过执行以下代码来完成:

        fig, ax = plt.subplots(figsize=(14, 14))
        ax.imshow(image, cmap='gray')
        ax.plot(init[:, 1], init[:, 0], '--r', lw=3) # pylint: disable=unsubscriptable-object
        ax.plot(snake[:, 1], snake[:, 0], '-b', lw=3)
        ax.set_xticks([]), ax.set_yticks([])
        ax.axis([0, img.shape[1], img.shape[0], 0])

        fig.canvas.draw()
        
        #Save the segmented image into a numpy array:
        data = np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
        data = data.reshape(fig.canvas.get_width_height()[::-1] + (3,))
        segmentedImage = resize(data, (512, 512))

现在,问题是我不希望显示图像,我只需要将其存储在segmentedImage变量中,而是一直显示。我不能完全弄清楚哪一行特定的代码实际上在显示图像,而我为防止matplotlib所做的任何工作(例如禁用交互模式)似乎都是失败的,因为我不是在jupyter中工作,而是在可视代码中。

非常感谢。

2 个答案:

答案 0 :(得分:0)

将图像创建放入函数中时,Matplotlib不显示图像。我认为,图像将绘制在单元格/执行的末尾。如果将其放入函数中,它将被返回并且不会被绘制出来。

答案 1 :(得分:0)

您是否尝试过使用savefig()方法。它既可以采用面向对象的方法fig.savefig(...),也可以采用Matlab风格的pyplot方法plt.savefig(...)来使用。有关详细信息,请参见下面的链接:

matplotlib.figure.Figure.savefig

matplotlib.pyplot.savefig