我想水平遮罩多个图像

时间:2019-10-18 10:55:14

标签: python numpy opencv image-processing mask

我的期刊页面图像很少,我想在不更改尺寸的情况下将两列遮盖两列白色,这意味着即使有一列,输出图像也应该与输入图像具有相同的尺寸。

我能够遮盖图像,但是遮盖部分变成了黑色,我希望将其变成白色。

import cv2

import numpy as np

# Load the original image

image = cv2.imread(filename = "D:\output_final_word5\image1.jpg")

# Create the basic black image 

mask = np.zeros(shape = image.shape, dtype = "uint8")

# Draw a white, filled rectangle on the mask image

cv2.rectangle(img = mask, pt1 = (0, 0), pt2 = (795, 3000), color = (255, 255, 

255), thickness = -1)

# Apply the mask and display the result

maskedImg = cv2.bitwise_and(src1 = image, src2 = mask)

#cv2.namedWindow(winname = "masked image", flags = cv2.WINDOW_NORMAL)

cv2.imshow("masked image",maskedImg)

cv2.waitKey(delay = 0)

cv2.imwrite("D:\Test_Mask.jpg",maskedImg)

我的最终目标是读取一个文件夹,该文件夹中有多个日记帐页面,需要在不影响输入图像尺寸的情况下通过遮蔽第一列然后遮盖另一列的方式将其保存为白色。 下面是附加的输入图像...

Input_Image1

Input_Image2

输出应该是这样的。...

Output_Image1

Output_Image2

1 个答案:

答案 0 :(得分:1)

您不需要遮罩即可绘制矩形。您可以直接在图像上绘制。

您还可以使用image.copy()与其他列创建第二张图片

顺便说一句:如果795在宽度的中间,那么您可以使用image.shape来获取其(height,width)并使用width//2而不是795将适用于具有不同宽度的图像。但是,如果795不在中间,则使用half_width = 795

import cv2

image_1 = cv2.imread('image.jpg')
image_2 = image_1.copy()

height, width, depth = image_1.shape # it gives `height,width`, not `width,height`
half_width = width//2
#half_width = 795

cv2.rectangle(img=image_1, pt1=(0, 0), pt2=(half_width, height), color=(255, 255, 255), thickness=-1)
cv2.rectangle(img=image_2, pt1=(half_width, 0), pt2=(width, height), color=(255, 255, 255), thickness=-1)

cv2.imwrite("image_1.jpg", image_1)
cv2.imwrite("image_2.jpg", image_2)

cv2.imshow("image 1", image_1)
cv2.imshow("image 2", image_2)

cv2.waitKey(0)
cv2.destroyAllWindows()