opencv python 画像処理

Question

# -*- coding: utf-8 -*-
import cv2
import numpy as np

# フレーム差分の計算
def frame_sub(src1, src2, src3, th):
    # フレームの絶対差分
    d1 = cv2.absdiff(src1, src2)
    d2 = cv2.absdiff(src2, src3)


    diff = cv2.bitwise_and(d1, d2)

    # 二値化処理
    diff[diff < th] = 0
    diff[diff >= th] = 255

    # メディアンフィルタ処理（ゴマ塩ノイズ除去）
    mask = cv2.medianBlur(diff, 3)

    return  mask

def main():
    # カメラのキャプチャ
    cap = cv2.VideoCapture(0)

    # フレームを3枚取得してグレースケール変換
    frame1 = cv2.cvtColor(cap.read()[1], cv2.COLOR_RGB2GRAY)
    frame2 = cv2.cvtColor(cap.read()[1], cv2.COLOR_RGB2GRAY)
    frame3 = cv2.cvtColor(cap.read()[1], cv2.COLOR_RGB2GRAY)

    while(cap.isOpened()):
        # フレーム間差分を計算
        mask = frame_sub(frame1, frame2, frame3, th=30)

        # 結果を表示

        cv2.imshow("Mask", mask)

        # 3枚のフレームを更新
        frame1 = frame2
        frame2 = frame3
        frame3 = cv2.cvtColor(cap.read()[1], cv2.COLOR_RGB2GRAY)

        # qキーが押されたら途中終了
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break

    cap.release()
    cv2.destroyAllWindows()

if __name__ == '__main__':
    main()

このコードでフレームの差分が出せて動いたものが白く映ります。
これを利用して『白く写った時に一秒間隔でカラーの写真を撮り保存、白く映らない時は写真を撮らない』というようなプログラムを作りたいです。

import cv2
import numpy as np
import time

cap = cv2.VideoCapture(0)
size = (640, 480)
ret, frame1 = cap.read()
#frame1 = cv2.resize(frame1, size)
ret, frame2 = cap.read()
#frame2 = cv2.resize(frame2, size)
ret, frame3 = cap.read()

def sub(a, b, c):
    c = a - b
    if c > 0:
        return c
    if c < 0:
        return -1 * c

#ave1 = frame1.mean(0).mean(0).mean(0)
#ave2 = frame2.mean(0).mean(0).mean(0)

while(True):
    frame1 = frame2
    #frame2 = frame3
    frame2 = cap.read()
    #frame3 = cap.read()

    #ave1 = ave2
    ave2 = frame2.mean(0).mean(0).mean(0)
    print ave2


    #subave = sub(ave1, ave2, 0)
    #print subave

    time.sleep(0.7)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

平均値を出すところまで行きましたがwhileの外にave1,ave2をかくと、もちろん値が更新されないので内側にかかなければならいと思いますが、
ave2 = frame2.mean(0).mean(0).mean(0)
AttributeError: 'tuple' object has no attribute 'mean'
というエラーが出てしまいます。これを解決するにはどうしたら良いでしょうか？

Answer 1

Example 3.16. タプルはメソッドを～のくだりをご覧ください。listの計算はnumpyでいけますね。

Comments

[退会済みユーザー]

タプルは不変なものだとわかりました。numpyの計算というのは具体的にはave2 = frame2.mean(0).mean(0).mean(0)はダメでnp.mean(frame2)かnp.average(frame2)を繰り返したほうがいいということでしょうか？

[退会済みユーザー]

そうだと思います。
(もしかしたら繰り返し並べる処理もあるのかもしれませんが) .mean(0)と一回するだけで値は取れそうですね。もし、この式の言わんとするところが明るさを判定したい、というところにあるのであれば、
1. BGRをgrayscaleに(BGRを1chに統合)してしまって、それを平均or中央値処理
2. BGRをBとGとRに分割してそれぞれを平均or中央値処理
でよいと思います。

Answer 2

# -*- coding: utf-8 -*-
import cv2
import numpy as np
import time
import sys

def frame_sub(src1, src2, src3, th):
    d1 = cv2.absdiff(src1, src2)
    d2 = cv2.absdiff(src2, src3)
    diff = cv2.bitwise_and(d1, d2)
    diff[diff < th] = 0
    diff[diff >= th] = 255
    mask = cv2.medianBlur(diff, 3)
    mas = np.mean(mask)
    #print mas
    return  mas

cap = cv2.VideoCapture(0)

frame1 = cv2.cvtColor(cap.read()[1], cv2.COLOR_RGB2GRAY)
frame2 = cv2.cvtColor(cap.read()[1], cv2.COLOR_RGB2GRAY)
frame3 = cv2.cvtColor(cap.read()[1], cv2.COLOR_RGB2GRAY)

count = 0
while(True):
    mask = frame_sub(frame1, frame2, frame3, th=30)
    frame1 = frame2
    frame2 = frame3
    frame3 = cv2.cvtColor(cap.read()[1], cv2.COLOR_RGB2GRAY)

    if mask > 0.001:
        while(True):
            mask = frame_sub(frame1, frame2, frame3, th=30)
            frame1 = frame2
            frame2 = frame3
            frame3 = cv2.cvtColor(cap.read()[1], cv2.COLOR_RGB2GRAY)

            cap = cv2.VideoCapture(0)
            ret, frame = cap.read()

            count += 1
            count_padded = '%05d' % count
            write_file_name = count_padded + ".jpg"
            cv2.imwrite(write_file_name, frame)
            #time.sleep(0.05)
            if mask < 0.001:
                break