opencvの処理をリアルタイムに

Question

処理が途中でとまり、
処理後フォルダにaviファイルは保存されるのですが2秒ほどの動画となってしまいます
動画を撮り続けて自分で止めたい場合どのようにすればよいでしょうか
アドバイスお願いします。

Comments

[coco_bauer]

リアルタイムでエッジ検出をするのは無理があると思います。参考にされた[https://qiita.com/hitomatagi/items/12a2eceaf65f142ec3df]には、リアルタイムで動画をキャプチャーする事と、静止画のエッジ検出をする事が別々に説明されています。それらを混ぜるのは無茶ではないでしょうか。

Answer 1

リアルタイムでエッジ処理をして動画ファイルを作成します。
処理の流れは以下のとおり

1.カレントディレクトリ直下に新しくフォルダを作成しそのフォルダに移動する。
2.webカメラに接続
3.1フレームを読み込みリアルタイムで画像加工処理
4.処理したフレームを表示とともに画像として保存(フレームごとに画像は保存し蓄積される)
5.３と４を繰り返す。「esc」ボタンを押すと終了処理に入る
6.今まで保存した画像から動画ファイルを作成

import cv2
import numpy as np
import os
import datetime

##カレントディレクトリのパスを取得
current = os.getcwd()
##ディレクトリ作成
now = datetime.datetime.now()
dir_name = "/IMG_{0:%m%d%H%M}".format(now)
if os.path.isdir(current + dir_name) == False:
    os.mkdir(current + dir_name)  #make directory
os.chdir(current + dir_name)  #move to directory

# VideoCaptureのインスタンスを作成する。
cap = cv2.VideoCapture(0)
cap.set(cv2.CAP_PROP_FPS, 30)

#フレーム数カウント用
count = 1

while True:
    # VideoCaptureから1フレーム読み込む
    ret, frame = cap.read()
    w, h = frame.shape[0] , frame.shape[1]

    # 加工なし画像を表示する
    cv2.imshow('Raw Frame', frame)

    # 画像加工処理
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    blur = cv2.GaussianBlur(gray,(11,11),0)
    _, binary = cv2.threshold(blur, 0, 255, cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)
    nega = cv2.bitwise_not(binary)
    canny = cv2.Canny(nega, 100, 200)
    #2値画像をrgbに変換
    backtorgb = cv2.cvtColor(canny,cv2.COLOR_GRAY2RGB)


    # 加工済の画像を表示する
    cv2.imshow('Edited Frame', backtorgb)

    # フレームを保存
    filename = "IMG_{}.jpg".format(count)
    cv2.imwrite(filename, backtorgb)
    count += 1

    # キー入力を1ms待って、k が27（ESC）だったらBreakする
    k = cv2.waitKey(1)
    if k == 27:
        break

# キャプチャをリリースして、ウィンドウをすべて閉じる
cap.release()
cv2.destroyAllWindows()


# 連番画像から動画作成
fourcc = cv2.VideoWriter_fourcc('m','p','4','v')
video = cv2.VideoWriter('video.mp4', fourcc, 20.0, (h, w))
for i in range(1, count):
    img = cv2.imread("IMG_{}.jpg".format(i))
    img = cv2.resize(img, (h,w))
    video.write(img)

video.release()

print("完了")