14-10. 画像差分で鮭の遡上を監視してみた

pip install opencv-python
pip install opencv-contrib-python
pip install matplotlib

import cv2
import matplotlib.pyplot as plt

#=============================
# パラメータ
#=============================
#動画
movie_path = '14-17_movie.mp4'

#=============================
# 画像表示関数
#=============================
def show(img):
    plt.figure(figsize=(10, 10))
    plt.imshow(img, vmin = 0, vmax = 255)
    plt.show()
    plt.close()
    print()

#=============================
# 処理
#=============================
#動画読み込み
video = cv2.VideoCapture(movie_path)

#画像差分の準備
fgbg = cv2.bgsegm.createBackgroundSubtractorMOG(history=4)

#繰り返し処理
while(video.isOpened()):
    #フレームを取得
    ret, frame = video.read()
    
    #フレームがある場合
    if ret == True:
        #差分マスクの計算
        fgmask = fgbg.apply(frame)
        
        #輪郭検出
        contours, _ = cv2.findContours(fgmask, cv2.RETR_TREE, cv2.CHAIN_APPROX_NONE)
        
        #最小面積で選別
        contours = [contour for contour in contours if cv2.contourArea(contour) > 200]
        
        #すべての輪郭に矩形を引く
        for contour in contours:
            x, y, w, h = cv2.boundingRect(contour)
            cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 0, 255), 2)
        
        #表示
        show(frame)
        
        #
        frame_before = frame
    else:
        break
video.release()