Какой cmap (цветовая карта) использовать с matplotlib imshow() для отображения изображений набора данных MNIST в виде черных символов на белом фоне

Исходные изображения MNIST представляют собой белый фон с символами в градациях серого. 0 соответствует белому цвету, а 255 — черному, между ними находятся оттенки серого. Я использую копию набора данных, предоставленного keras, в том же формате.

Просто чтобы посмотреть на вещи, я использую matplotlib для отображения примеров из набора данных, но когда я выбираю cmap 'gray', я получаю черный фон с белыми символами, как показано ниже.

  import matplotlib.pyplot as plt
  plt.imshow(X_train[0], cmap='gray')

введите здесь описание изображения

Есть ли другая цветовая карта, которая будет правильно отображать изображения черным цветом на белом фоне?


python matplotlib mnist colormap
person Ray    schedule 15.04.2016    source источник

Ответы (1)


arrow_upward
5
arrow_downward

Вы можете либо инвертировать цветовую карту (используя версию *_r вашей палитры) или инвертировать (отменить) ваши данные.

# Invert the colormap
plt.imshow(X_train[0], cmap='gray_r')

# Invert your data
plt.imshow(-X_train[0], cmap='gray')
person Suever    schedule 15.04.2016