OpenCVで飛行機を検出する初心者向け手法を解説します

PythonのOpenCVを使って動画から映像を取得し、飛んでいる飛行機を検知します。

今回はピクセルを比較する方法で動体検知します。
とてもシンプルな手法ですが、飛んでいる飛行機を検知する上では意外と正確に検知出来たので記事にしました。

結果

※使用させていただいた動画のリンクは記事の最後に記載しました。

前提条件

python3.9
opencv-pytho 4.5.5

仕組み概要

1.動画を読み込む。
2.グレースケールに変換
3.比較用にフレームを保存しておく
4.ガウシアンブラーでノイズを軽減
5.画像を比較して差分の画像を生成する
6.差分の画像を２値化する（閾値処理）
7.輪郭を検知する
8.輪郭がはいった配列から、領域が小さすぎる要素は無視する
9.輪郭情報から短形で領域を囲み、描画する

コード

import cv2

filepath = "sample.mov"
cap = cv2.VideoCapture(filepath)
avg = None

while True:
    # 1フレームずつ取得する。
    ret, frame = cap.read()
    if not ret:
        break
    # グレースケールに変換
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    # 比較用のフレームを取得する
    if avg is None:
        avg = gray.copy().astype("float")
        continue

    # ブラーを掛けてノイズを軽減する
    blur = cv2.GaussianBlur(gray, (1, 1), 1)
    # 現在のフレームと移動平均との差を計算
    cv2.accumulateWeighted(blur, avg, 0.7)
    frameDelta = cv2.absdiff(blur, cv2.convertScaleAbs(avg))
    # デルタ画像を閾値処理を行う
    thresh = cv2.threshold(frameDelta, 3, 255, cv2.THRESH_BINARY)[1]
    # 画像の閾値に輪郭線を入れる
    contours, hierarchy = cv2.findContours(thresh.copy(), cv2.RETR_LIST, cv2.CHAIN_APPROX_NONE)

    for i in range(0, len(contours)):
        if len(contours[i]) > 0:
            # しきい値より小さい領域は無視する
            if cv2.contourArea(contours[i]) < 100:
                continue
            # 短形で領域を囲む
            rect = contours[i]
            x, y, w, h = cv2.boundingRect(rect)
            cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 4)
    # 結果を出力
    cv2.imshow("Frame", frame)
    # Escキーで終わる
    key = cv2.waitKey(30)
    if key == 27:
        break
cap.release()
cv2.destroyAllWindows()

コード解説

gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

グレースケールに変換します。
OpenCVでは画像のチャンネルの並び順はRGBではなくBGRです。

blur = cv2.GaussianBlur(gray, (1, 1), 1)

ブラーを掛けてノイズを軽減します。
(1,1),1がブラーの強さになりますが、大きくする（大きくぼかす）と輪郭の情報もボケてしまって検出しにくくなります。

cv2.accumulateWeighted(blur, avg, 0.7)

2つの画像blur,avgの移動した値を求めます。0.7は入力画像の重みです。
重みは更新速度（どのくらいの早さで，以前の画像を「忘れる」か）を調節します。*参照 opencv.jp
重みを調整することでより正確に検出できることがあります。

contours, hierarchy = cv2.findContours(thresh.copy(), cv2.RETR_LIST, cv2.CHAIN_APPROX_NONE)

輪郭を検知します。
contoursにはオブジェクトの輪郭座標を保持している配列。
hierarchyにはオブジェクトの階層構造情報を保持している配列。
が入ります。

for i in range(0, len(contours)):
        if len(contours[i]) > 0:
            # しきい値より小さい領域は無視する
            if cv2.contourArea(contours[i]) < 100:
                continue
            # 短形で領域を囲む
            rect = contours[i]
            x, y, w, h = cv2.boundingRect(rect)
            cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 4)

forで回して順番に処理します。
cv2.contourAreaは領域の大きさを返します。しきい値100より小さいなら無視します。
cv2.boundingRect(rect)で短形の領域を計算し、
cv2.rectangleで描画します。
(0, 255, 0)は色（RGB)で、4は線の太さです。

cv2.imshow("Frame", frame) でウインドウに描画します。