memef4rmer/edit_anything

1import numpy as np2import matplotlib.pyplot as plt3 4 5def mpl_fig_to_buffer(fig):6    fig.canvas.draw()7    plot = np.array(fig.canvas.renderer.buffer_rgba())8    plt.close(fig)9    return plot10