グレースケール化のためのPIL形式への変換

Question

ディープラーニングを行うために画像のグレースケール化を行おうとしているのですが，現在自分の持っている画像では，添付されているコードで実行することができませんでした．
調べたところ，PIL 画像オブジェクトを numpy 配列に変換することで，グレースケール化しているようでした．
そこで，現在手元にある画像をPIL形式に変換したいと考えています．
以下にグレースケール化のコードを示します．

import numpy as np
from PIL import Image


def _remove_colormap(filename):
  return np.array(Image.open(filename))


def _save_annotation(annotation, filename):
  pil_image = Image.fromarray(annotation.astype(dtype=np.uint8))
  pil_image.save(filename)


def main():
  raw_annotation = _remove_colormap('2007_000032.png')
  _save_annotation(raw_annotation, '2007_000032_output.png')


if __name__ == '__main__':
  main()

Answer 1

改めて調べたところ，インデックスカラーに変換することでグレースケール化に成功しました．

Answer 2

以下のようなコードでグレースケール化できると思います。

from PIL import Image

img = Image.open('test.png')
gray = img.convert('L')
gray.save('test-gray.png')

Comments

[barisuton3]

ご回答ありがとうございます