python - matplotlib.pyplot сохраняет двоичную маску с 4 каналами (?) - PullRequest
0 голосов
/ 27 мая 2018

Я пытаюсь сохранить двоичные маски, т. Е. Массивы со значениями False и True, в виде файлов .png.

Однако при этом с помощью matplotlib.pyplot изображение сохраняется с 4 каналами, и яне знаю почему.

Например, чтобы сохранить изображение:

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm

m = np.array([[False, True],
             [False, True]])

plt.imsave("mask.png", m, cmap=cm.gray)

Затем прочитайте его снова и напечатайте форму:

plt.imread("mask.png").shape

Дает мне:

(2, 2, 4)

Есть ли для этого какая-либо причина и как просто сохранить ее как обычную шкалу серого со значениями 0/1, чтобы форма по существу просто была (2,2)?

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