Конвертировать матрицу Rainbow в RBG и сохранить как PNG - PullRequest
0 голосов
/ 10 января 2019

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

Есть идеи?

1 Ответ

0 голосов
/ 11 января 2019

Вы пробовали это?

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

Глядя на документацию, кажется, вы забыли cmap='rainbow'

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...