我只能仅获得图像的面具？

Question

嗨，我试图拿到一件T恤的面具。

这是我得到的：

但是，我只想在中间收到衣服的形状。我该怎么做？

谢谢！

这是代码：

import cv2
import numpy as np

# load image whose you want to create mask
img = cv2.imread('01430_00.jpg')

# convert to graky
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# threshold input image as mask
mask = cv2.threshold(gray,220,220, cv2.THRESH_BINARY)[1]

# negate mask
mask = 255 - mask

# apply morphology to remove isolated extraneous noise
# use borderconstant of black since foreground touches the edges
kernel = np.ones((3,3), np.uint8)
mask = cv2.morphologyEx(mask, cv2.MORPH_OPEN, kernel)
mask = cv2.morphologyEx(mask, cv2.MORPH_CLOSE, kernel)

# anti-alias the mask -- blur then stretch
# blur alpha channel
mask = cv2.GaussianBlur(mask, (0,0), sigmaX=2, sigmaY=2, borderType = 
cv2.BORDER_DEFAULT)

# linear stretch so that 127.5 goes to 0, but 255 stays 255
mask = (2*(mask.astype(np.float32))-255.0).clip(0,255).astype(np.uint8)



# Show Image in Opencv Windows
cv2.imshow("Original", img)
cv2.imshow("MASK", mask)

# Save mask Image
cv2.imwrite("Mask2.jpg",mask)

cv2.waitKey(0)
cv2.destroyAllWindows()

Answer 1

您已经有一个大的“服装”轮廓。您可以使用以下代码填充最大轮廓

contours, _ = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
cnt = max(contours, key = cv2.contourArea)
cv2.drawContours(mask,[cnt],0,255,-1)