PythonのOpenCVをインストールして画像処理を行う

PythonのOpenCVというライブラリをpipenvにインストールして画像処理をするまでを紹介します。

OpenCVというのは画像や動画の処理をするためのライブラリになります。

「pipenvって何?」って人はまずはこちらの記事を参照ください。

【Python】pipenvでpipfile内環境をローカルにインストールする

コンテンツ

今回のゴール

ゴールとしては用意した画像を加工して、輪郭を抽出してラフ画のような感じに表示するまでです。

開発環境を確認する

まずは今回実行する開発環境を確認します。

環境

・OS:Mac

・Python3.9

・Pipenv

OpenCVって何？

「そもそもOpenCVって何？」って話になりますがOpenCVとは、画像や動画の処理をするためのライブラリになります。

OpenCVができることとしては、

画像の輪郭検出
画像に文字や図形の挿入
画像の色彩変換
画像の回転・上下反転・左右反転
モザイク処理
マスク処理
合成処理
…

とにかく、画像の加工に関しては多機能で多くの処理がPythonで行えるライブラリになります。

また画像に限らず動画にもその加工処理ができます。

OpenCVをインストールする

まずはライブラリのインストールから始めます。

% pipenv install opencv-python

1	% pipenv install opencv-python

無事インストールできたかどうか、Pythonの対話モードで実際に確認します。

(※対話モードとは入力したコードをすぐに実行して利用できる機能です。)

% python pipenv shell # pipenvの起動
% python   # pythonの対話モード
>>> import cv2
>>> print(cv2.__version__)
4.5.1

% python pipenv shell # pipenvの起動

% python # pythonの対話モード

>>> import cv2

>>> print(cv2.__version__)

4.5.1

こちらのコマンドがうまくいけば、インストールしたバージョンが4.5.1だということが確認できました。

コードを実装する

インストールもして準備もできたので早速コードを実装します。

画像は相対パス(同じディレクトリ)に用意します。

エヴ◯ンゲリオン初号機の画像を用意します。

こちらの画像のファイル名は、sample.pngとします。

import cv2

img = cv2.imread('sample.jpg', 0) # 画像の読み込み
edge_img = cv2.Canny(img, 50, 100) # 読み込んだ画像の輪郭の検出

cv2.imwrite('edge_sample.png',edge_img) # edge_imge.pngというファイルを生成し書き込む
print('画像処理を実行しました')