Преобразуйте матрицу радуги в RBG и сохраните как PNG

Я следую этому руководству: https://pyts.readthedocs.io/en/latest/auto_examples/plot_gaf.html, чтобы сгенерировать матрицу GADF и GASF.

Все работает хорошо. Мне нужно только сохранить в формате png результаты, содержащиеся в X_gasf[0], но я не знаю, как это сделать.

Я уже пробовал это:

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] - это странный тип образа, с которым я не могу справиться.

Любые идеи?