opencvによるエッジ抽出のサンプル

opencv-prewitt-sample.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import cv2
import numpy as np


def main():
    gray = cv2.imread('lena_std.png', 0)  # 入力画像をグレースケールで読み込み
    # gray = np.asarray([[0, 0, 1, 1], [0, 0, 1, 7], [1, 0, 8, 8], [0, 1, 8, 9]], dtype=np.uint8)

    # カーネル（縦方向の輪郭検出用）
    kernel = np.array([[-1, 0, 1],
                       [-1, 0, 1],
                       [-1, 0, 1]])

    dst1 = cv2.filter2D(gray, -1, kernel)
    # print(dst1)

    # Output
    cv2.imwrite('lena_prewitt.png', dst1)


if __name__ == '__main__':
    main()