Как преобразовать фигуру matplotlib в объект изображения PIL (без сохранения изображения) ⇐ Python

[Anonymous]

Как говорится в заголовке, я пытаюсь преобразовать рис в pil.image . В настоящее время я могу сделать это, сначала сохранив рис < /code> на диск, а затем открыв этот файл, используя image.open () < /code>, но процесс занимает больше времени, чем ожидалось, и я надеюсь, что, пропустив сохранение локального шага, это будет немного быстрее. < /P>

Здесь я будет немного быстрее.# build fig
figsize, dpi = self._calc_fig_size_res(img_height)
fig = plt.Figure(figsize=figsize)
canvas = FigureCanvas(fig)
ax = fig.add_subplot(111)
ax.imshow(torch.from_numpy(S).flip(0), cmap = cmap)
fig.subplots_adjust(left = 0, right = 1, bottom = 0, top = 1)
ax.axis('tight'); ax.axis('off')

# export
fig.savefig(export_path, dpi = dpi)

# open image as PIL object
img = Image.open(export_path)
< /code>

Я попытался сделать это после того, как я построил рис (он будет прямо перед стадией экспорта): < /p>

pil_img = Image.frombytes('RGB', canvas.get_width_height(), canvas.tostring_rgb())
< /code>

Но он не показывает все изображение. Похоже, это урожай верхнего левого угла, но это может быть просто странное представление данных - я работаю со спектрограммами, чтобы изображения довольно абстрактные.

Подробнее здесь: https://stackoverflow.com/questions/573 ... ving-image