scipy: savefig без фреймів, осей, лише вміст

У numpy / scipy у мене є зображення, що зберігається в масиві. Я можу показати його, я хочу зберегти його, використовуючи savefig без будь-яких меж, осей, міток, заголовків, ... Просто чисте зображення, нічого іншого.

Я хочу уникати таких пакунків, як PyPNGабо scipy.misc.imsave, вони часом бувають проблематичними (вони не завжди добре встановлюються, savefig()для мене лише базові

РЕДАГУВАТИ

Змінено aspect='normalна, aspect='auto'оскільки це змінилося в останніх версіях matplotlib (завдяки @ Luke19).

Припускаючи:

import matplotlib.pyplot as plt

Щоб зробити фігуру без рамки:

fig = plt.figure(frameon=False)
fig.set_size_inches(w,h)

Щоб зміст заповнив всю фігуру

ax = plt.Axes(fig, [0., 0., 1., 1.])
ax.set_axis_off()
fig.add_axes(ax)

Потім намалюйте на ньому своє зображення:

ax.imshow(your_image, aspect='auto')
fig.savefig(fname, dpi)

aspectПараметр змінює розмір пікселя , щоб переконатися , що вони заповнюють розмір фігури , зазначений в fig.set_size_inches(…). Щоб відчути, як грати з подібними речами, прочитайте документацію matplotlib , особливо на тему Сокири, Вісь та Художник.

Більш простим рішенням є:

fig.savefig('out.png', bbox_inches='tight', pad_inches=0)

Ви можете знайти bbox зображення всередині осі (за допомогою get_window_extent) і використати bbox_inchesпараметр, щоб зберегти лише ту частину зображення:

import numpy as np
import matplotlib.pyplot as plt

data=np.arange(9).reshape((3,3))
fig=plt.figure()
ax=fig.add_subplot(1,1,1)
plt.axis('off')
plt.imshow(data)

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

Я дізнався цей фокус від Джо Кінгтона тут .

Я перепробував кілька варіантів у своєму випадку, і найкращим рішенням було таке:

fig.subplots_adjust(bottom = 0)
fig.subplots_adjust(top = 1)
fig.subplots_adjust(right = 1)
fig.subplots_adjust(left = 0)

потім збережіть свою фігуру за допомогою savefig

Я запропоную відповідь heron13 з невеликим доповненням, запозиченим тут, щоб видалити відступ, що залишився після встановлення bbox у жорсткий режим, тому:

axes = fig.axes()
axes.get_xaxis().set_visible(False)
axes.get_yaxis().set_visible(False)
fig.savefig('out.png', bbox_inches='tight', pad_inches=0)

Цей працює для мене

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

У мене була та ж проблема під час деякої візуалізації за допомогою librosa, де я хотів витягти вміст сюжету без будь-якої іншої інформації. Тож це мій підхід. відповідь unutbu також допомагає мені змусити працювати.

    figure = plt.figure(figsize=(500, 600), dpi=1)
    axis = plt.subplot(1, 1, 1)
    plt.axis('off')
    plt.tick_params(axis='both', left='off', top='off', right='off', bottom='off', labelleft='off', labeltop='off',
                    labelright='off', labelbottom='off')

     # your code goes here. e.g: I used librosa function to draw a image
    result = np.array(clip.feature_list['fft'].get_logamplitude()[0:2])
    librosa.display.specshow(result, sr=api.Clip.RATE, x_axis='time', y_axis='mel', cmap='RdBu_r')


    extent = axis.get_window_extent().transformed(figure.dpi_scale_trans.inverted())
    plt.savefig((clip.filename + str("_.jpg")), format='jpg', bbox_inches=extent, pad_inches=0)
    plt.close()

Для тих, хто намагається зробити це в Юпітері

 plt.axis('off')

 spec = plt.imshow

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

Хоча наведені вище відповіді стосуються видалення полів та відступів, вони не спрацювали для мене при видаленні міток. Ось що вдалося для тих, хто пізніше натрапить на це питання:

Якщо припустити, що вам потрібна сітка розмірів 2х2 із чотирьох зображень, що зберігаються в images:

matplotlib.pyplot.figure(figsize = (16,12)) # or whatever image size you require
for i in range(4):
    ax = matplotlib.pyplot.subplot(2,2,i+1)
    ax.axis('off')
    imshow(images[i])
matplotlib.pyplot.savefig(path, bbox_inches='tight')

Я намагався позбутися кордону теж, користуючись порадами тут, але нічого насправді не спрацювало. Деякі возилися, і я виявив, що зміна фасолі не дала мені кордону в лабораторіях jupyter (будь-який колір призвів до позбавлення від білого кордону). Сподіваюся, це допомагає.

def show_num(data):
data = np.rot90(data.reshape((16,16)), k=3)
data = np.fliplr(data)
fig = plt.figure(frameon=False, facecolor='white')
ax = plt.Axes(fig, [0., 0., 1., 1.])
ax.set_axis_off()
fig.add_axes(ax)
ax.imshow(data)
plt.show()

Насправді я нещодавно пробував це, і замість усіх цих рядків ви можете використовувати

plt.imsave(image_path, image)

Працює як шарм. лише один рядок і проблема вирішена.

imsave()

Документація ( https://matplotlib.org/3.1.1/api/_as_gen/matplotlib.pyplot.imsave.html )

Для мене цей код зробив подібний розмір вхідного зображення без кадру та осей із кодів matehat , unutbu та WHZW :

fig = plt.figure()
ax = fig.add_subplot(1,1,1)
plt.axis('off')
viridis = cm.get_cmap('gist_gray', 256)
plt.imshow(data, aspect='auto', cmap=viridis)
plt.tight_layout()
plt.savefig(out_file, bbox_inches='tight', transparent=True, pad_inches=0)

Середовище виконання:
Python 3.6.10
Matplotlib 3.2.1
OS Windows 10