删除已保存图像周围的空白

Question

我需要拍摄一张图片并在经过一些处理后保存。当我显示它时，这个图看起来很好，但是在保存图之后，我在保存的图像周围有一些空白。我已经尝试了'tight' 方法的'tight' 选项，也没有工作。代码：

  import matplotlib.image as mpimg
  import matplotlib.pyplot as plt

  fig = plt.figure(1)
  img = mpimg.imread("image.jpg")
  plt.imshow(img)
  ax=fig.add_subplot(1, 1, 1)

  extent = ax.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
  plt.savefig('1.png', bbox_inches=extent)

  plt.axis('off') 
  plt.show()

我正在尝试通过在图形上使用 NetworkX 来绘制基本图形并保存它。我意识到没有图表也可以工作，但是当添加图表时，我会在保存的图像周围出现空白；

import matplotlib.image as mpimg
import matplotlib.pyplot as plt
import networkx as nx

G = nx.Graph()
G.add_node(1)
G.add_node(2)
G.add_node(3)
G.add_edge(1, 3)
G.add_edge(1, 2)
pos = {1:[100, 120], 2:[200, 300], 3:[50, 75]}

fig = plt.figure(1)
img = mpimg.imread("image.jpg")
plt.imshow(img)
ax=fig.add_subplot(1, 1, 1)

nx.draw(G, pos=pos)

extent = ax.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
plt.savefig('1.png', bbox_inches = extent)

plt.axis('off') 
plt.show()

Answer 1

您可以通过在savefig 中设置bbox_inches="tight" 来移除空白填充：

plt.savefig("test.png",bbox_inches='tight')

您必须将bbox_inches 的参数作为字符串，也许这就是为什么它不能早些为您工作的原因。

---

可能的重复：

Matplotlib plots: removing axis, legends and white spaces

How to set the margins for a matplotlib figure?

Reduce left and right margins in matplotlib plot

Answer 2

我不能声称我确切地知道我的“解决方案”为什么起作用或如何起作用，但是当我想将几个机翼截面的轮廓（没有白边）绘制到 PDF 文件时，我必须这样做。 （请注意，我在 IPython 笔记本中使用了 matplotlib，带有 -pylab 标志。）

plt.gca().set_axis_off()
plt.subplots_adjust(top = 1, bottom = 0, right = 1, left = 0, 
            hspace = 0, wspace = 0)
plt.margins(0,0)
plt.gca().xaxis.set_major_locator(plt.NullLocator())
plt.gca().yaxis.set_major_locator(plt.NullLocator())
plt.savefig("filename.pdf", bbox_inches = 'tight',
    pad_inches = 0)

我已尝试停用此功能的不同部分，但这总是会导致某处出现白边。您甚至可以对其进行修改，以防止靠近图形边界的粗线因缺少边距而被剃掉。

Answer 3

在尝试上述答案但没有成功（以及大量其他堆栈帖子）之后，最终对我有用的只是

plt.gca().set_axis_off()
plt.subplots_adjust(top = 1, bottom = 0, right = 1, left = 0, 
            hspace = 0, wspace = 0)
plt.margins(0,0)
plt.savefig("myfig.pdf")

重要的是，这不包括 bbox 或填充参数。

Answer 4

