python - 使用 Numpy 或 OpenCV 将图像中的剪切内容复制到新图像

Question

是否可以使用 Python 中的 OpenCV 或 Numpy 仅复制图像中的特定区域并将其粘贴到另一个图像？

假设我有一个 RGB 图像和一个图像的灰度蒙版。 (两者尺寸相同)

是否可以仅将 mask 像素重叠到新图像的特定部分的船只复制？

这是我尝试过的代码。

image1 = cv2.imread("boat.png")
mask = cv2.imread("boat_mask.png")
image2 = cv2.imread("sea.png")
width, height = image1.shape
image2[y:y+height,x:x+width] = image1[mask==255]

这不起作用，因为 image1[mask==255] 不再具有 2d 形状。

解决方案:

image2_cutout = image2[y:y+height,x:x+width]
image2_cutout[mask==255] = image1[mask==255]
image2[y:y+height,x:x+width] = image2_cutout

Answer 1

这里是:

import cv2


image1 = cv2.imread("data/boat.png")
_, mask = cv2.threshold(cv2.cvtColor(cv2.imread("data/masked_boat.png"), cv2.COLOR_BGR2GRAY), 120, 255, cv2.THRESH_BINARY)
image2 = cv2.imread("data/sea.jpg")
width, height, _ = image1.shape
y = 300
x = 300

masked_normalized = cv2.bitwise_and(image1, image1, mask=mask)

cut_from_bg = image2[x:x+width, y:y+height]
img1_bg = cv2.bitwise_and(cut_from_bg, cut_from_bg, mask=cv2.bitwise_not(mask))

dst = cv2.add(img1_bg, masked_normalized)

image2[x:width + x, y:height + y] = dst
cv2.imshow("sea_with_boat", image2)

cv2.waitKey(0)
cv2.destroyAllWindows()

我在这里做的第一件事是获取船的mask，并使用cv2.bitwise_and获得船的真实蒙版图像。然后我在想要放置船的地方切出了一部分海，并在这里切出了船的形状。然后我使用cv2.add组合蒙版船图像和剪切的海洋图像，然后我将这部分图像放在原始海洋图像上。