Я получаю ошибку с морфологической маской - PullRequest
0 голосов
/ 21 марта 2019

Я написал программу, которая находит 2 области интереса, выбирает их в 2 отдельных фреймах и затем подсчитывает количество зеленых пикселей каждого фрейма. Это работает нормально, но когда я применяю морфологические маски для камеры, это выдает мне ошибку: /home/pi/opencv/opencv-3.4.0/modules/core/src/arithm.cpp:1769: ошибка: (-209) Нижняя Граница не является ни массивом того же размера и того же типа, что и src, ни скаляром в функции inRange. Как я могу это исправить?

Проблема возникает здесь

maskClose=cv2.morphologyEx(maskOpen,cv2.MORPH_CLOSE,kernelClose) maskFinal=maskClose ... for i in range(len(conts)): x,y,w,h=cv2.boundingRect(conts[i]) area=maskFinal[y:y+h, x:x+w] pixcount = cv2.inRange(area,lowerBound,upperBound ) pixNum = cv2.countNonZero(pixcount)

Полный код

import cv2
import numpy as np
from picamera.array import PiRGBArray
from picamera import PiCamera
import time
import sys
prevNrOfContours = 0
lowerBound=np.array([43,53,30])
upperBound=np.array([132,255,255])


camera = PiCamera()
camera.rotation = 180
camera.resolution = (640, 480)
camera.framerate = 30
font=cv2.FONT_HERSHEY_SIMPLEX

rawCapture = PiRGBArray(camera, size=(640, 480))
GREEN_MIN = np.array([0, 0, 0])
GREEN_MAX = np.array([0, 0, 0])



# allow the camera to warmup
time.sleep(0.1)
kernelOpen=np.ones((5,5))
kernelClose=np.ones((20,20))
# capture frames from the camera
for frame in camera.capture_continuous(rawCapture, format="bgr", use_video_port=True):
        img = frame.array
        imgHSV= cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
        mask=cv2.inRange(imgHSV,lowerBound,upperBound)
#morphology
        maskOpen=cv2.morphologyEx(mask,cv2.MORPH_OPEN,kernelOpen)
        maskClose=cv2.morphologyEx(maskOpen,cv2.MORPH_CLOSE,kernelClose)
        maskFinal=maskClose
        _, conts, _=cv2.findContours(maskFinal.copy(),cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_NONE)
        nrOfContours = len(conts)
        for i in range(len(conts)):
            x,y,w,h=cv2.boundingRect(conts[i])
            area=maskFinal[y:y+h, x:x+w] ######## problem is here
            pixcount = cv2.inRange(area,lowerBound,upperBound )
            pixNum = cv2.countNonZero(pixcount)
            print("Area No."+str(i), "Green pixels = " + str(no_black))
            cv2.rectangle(img,(x,y),(x+w,y+h),(0,0,255), 2)
            cv2.imshow("area" + str(i), area)
        # close unnecessary windows
        if prevNrOfContours > nrOfContours:
          for i in range(nrOfContours, prevNrOfContours):
            cv2.destroyWindow("area" + str(i))
        prevNrOfContours = nrOfContours
        cv2.imshow("cam",img)

        # clear the stream in preparation for the next frame
        key = cv2.waitKey(1) & 0xFF
        rawCapture.truncate(0)
        if key == ord("q"):
                break

1 Ответ

0 голосов
/ 22 марта 2019

Вы создаете подизображение maskFinal. maskFinal - это двоичное изображение, которое несовместимо с 3 каналами lowerBound.

Чтобы решить эту проблему, вы можете просто убрать строку pixcount = cv2.inRange(area,lowerBound,upperBound ) Маскированное изображение имеет белый цвет для зеленых областей, поэтому достаточно подсчета ненулевых значений.

Примечание: текущее изображение включает морфологические преобразования. Если вы не хотите считать пиксели, вызванные ими, вам нужно создать подизображение mask

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...