将彩虹矩阵转换为 RBG 并另存为 PNG

Question

我正在按照本教程进行操作：https://pyts.readthedocs.io/en/latest/auto_examples/plot_gaf.html 以生成GADF和GASF矩阵。

一切正常。我只需要将 X_gasf[0] png 中包含的结果保存为 png，但我不知道该怎么做。

我已经试过了：

fig = plt.imshow(X_gadf[0], cmap='rainbow', aspect='equal')
fig.axes.get_xaxis().set_visible(False)
fig.axes.get_yaxis().set_visible(False)
plt.axis('off')
plt.savefig(GADF_path + filename, pad_inches=0, bbox_inches='tight')

这段代码有效，但仍然在图像周围给了我一个白色边框（对我来说，没有任何白色边框非常重要）。

我也试过这个：

# Test 1
matplotlib.image.imsave(filename, X_gadf[0])
# Test 2
im = Image.fromarray(X_gadf[0], mode='HSV')
im.save(GADF_path + filename)
# Test 3
cv2.imread(X_gadf[0])
imsave(filename, X_gadf[0])

但这些解决方案都不起作用。看来X_gadf[0]是一种我无法管理的奇怪形象。

有什么想法吗？

Answer 1

你试过这个吗？

matplotlib.image.imsave(filename, X_gasf[0], cmap='rainbow')

查看文档，您似乎忘记了cmap='rainbow'