我从 Arvind Pereira (http://robotics.usc.edu/~ampereir/wordpress/?p=626) 那里找到了一些东西，并且似乎对我有用：

plt.savefig(filename, transparent = True, bbox_inches = 'tight', pad_inches = 0)

Answer 5

以下函数包含上面的 johannes-s 答案。我已经用plt.figure 和plt.subplots() 对它进行了多轴测试，效果很好。

def save(filepath, fig=None):
    '''Save the current image with no whitespace
    Example filepath: "myfig.png" or r"C:\myfig.pdf" 
    '''
    import matplotlib.pyplot as plt
    if not fig:
        fig = plt.gcf()

    plt.subplots_adjust(0,0,1,1,0,0)
    for ax in fig.axes:
        ax.axis('off')
        ax.margins(0,0)
        ax.xaxis.set_major_locator(plt.NullLocator())
        ax.yaxis.set_major_locator(plt.NullLocator())
    fig.savefig(filepath, pad_inches = 0, bbox_inches='tight')

Answer 6

我发现以下代码非常适合这项工作。

fig = plt.figure(figsize=[6,6])
ax = fig.add_subplot(111)
ax.imshow(data)
ax.axes.get_xaxis().set_visible(False)
ax.axes.get_yaxis().set_visible(False)
ax.set_frame_on(False)
plt.savefig('data.png', dpi=400, bbox_inches='tight',pad_inches=0)

Answer 7

我按照这个顺序，它就像一个魅力。

plt.axis("off")
fig=plt.imshow(image array,interpolation='nearest')
fig.axes.get_xaxis().set_visible(False)
fig.axes.get_yaxis().set_visible(False)
plt.savefig('destination_path.pdf',
    bbox_inches='tight', pad_inches=0, format='pdf', dpi=1200)

Answer 8

这对我有用 plt.savefig(save_path,bbox_inches='tight', pad_inches=0, transparent=True)

Answer 9

最直接的方法是使用plt.tight_layout 转换，这实际上更可取，因为它在使用plt.savefig 时不会进行不必要的裁剪

import matplotlib as plt    
plt.plot([1,2,3], [1,2,3])
plt.tight_layout(pad=0)
plt.savefig('plot.png')

但是，这对于修改图形的复杂绘图可能并不可取。如果是这种情况，请参阅使用plt.subplots_adjust 的Johannes S's answer。

Answer 10

我发现一个更简单的方法是使用plt.imsave：

    import matplotlib.pyplot as plt
    arr = plt.imread(path)
    plt.imsave('test.png', arr)

Answer 11

对于任何想要以像素而不是英寸工作的人来说，这都可以。

加上平时你也需要的

from matplotlib.transforms import Bbox

然后你可以使用以下内容：

my_dpi = 100 # Good default - doesn't really matter

# Size of output in pixels
h = 224
w = 224

fig, ax = plt.subplots(1, figsize=(w/my_dpi, h/my_dpi), dpi=my_dpi)

ax.set_position([0, 0, 1, 1]) # Critical!

# Do some stuff
ax.imshow(img)
ax.imshow(heatmap) # 4-channel RGBA
ax.plot([50, 100, 150], [50, 100, 150], color="red")

ax.axis("off")

fig.savefig("saved_img.png",
            bbox_inches=Bbox([[0, 0], [w/my_dpi, h/my_dpi]]),
            dpi=my_dpi)

Answer 12

因此解决方案取决于您是否调整子图。如果您指定 plt.subplots_adjust (top, bottom, right, left)，您不想将 bbox_inches='tight' 的 kwargs 与 plt.savefig 一起使用，因为它自相矛盾地创建了空白填充。它还允许您将图像保存为与输入图像相同的暗度（600x600 输入图像另存为 600x600 像素输出图像）。

如果您不关心输出图像大小的一致性，您可以省略 plt.subplots_adjust 属性，只需将 bbox_inches='tight' 和 pad_inches=0 kwargs 与 plt.savefig 一起使用。

此解决方案适用于 matplotlib 版本 3.0.1、3.0.3 和 3.2.1。当您有超过 1 个子图（例如 plt.subplots(2,2,...)）时，它也可以工作。

def save_inp_as_output(_img, c_name, dpi=100):
    h, w, _ = _img.shape
    fig, axes = plt.subplots(figsize=(h/dpi, w/dpi))
    fig.subplots_adjust(top=1.0, bottom=0, right=1.0, left=0, hspace=0, wspace=0) 
    axes.imshow(_img)
    axes.axis('off')
    plt.savefig(c_name, dpi=dpi, format='jpeg') 

Answer 13

你可以试试这个。它解决了我的问题。

import matplotlib.image as mpimg
img = mpimg.imread("src.png")
mpimg.imsave("out.png", img, cmap=cmap)

Answer 14

这适用于我将使用 imshow 绘制的 numpy 数组保存到文件

import matplotlib.pyplot as plt

fig = plt.figure(figsize=(10,10))
plt.imshow(img) # your image here
plt.axis("off")
plt.subplots_adjust(top = 1, bottom = 0, right = 1, left = 0, 
        hspace = 0, wspace = 0)
plt.savefig("example2.png", box_inches='tight', dpi=100)
plt.show()