pythonで画像を楕円形にトリミングしたい

Question

前提・実現したいこと

pythonで画像を楕円形にトリミングしたいです。

発生している問題・エラーメッセージ

以下のコードで四角にトリミングすることはできたのですが、これと同じように楕円でもトリミングしたいです。
target1=img0[y1:y1+h1, x1:x1+w1,0];

該当のソースコード

import cv2
import numpy as np

%matplotlib inline
from matplotlib import pyplot as plt

cap = cv2.VideoCapture('oclusion1.MOV')

(flag,frame)=cap.read()
plt.imshow(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) );

img0=cv2.cvtColor(frame, cv2.COLOR_BGR2RGB);

#初期位置の左上の座標
x1=400;
y1=720;
#四角の大きさ
w1=140;
h1=400;

#囲う
cv2.rectangle(img0,(x1,y1),(x1+w1,y1+h1),(255,0,0),3); 
plt.imshow(img0);

Answer 1

ellipse関数でマスク画像に楕円描画しAND演算にてトリミングできます。

import matplotlib.pyplot as plt
import cv2
import numpy as np

img = cv2.imread('lena.png')

# 楕円でマスク
mask = np.zeros(img.shape, dtype=np.uint8)
cv2.ellipse(mask,((256,256),(480,320),0),(255,255,255),-1)
img = img & mask

plt.imshow(cv2.cvtColor(img,cv2.COLOR_BGR2RGB)) #array to 2Dfigure
plt.show()

Comments

[退会済みユーザー]

ありがとうございます。やってみます。

Answer 2

.opencvではありませんが、画像を円形で切り抜く方法が書いてあるサイトを見つけました
(楕円ではなく円ですが...)

画像処理についてあれこれ

Comments

[退会済みユーザー]

ありがとうございます。参考にしてみます。

Answer 3

提案

画像自体を楕円にすることは矩形の配列で表現されているため難しいので、楕円のマスクを作成して矩形から楕円の範囲内のみを取り出す方法があると思います。
以下に質問者と同様の方法で矩形に切り取ったあとで、楕円のマスクを作成するコードを書いてみました。
方法は楕円の方程式を用いているだけなので、そちらは別途検索してみてください。

# 矩形にトリミング
target1 = img0[y1:y1+h1, x1:x1+w1]

# 楕円のマスクを作成
mask = np.zeros_like(target1)
cx = w1/2
cy = h1/2
for iy in range(h1):
    for ix in range(w1):
        if (ix-cx)**2/(w1/2)**2 + (iy-cy)**2/(h1/2)**2 <= 1:
            mask[iy, ix] = 1

plt.imshow(mask*target1)

Comments

[退会済みユーザー]

ありがとうございます。やってみます。