我正在执行将心音转换为图以馈送卷积神经网络的过程。我使用matplotlib.pyplot.plot绘制了心音图,但我不知道如何获取图像,有人将其保存使用matplotlib.pyplot.savefig作为.jgp或.png,然后通过cv2.imread()获得图像矩阵。我的问题是我是否可以不使用matplotlib.pyplt.savefig获得图像矩阵,而是直接获得图像矩阵。
import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(-3,3,50)
y1 = 0.1*x
y2 = x**2
plt.figure()
plt.plot(x,y1,linewidth = 10,zorder = 2,label = r'$y_1\ =\ 0.1*x$')
plt.plot(x,y2,linewidth = 10,zorder = 1,label = r'$y_2\ =\ x^{2}$')
plt.legend(loc = 'lower right')
plt.show()
plt.savefig("image path")