ラズパイとopencvとPiCameraを使って赤色の物体の認識をする

Question

前提・実現したいこと

ラズパイにopencvとPiCameraを乗せて赤色の物体の認識をしようとしているのですが下記のエラーが出て実行できません

発生している問題・エラーメッセージ

[ INFO:0] VIDEOIO: Enabled backends(4, sorted by priority): FFMPEG(1000); V4L2(990); CV_IMAGES(980); CV_MJPEG(970)
(DEBUG) V4L: opening /dev/video0
Traceback (most recent call last):
  File "cameratest.py", line 23, in <module>
    rects = find_rect_of_target_color(frame)
  File "cameratest.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

該当のソースコード

# coding:utf-8
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]
  mask = np.zeros(h.shape, dtype=np.uint8)
  mask[((h < 20) | (h > 200)) & (s > 128)] = 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

if __name__ == "__main__":
  capture = cv2.VideoCapture(0)
  while cv2.waitKey(30) < 0:
    _, frame = capture.read()
    rects = find_rect_of_target_color(frame)
    if len(rects) > 0:
      rect = max(rects, key=(lambda x: x[2] * x[3]))
      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()

Comments

[can110]

前回の質問と同じようですが、cv2.cvtColor(までは正常に到達するようになったということでしょうか？

Answer 1

エラーを見る限り、次の環境をご利用であると推察します。

• OpenCV3以前
• Python2.7

問題解決に役立ちますので、環境は質問に明記するようにしてください。

---

Traceback (most recent call last):
...
contours, _ = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
ValueError: too many values to unpack

cv2.findContoursの返り値が二要素のタプルになったのは、OpenCV4以降です。参考
次のように返り値を受け取ってください。

_, contours, _ = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

あるいは

contours = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)[1]

Comments

[Leo0121]

ありがとうございます．
その部分のエラーは解決できましたが新しいエラーが出てきてしまいました．
[ INFO:0] VIDEOIO: Enabled backends(4, sorted by priority): FFMPEG(1000); V4L2(990); CV_IMAGES(980); CV_MJPEG(970)
(DEBUG) V4L: opening /dev/video0

(red:1210): Gtk-WARNING **: cannot open display:
調べてみるとunset SSH_ASKPASSで解決できると分かったのですが，変わりませんでした．

[LouiS0616]

ディスプレイはちゃんと積んでますか？

[Leo0121]

ssh接続やvnc接続ではできませんか？

[LouiS0616]

ラズパイは触ったことが無いので解決方法には全然見当が付きませんが、一般論として。
cannot open displayと出ている以上、まずは画像を読み込んで表示するだけのコードをテストした方が良いと思います。

[Leo0121]

わかりました．
ディスプレイを接続して実行してみたのですがさらにエラーが出てきました．
このエラーの解決策を教えていただけないでしょうか
[ INFO:0] VIDEOIO: Enabled backends(4, sorted by priority): FFMPEG(1000); V4L2(990); CV_IMAGES(980); CV_MJPEG(970)
(DEBUG) V4L: opening /dev/video0
(DEBUG) try_init_v4l2 open "/dev/video0": No such file or directory
VIDEOIO ERROR: V4L: can't open camera by index 0
Traceback (most recent call last):
File "cameratest.py", line 23, in <module>
rects = find_rect_of_target_color(frame)
File "cameratest.py", line 6, in find_rect_of_target_color
hsv= cv2.cvtColor(image, cv2.COLOR_BGR2HSV_FULL)
cv2.error: OpenCV(3.4.2) /home/pi/opencv/modules/imgproc/src/color.hpp:253: error: (-215:Assertion failed) VScn::contains(scn) && VDcn::contains(dcn) && VDepth::contains(depth) in function 'CvtHelper'

[LouiS0616]

cvtColorの前の行に assert image is not None と入れて実行するとどうなりますか？

[Leo0121]

下記のようにエラーが出てきます．
[ INFO:0] VIDEOIO: Enabled backends(4, sorted by priority): FFMPEG(1000); V4L2(990); CV_IMAGES(980); CV_MJPEG(970)
(DEBUG) V4L: opening /dev/video0
(DEBUG) try_init_v4l2 open "/dev/video0": No such file or directory
VIDEOIO ERROR: V4L: can't open camera by index 0
Traceback (most recent call last):
File "cameratest.py", line 24, in <module>
rects = find_rect_of_target_color(frame)
File "cameratest.py", line 6, in find_rect_of_target_color
assert image is not None
AssertionError

[LouiS0616]

imageがNoneであるということは、つまりそもそも画像が読み込めて無いということです。

[Leo0121]

今回VideoCaptureを使用して画像ではなくPiCameraからリアルタイムで処理しようとしましたがこの場合はVideoCaptureが正しく動作していないという事でしょうか？

[LouiS0616]

そうなります。
そしてVideoCaptureが正しく動作していないということは、カメラの接続が上手くいっていないということですね。

Answer 2

使用しているカメラがpicameraだったのでその設定が必要でした．

sudo modprobe bcm2835-v4l2