Function fig2img

utils.py:59–65

(fig)

def fig2img(fig):
    import io
    buf = io.BytesIO()
    fig.savefig(buf, bbox_inches='tight')
    buf.seek(0)
    img = Image.open(buf)
    return img


def show_img(matrix, title):