# 分別使用矩形和圓形遮罩從圖像中提取身體和臉部。
# USAGE
# python opencv_masking.py

import argparse

import cv2
# 導入必要的包
import numpy as np

# 構建命令行參數及解析
# --image 輸入圖像路徑
ap = argparse.ArgumentParser()
ap.add_argument("-i", "--image", type=str, default="yc.jpg",
                help="path to the input image")
args = vars(ap.parse_args())

# 加載原始輸入圖像，并展示
image = cv2.imread(args["image"])
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, (155, 200), 140, 255, -1)
masked = cv2.bitwise_and(image, image, mask=mask)

# 展示輸出圖像
cv2.imshow("Circular Mask", mask)
cv2.imshow("Circular Mask Applied to Image", masked)
cv2.waitKey(0)