OpenCV制作Mask图像掩码的案例

def main():
    '''
    分别使用矩形和圆形遮罩从图像中提取矩形区域和圆形区域。
    '''
    import argparse
    import cv2
    import numpy as np

    images_bg = r'C:\dest\speed_bg.png'
    # 加载原始输入图像，并展示
    image = cv2.imread(images_bg)
    cv2.imshow("Original", image)
    # 掩码和原始图像具有相同的大小，但是只有俩种像素值：0（背景忽略）、255（前景保留）
    mask = np.zeros(image.shape[:2], dtype="uint8")
    cv2.rectangle(mask, (30, 90), (280, 440), 255, -1)
    cv2.imshow("Rectangular Mask", mask)
    # 应用掩码图像
    masked = cv2.bitwise_and(image, image, mask=mask)
    cv2.imshow("Rectangular Mask Applied to Image", masked)
    cv2.waitKey(0)
    # 构造一个圆形掩码（半径为140px，并应用位运算）
    mask = np.zeros(image.shape[:2], dtype="uint8")
    cv2.circle(mask, (210, 210), 160, 255, -1)
    # cv2.circle(mask, (202, 202), 110, 255, -1)
    masked = cv2.bitwise_and(image, image, mask=mask)
    bitwiseNot = cv2.bitwise_not(mask)
    cv2.imshow("bitwiseNot", bitwiseNot)
    # 展示输出图像
    cv2.imshow("Circular Mask", mask)
    cv2.imshow("Circular Mask Applied to Image", masked)
    cv2.waitKey()
    # TACH_PATTERN_PATH = r'C:\dest\new\tach_mask.png'
    # aircv.imwrite(TACH_PATTERN_PATH, bitwiseNot)
    TACH_PATTERN_PATH2 = r'C:\dest\new\speed_mask.png'
    aircv.imwrite(TACH_PATTERN_PATH2, mask)

if __name__ == "__main__":
    main()