Raspberry Piカメラの映像で赤色をリアルタイムで認識したい。

Question

赤色を認識するコードををインターネットから拾ってきたのですが、これをraspberry piのカメラで行うにはどのようにプログラムを組んだらよいのでしょうか。
また、普通にプログラムを実行するとエラーが発生しました。どのように改善すればよいでしょうか。

import cv2
import numpy as np

def find_rect_of_target_color(image):
  hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV_FULL)
  h = hsv[:, :, 0]
  s = hsv[:, :, 1]
  v = hsv[:, :, 0]
  mask = np.zeros(h.shape, dtype=np.uint8)
  mask[((h < 50) | (h > 200)) & (s > 100)] = 255
  contours, _ = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
  rects = []
  for contour in contours:
    approx = cv2.convexHull(contour)
    rect = cv2.boundingRect(approx)
    rects.append(np.array(rect))
  return rects

capture = cv2.VideoCapture(0)
while cv2.waitKey(30) < 0:
    _, frame = capture.read()
    rects = find_rect_of_target_color(frame)
    for rect in rects:
        cv2.rectangle(frame, tuple(rect[0:2]), tuple(rect[0:2] + rect[2:4]), (0, 0, 255), thickness=2)
        cv2.imshow('red', frame)
capture.release()
cv2.destroyAllWindows()

エラーコード

Traceback (most recent call last):
  File "red.py", line 22, in <module>
    rects = find_rect_of_target_color(frame)
  File "red.py", line 11, in find_rect_of_target_color
    contours, _ = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
ValueError: too many values to unpack

Answer 1

snow1231さん

sudo modprobe bcm2835-v4l2 を実行して、v4l2ドライバをロードすれば、OpenCVのVideoCapture関数が使えるようになるので、コードの変更なしで動くと思います。

Comments

[退会済みユーザー]

回答していただきありがとうございます。
そのコードを実行してみましたが、特に変化がなく、プログラムを実行したらエラーが発生しました。（エラーコードを質問のところに追加しました）
これはコードが間違っているということでしょうか。

[mt08]

snow1231さん、

"ValueError: too many values to unpack" というエラーメッセージをみればわかると思いますが、contours, _ = cv2.findContours(~~~)で返ってくる値の数が2つでより多い、ってことですね。

opencvのバージョンによって変わるようなので、使用しているバージョンとドキュメントを確認してみてください

[退会済みユーザー]

ありがとうございます。戻り値を３つにしたら出来ました。