Menghapus ruang putih di sekitar gambar yang disimpan di matplotlib

Saya perlu mengambil gambar dan menyimpannya setelah beberapa proses. Angka itu terlihat baik-baik saja ketika saya menampilkannya, tetapi setelah menyimpan gambar itu, saya mendapat ruang putih di sekitar gambar yang disimpan. Saya telah mencoba 'tight'opsi untuk savefigmetode, tidak berhasil. Kode:

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

  fig = plt.figure(1)
  img = mpimg.imread(path)
  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()

Saya mencoba menggambar grafik dasar dengan menggunakan NetworkX pada gambar dan menyimpannya. Saya menyadari bahwa tanpa grafik itu berfungsi, tetapi ketika menambahkan grafik saya mendapatkan ruang putih di sekitar gambar yang disimpan;

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("C:\\images\\1.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()

Saya tidak dapat mengklaim bahwa saya tahu persis mengapa atau bagaimana "solusi" saya bekerja, tetapi inilah yang harus saya lakukan ketika saya ingin merencanakan garis besar beberapa bagian aerofoil - tanpa margin putih - ke file PDF. (Perhatikan bahwa saya menggunakan matplotlib di dalam notebook IPython, dengan flag -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)

Saya telah mencoba untuk menonaktifkan bagian yang berbeda dari ini, tetapi ini selalu mengarah pada margin putih di suatu tempat. Anda bahkan mungkin telah memodifikasi ini untuk menjaga garis-garis lemak di dekat batas gambar agar tidak dicukur oleh kurangnya margin.

Anda dapat menghapus padding ruang putih dengan menetapkan bbox_inches="tight"di savefig:

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

Anda harus meletakkan argumen bbox_inchessebagai string, mungkin ini sebabnya tidak bekerja lebih awal untuk Anda.

Kemungkinan duplikat:

Plot matplotlib: menghilangkan sumbu, legenda dan spasi putih

Bagaimana cara mengatur margin untuk angka matplotlib?

Kurangi margin kiri dan kanan di plot matplotlib

Setelah mencoba jawaban di atas tanpa hasil (dan banyak posting stack lainnya) apa yang akhirnya berhasil untuk saya adalah adil

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")

Yang penting ini tidak termasuk argumen bbox atau padding.

Saya menemukan sesuatu dari Arvind Pereira ( http://robotics.usc.edu/~ampereir/wordpress/?p=626 ) dan sepertinya bekerja untuk saya:

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

Fungsi berikut menggabungkan jawaban johannes di atas. Saya telah mengujinya dengan plt.figuredan plt.subplots()dengan beberapa sumbu, dan bekerja dengan baik.

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')

Saya menemukan kode-kode berikut berfungsi dengan baik untuk pekerjaan itu.

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)

Saya mengikuti urutan ini dan itu bekerja seperti pesona.

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)

Bagi siapa pun yang ingin bekerja dalam piksel daripada inci, ini akan berfungsi.

Plus yang biasa Anda juga perlu

from matplotlib.transforms import Bbox

Kemudian Anda dapat menggunakan yang berikut ini:

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)

Pendekatan yang jauh lebih sederhana yang saya temukan adalah menggunakan plt.imsave:

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

Anda dapat mencoba ini. Itu memecahkan masalah saya.

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

Jika Anda ingin menampilkan apa yang harus disimpan, saya sarankan menggunakan plt.tight_layouttransformasi yang sebenarnya lebih disukai karena tidak melakukan pemotongan yang tidak perlu saat menggunakanplt.savefig

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

Ini berfungsi untuk saya menyimpan array numpy diplot dengan imshow ke file

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()