2022年4月27日水曜日

PySimpleGUIとOpenCVでwebカメラ画像の輪郭線を描画する

OpenCVで輪郭線を描画するには画像を以下の順序で処理します。
1. threshold関数などで画像を二値化します
2. findContours関数で輪郭線検出を行う
3. drawContours関数で輪郭線を描画

サンプルコードの実行手順

1. PySimpleGUIとOpenCVがインストールされた環境の構築
以下のページを参照して、環境を構築します。
PySimpleGUIとOpenCVをインストールしてwebカメラの映像をウインドウを表示する

2. サンプルプログラムの作成と実行
以下のファイルを保存して、実行します。
psgui_opencv_contours.py
import PySimpleGUI as sg
import cv2

sg.theme('SystemDefault')
layout = [
  [sg.Slider(key='slider', range=(0, 255), default_value=128, orientation='horizontal')],
  [sg.Image(key='img1'), sg.Image(key='img2')]
]

# webカメラをキャプチャー
capture = cv2.VideoCapture(0)

# webカメラの解像度を取得
width = int(capture.get(cv2.CAP_PROP_FRAME_WIDTH)/2)
height = int(capture.get(cv2.CAP_PROP_FRAME_HEIGHT)/2)
window = sg.Window("webカメラ画面", layout=layout, finalize=True)
# イベントループ
while True:
  event, values = window.read(timeout=50)
  print(event, values)
  if event == sg.WIN_CLOSED:
    break
  rv, frame = capture.read()
  if rv is True:
    # 左右に並べるために縦横のサイズを半分にリサイズ
    resized = cv2.resize(frame, (width, height))
    # グレースケールに変換
    gray = cv2.cvtColor(resized, cv2.COLOR_BGR2GRAY)
    # 2値化
    slider = values['slider']
    threshold, binary = cv2.threshold(gray, slider, 255, cv2.THRESH_BINARY)
    # 輪郭抽出
    (contours, hierarchy) = cv2.findContours(binary,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)
    # 輪郭描画
    cnt = resized.copy()
    cnt = cv2.drawContours(resized, contours, -1, (0,192,0), 2)

    # pngに変換して、Image更新
    img = cv2.imencode('.png', gray)[1].tobytes()
    img2 = cv2.imencode('.png', cnt)[1].tobytes()
    window['img1'].update(data=img)
    window['img2'].update(data=img2)

capture.release()
window.close()

・実行方法
以下のコマンドを実行します。
python3 psgui_opencv_contours.py

関連情報

PySimpleGUIで画像を表示する

・OpenCVに関する他の記事はこちらを参照してください。

0 件のコメント:

コメントを投稿