首页 文章

OpenCV与Python错误 - 断言失败((mask.type()== CV_8UC1 || mask.type()== CV_8SC1))在binary_op中

提问于
浏览
2

我正在尝试在网络摄像头Feed上叠加图像 . 这是代码的主要部分 -

# Load our overlay image: glasses.png
imgGlasses = cv2.imread('1.png')

# Create the mask for the glasses
imgGlassesGray = cv2.cvtColor(imgGlasses, cv2.COLOR_BGR2GRAY)
ret, orig_mask = cv2.threshold(imgGlasses, 10, 255, cv2.THRESH_BINARY)

# Create the inverted mask for the glasses
orig_mask_inv = cv2.bitwise_not(orig_mask)

# Convert glasses image to BGR
# and save the original image size (used later when re-sizing the image)
imgGlasses = imgGlasses[:,:,0:3]
origGlassesHeight, origGlassesWidth = imgGlasses.shape[:2]

video_capture = cv2.VideoCapture(0)

while True:
    # Capture frame-by-frame
    ret, frame = video_capture.read()

    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    faces = face_cascade.detectMultiScale(gray, 1.3, 5)
    for (x,y,w,h) in faces:
        #cv2.rectangle(frame,(x,y),(x+w,y+h),(255,0,0),2)
            roi_gray = gray[y:y+h, x:x+w]
            roi_color = frame[y:y+h, x:x+w]

            eyes = eye_cascade.detectMultiScale(roi_gray)
            for (ex,ey,ew,eh) in eyes:
            cv2.rectangle(roi_color,(ex,ey),(ex+ew,ey+eh),(0,255,0),2)

        for (ex, ey, ew, eh) in eyes:
            glassesWidth = 3*ew
            glassesHeight = glassesWidth * origGlassesHeight / origGlassesWidth

            # Center the glasses on the bottom of the nose
            x1 = ex - (glassesWidth/4)
            x2 = ex + ew + (glassesWidth/4)
            y1 = ey + eh - (glassesHeight/2)
            y2 = ey + eh + (glassesHeight/2)

            # Re-calculate the width and height of the glasses image
            glassesWidth = x2 - x1
            glassesHeight = y2 - y1

            # Re-size the original image and the masks to the glasses sizes
            # calcualted above
            glasses = cv2.resize(imgGlasses, (glassesWidth,glassesHeight), interpolation = cv2.INTER_AREA)
            mask = cv2.resize(orig_mask, (glassesWidth,glassesHeight), interpolation = cv2.INTER_AREA)
            mask_inv = cv2.resize(orig_mask_inv, (glassesWidth,glassesHeight), interpolation = cv2.INTER_AREA)

            # take ROI for glasses from background equal to size of glasses image
            roi = roi_color[y1:y2, x1:x2]

                # roi_bg contains the original image only where the glasses is not
                # in the region that is the size of the glasses.
            roi_bg = cv2.bitwise_and(roi,roi,mask = mask_inv)

                # roi_fg contains the image of the glasses only where the glasses is
            roi_fg = cv2.bitwise_and(glasses,glasses,mask = mask)

                # join the roi_bg and roi_fg
            dst = cv2.add(roi_bg,roi_fg)

                # place the joined image, saved to dst back over the original image
            roi_color[y1:y2, x1:x2] = dst

            break

尝试运行程序时出现以下错误,我不知道如何解决此问题 -

OpenCV错误:断言失败((mask.type()== CV_8UC1 || mask.type()== CV_8SC1))在binary_op中,文件/home/user/opencv-2.4.10/modules/core/src/arithm .cpp,第1035行追溯(最近一次调用最后一次):文件“main_eye.py”,第86行,在roi_bg = cv2.bitwise_and(roi,roi,mask = mask_inv)cv2.error:/home/user/opencv-2.4 .10 / modules / core / src / arithm.cpp:1035:错误:(-215)(mask.type()== CV_8UC1 || mask.type()== CV_8SC1)函数binary_op

1 回答

  • 5

    你的面具是3个通道的图像,但它应该只有一个通道 . 您可以使用 mask.shape 进行检查 - 它将返回具有掩码尺寸的元组 . 问题在这里:

    imgGlassesGray = cv2.cvtColor(imgGlasses, cv2.COLOR_BGR2GRAY)
    ret, orig_mask = cv2.threshold(imgGlasses, 10, 255, cv2.THRESH_BINARY)
    

    您正在使用具有3个通道的图像创建蒙版,因此您的蒙版也将具有3个通道 . 使用单通道图像创建它(很可能你想使用imgGlassesGray而不是imgGlasses) - 它应该解决你的问题 .

    imgGlassesGray = cv2.cvtColor(imgGlasses, cv2.COLOR_BGR2GRAY)
    ret, orig_mask = cv2.threshold(imgGlassesGray, 10, 255, cv2.THRESH_BINARY)
    

相关问题