【Python/OpenCV】膨張・収縮フィルタ処理(dilate, erode)

この記事では、Python版OpenCV(cv2.dilate, cv2.erode)もしくはNumPyで膨張・収縮フィルタ処理を実装する方法をソースコード付きで解説します。

膨張・収縮フィルタ

膨張・収縮フィルタとは、二値画像からノイズ除去したりするのに使われるアルゴリズムです。

膨張処理・・・白色(255)の領域を膨張
収縮処理・・・白色(255)の領域を収縮

アルゴリズムの原理については下記事で解説しています。
【参考】膨張・収縮処理の原理・特徴・計算式

今回は、以下の2通りの方法で処理を実装してみました。

方法①・・・NumPyで膨張・収縮フィルタのアルゴリズムを書いて実装
方法②・・・cv2.Cannyで簡単に実装(1行で書けちゃうので楽チン)

書式

dst = cv2.dilate(src, kernel, iterations) # 膨張処理
dst = cv2.erode(src, kernel, iterations) # 収縮処理
パラメータ名 説明
src 入力画像
kernel カーネル(1の近傍画素を膨張・収縮処理に利用)
dst 出力画像

ソースコード(Python3+OpenCV3)

サンプルプログラムのソースコードです。

方法①

#-*- coding:utf-8 -*-
import cv2
import numpy as np

# 膨張処理
def dilate(src, ksize=3):
    h, w = src.shape
    dst = src.copy()
    d = int((ksize-1)/2)

    for y in range(0, h):
        for x in range(0, w):
            if np.count_nonzero(src[y-d:y+d+1, x-d:x+d+1]) > 0:
                dst[y][x] = 255

    return dst

# 収縮処理
def erode(src, ksize=3):

    h, w = src.shape
    dst = src.copy()
    d = int((ksize-1)/2)

    for y in range(0, h):
        for x in range(0, w):
            if np.count_nonzero(src[y-d:y+d+1, x-d:x+d+1]) < ksize**2:
                dst[y][x] = 0

    return dst

def main():
    # 入力画像を読み込み
    img = cv2.imread("input.jpg")

    # グレースケール変換
    gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)

    # 二値化処理
    gray[gray<127] = 0
    gray[gray>=127] = 255

    # 膨張・収縮処理(方法1)
    dilate1 = dilate(gray, ksize=3)
    erode1 = erode(dilate1, ksize=3)

    # 結果を出力
    cv2.imwrite("dilate1.jpg", dilate1)
    cv2.imwrite("erode1.jpg", erode1)


if __name__ == "__main__":
    main()

方法②

#-*- coding:utf-8 -*-
import cv2
import numpy as np

def main():
    # 入力画像を読み込み
    img = cv2.imread("input.jpg")

    # グレースケール変換
    gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)

    # 二値化処理
    gray[gray<127] = 0
    gray[gray>=127] = 255

    # 膨張・収縮処理(方法2)
    # 8近傍で処理
    kernel = np.array([[1, 1, 1],
                       [1, 1, 1],
                       [1, 1, 1]], np.uint8)

    dilate2 = cv2.dilate(gray, kernel)
    erode2 = cv2.erode(dilate2, kernel)

    # 結果を出力
    cv2.imwrite("dilate2.jpg", dilate2)
    cv2.imwrite("erode2.jpg", erode2)


if __name__ == "__main__":
    main()

※動作には、OpenCVライブラリのインストールが必要です。

実行結果

サンプルプログラムの実行結果です。

■入力画像(左)→膨張画像(中央)→収縮画像(右)

お借りした画像:プロ生ちゃん(暮井 慧)

【おすすめ記事】
PythonでOpenCV入門 サンプル集
【Python】画像処理プログラミング入門
【画像処理入門】アルゴリズム&プログラミング

シェア&フォローお願いします!