長方形のマスク画像の作成方法

Question

実現したいこと

このようなマスク画像をpythonで作りたいです。
黒の画像のサイズは(3280, 2464)
白の画像のサイズは(800,600)
白の画像を置く座標も指定したいです。白の長方形の左上の座標は（525,850）を想定しています。

黒い画像をつくることはできています。
白い画像を描画するのもおそらくできていると思うのですが
これらの２つの画像を組み合わせる方法が良く分からないです。
よろしくお願いします。

このサイトを参考にしました。
https://note.nkmk.me/python-pillow-composite/
https://note.nkmk.me/python-pillow-imagedraw/

プログラムコード

%matplotlib inline  
import cv2
import numpy as np
from matplotlib import pyplot as plt
from PIL import Image, ImageDraw

#黒い画像を保存
width = 3280
height = 2464
imageArray = np.zeros((height, width, 3), np.uint8)

cv2.imwrite("black.png", imageArray);

#白い画像を描画
draw.rectangle((525, 850, 1325, 250), fill=(0, 0, 0))

Answer 1

draw.rectangleしているあたり、二つの画像の合成でなくていいのでしょうか。
参考サイトのまま素直にコピペ実装すると次のようになると思います。

from PIL import Image, ImageDraw
# 背景黒の描画領域
width = 3280
height = 2464
im = Image.new('RGB', (width, height), (0, 0, 0))
# 白い四角を描画
draw = ImageDraw.Draw(im)
draw.rectangle((525, 850, 1325, 250), fill=(255, 255, 255), outline=(255, 255, 255))
# 保存
im.save('image.png')

Answer 2

1. np.zeros() で全画素が0の画像を作成
2. cv2.rectangle() で白い長方形を描画

cv2.rectangle(img, pt1, pt2, color[, thickness[, lineType[, shift]]])

thickness=-1 で塗り潰しになります。

import cv2

w, h = 3280, 2464
mask = np.zeros((w, h), dtype=np.uint8)
#白い画像を描画
x1, y1 = 525, 600  # 左上の座標
x2, y2 = 800, 800  # 右下の座標
cv2.rectangle(mask, (x1, y1), (x2, y2),
              color=255, thickness=-1)

cv2.imwrite("mask.png", mask)

Comments

[masukin]

ありがとうございます！
助かりました（o_ _)ｏ））