带有 Python 错误的 OpenCV - [=14= 中的断言失败 ((mask.type() == CV_8UC1 || mask.type() == CV_8SC1)) ]

Question

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

# 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 Error: Assertion failed ((mask.type() == CV_8UC1 || mask.type() == CV_8SC1)) in binary_op, file /home/user/opencv-2.4.10/modules/core/src/arithm.cpp, line 1035

Traceback (most recent call last): File "main_eye.py", line 86, in

roi_bg = cv2.bitwise_and(roi,roi,mask = mask_inv) cv2.error: /home/user/opencv-2.4.10/modules/core/src/arithm.cpp:1035: error: (-215) (mask.type() == CV_8UC1 || mask.type() == CV_8SC1) in function binary_op

Answer 1

您的遮罩是具有 3 个通道的图像，但它应该只有一个通道。您可以使用 mask.shape 检查它 - 它会 return 元组与您的面具尺寸。问题在这里：

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)

带有 Python 错误的 OpenCV - [=14= 中的断言失败 ((mask.type() == CV_8UC1 || mask.type() == CV_8SC1)) ]

OpenCV with Python error - Assertion failed ((mask.type() == CV_8UC1 || mask.type() == CV_8SC1)) in binary_op

python

webcam

opencv

computer-vision