операнды не могут быть переданы вместе с фигурами (1406,1100,3) (653,639,3) - PullRequest
0 голосов
/ 01 мая 2019

почему возникает эта ошибка? Я делюсь с вами изображением, которое я использую. Я попробовал этот же скрипт с другими изображениями, и ошибка не происходит Идея заключается в том, что это может быть общий сценарий для всех видов изображений.

Цель состоит в том, чтобы наконец получить белый фон для изображения.

import cv2 
import numpy as np 
from matplotlib import pyplot as plt 

#== Parameters ======================================================================= 
BLUR = 21 
CANNY_THRESH_1 = 10 
CANNY_THRESH_2 = 200 
MASK_DILATE_ITER = 10 
MASK_ERODE_ITER = 10 
MASK_COLOR = (0.0,0.0,1.0) # In BGR format 


#== Processing ======================================================================= 

#-- Read image ----------------------------------------------------------------------- 
img = cv2.imread('./remera.png') 

#-- Edge detection ------------------------------------------------------------------- 
edges = cv2.Canny(gray, CANNY_THRESH_1, CANNY_THRESH_2) 
edges = cv2.dilate(edges, None) 
edges = cv2.erode(edges, None) 

#-- Find contours in edges, sort by area --------------------------------------------- 
contour_info = [] 
_, contours, _ = cv2.findContours(edges, cv2.RETR_LIST, cv2.CHAIN_APPROX_NONE) 
for c in contours: 
    contour_info.append((
     c, 
     cv2.isContourConvex(c), 
     cv2.contourArea(c), 
    )) 
contour_info = sorted(contour_info, key=lambda c: c[2], reverse=True) 
max_contour = contour_info[0] 

# split image into channels 
c_red, c_green, c_blue = cv2.split(img) 

#-- Create empty mask, draw filled polygon on it corresponding to largest contour ---- 
# Mask is black, polygon is white 
mask = np.zeros(edges.shape) 
cv2.fillConvexPoly(mask, max_contour[0], (255)) 

#-- Smooth mask, then blur it -------------------------------------------------------- 
mask = cv2.dilate(mask, None, iterations=MASK_DILATE_ITER) 
mask = cv2.erode(mask, None, iterations=MASK_ERODE_ITER) 
mask = cv2.GaussianBlur(mask, (BLUR, BLUR), 0) 

mask_stack = np.dstack([mask]*3) # Create 3-channel alpha mask 

#-- Blend masked img into MASK_COLOR background -------------------------------------- 
mask_stack = mask_stack.astype('float32')/255.0   # Use float matrices, 
img   = img.astype('float32')/255.0     # for easy blending 

masked = (mask_stack * img) + ((1-mask_stack) * MASK_COLOR) # Blend 
masked = (masked * 255).astype('uint8')      # Convert back to 8-bit 



---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-71-dd2e80b083d7> in <module>()
     53 img   = img.astype('float32')/255.0     # for easy blending
     54 
---> 55 masked = (mask_stack * img) + ((1-mask_stack) * MASK_COLOR) # Blend
     56 masked = (masked * 255).astype('uint8')      # Convert back to 8-bit
     57 
[![enter image description here][1]][1]
ValueError: operands could not be broadcast together with shapes (1406,1100,3) (653,639,3) 

Любая помощь приветствуется.

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