sumit
4/29/2020 - 8:57 AM

convert plt in np array matplotlib to numpy array

from io import BytesIO

def get_img_from_fig(fig, dpi=180):
    buf = BytesIO()
    fig.savefig(buf, format="png", dpi=dpi)
    buf.seek(0)
    img_arr = np.frombuffer(buf.getvalue(), dtype=np.uint8)
    buf.close()
    img = cv2.imdecode(img_arr, 1)
    # img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)

    return img
    
plt.imshow(temp,cmap = c.jet)
plt.axis('off')
fig = plt.gcf()
cv2.imshow(get_img_from_fig(fig))