Pythonで画像を読み込み、表示し、保存する（OpenCV編）

PythonのOpenCV（cv2）モジュールで、画像ファイルの読み込みと、表示と、保存をしてみます。

OpenCV-Pythonのインストール

OpenCVといえば有名な画像処理ライブラリですが、 Pythonでもpipで利用できるようになりました。

> pip install opencv-python

このモジュールをインポートするときはcv2という名前でインポートします。

>>> import cv2

画像の読み込みと、表示と、保存をしてみます。

imreadというメソッドで画像ファイルの読み込みをします。

retval = cv2.imread(filename, [flags])

変数	型	内容
filename	str	画像ファイルのパス。
flags		省略可。既定値はIMREAD_COLOR。読み込み方法の指定。
retval	ndarray	画像データ。

flagsに読み込むときの方法を規定します。下表の他にもありますので、公式を参照してください。

読み込んだデータそのものはndarrayになります。

imshowメソッドで新しいウィンドウを開いて表示するのですが、imshowだけではウィンドウが開くだけで、一連の手続きが必要になります。

cv2.imshow(winname, mat)

変数	型	内容
winname	str	ウインドウ名。
mat	ndarray	画像データ。

cv2.waitKey([delay])

変数	型	内容
delay	int	省略可。既定値は0。キー入力待ち時間をミリ秒単位で指定する。

cv2.destroyAllWindows()

cv2モジュールで開いたウィンドウを全て閉じるメソッドです。

試してみましょう。プロ生ちゃんの画像をimreadで読み込んでimshowで表示してみます。

import cv2

im = cv2.imread('original.png')
cv2.imshow('test', im)
cv2.waitKey()
cv2.destroyAllWindows()

print(im.shape)

コンソールへの出力はこうなります。

(256, 256, 3)

カラーっぽい行列になってますね。

では同じ画像ファイルをグレースケールモードで読み込んでみましょう。

import cv2

im = cv2.imread('original.png', cv2.IMREAD_GRAYSCALE)
cv2.imshow('test', im)
cv2.waitKey()
cv2.destroyAllWindows()

print(im.shape)

コンソールへの出力はこうなります。

(256, 256)

各画素１バイトのグレースケールっぽい行列になってますね。

imwriteメソッドで画像をファイルに出力します。

cv2.imwrite(filename, img, [param])

paramには、画像フォーマット毎に特有なオプションのパラメータを指定します。詳しくは公式のリストを参照してください。

では、先ほどのプロ生ちゃんのカラー画像をグレースケールモードで読み込んで、そのままPNGファイルとして出力してみましょう。

import cv2

im = cv2.imread('original.png', cv2.IMREAD_GRAYSCALE)
cv2.imwrite('gray.png', im)

画像ファイルのプロパティを確認すると、確かに各画素8ビットのグレースケールで保存されていました。

公開日 2018-05-20