下のコードはraspberry piで撮影した映像をRGBからHSVに変換して特定色を検出するコードなのですが、これを映像ではなく、あらかじめ用意した画像で行いたくて、コードを自分なりに書き換えてみましたが、エラーが出てできません。
どのように変更すればよいのでしょうか。
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 < 50) | (h > 200)) & (s > 100)] = 255 mask, contours, hierarchy = 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) 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()
変更した部分
img = cv2.imread("home/pi/picture/alarm.jpg") while cv2.waitKey(30) < 0: rects = find_rect_of_target_color(img) if len(rects) > 0: rect = max(rects, key=(lambda x: x[2] * x[3])) cv2.rectangle(img, tuple(rect[0:2]), tuple(rect[0:2] + rect[2:4]), (0, 0, 255), thickness=2) cv2.imshow('red', img) cv2.destroyAllWindows()
エラーコード
Traceback (most recent call last): File "red3.py", line 20, in <module> rects = find_rect_of_target_color(img) File "red3.py", line 5, in find_rect_of_target_color hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV_FULL) cv2.error: OpenCV(3.4.5) /home/pi/opencv-3.4.5/modules/imgproc/src/color.cpp:181: error: (-215:Assertion failed) !_src.empty$
回答1件
あなたの回答
tips
プレビュー