ｇrabCutで背景を黒→アルファチャネルにしたい

Question

###前提・実現したいこと
python + openCVを用いて grabcutを利用しております。
ベースのプログラムでは背景が黒色のままなのですが、これをアルファチャネルにしたいです。
どの部分を変更すれば良いのか、それともopenCVではアルファチャネルにできないのでしょうか。
ご享受お願いします。

###該当のソースコード

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

for i in range(1, 468):
    img = cv2.imread("{0:03d}.jpg" .format(i))
    mask = np.zeros(img.shape[:2], np.uint8)

    bgdModel = np.zeros((1,65),np.float64)
    fgdModel = np.zeros((1,65),np.float64)

    rect = (39,5,100,250)
    cv2.grabCut(img,mask,rect,bgdModel,fgdModel,5,cv2.GC_INIT_WITH_RECT)

    mask2 = np.where((mask==2)|(mask==0),0,1).astype('uint8')
    img = img*mask2[:,:,np.newaxis]

    plt.imshow(img),plt.colorbar(),plt.show()
    cv2.imwrite("a{0:03d}.jpg".format(i), img)

###結果の表示画像

###補足情報(言語/FW/ツール等のバージョンなど)
python3系
ubuntu 14.xx
openCV を使用しております。

Answer 1

まず結果はこれでよいですね。

woman.jpg	grabbed.jpg	alpha.png

---

大事なところは、
*Alpha channel付き画像は、PNGで出力(JPEGでは使えません)
→cv2.imwrite("alpha.png",img_alpha)の部分です

*Alpha channelを付加する際には、[b,g,r]に分割して[a]を付加
→img_alpha = cv2.merge(img_bgr+[mask2])の部分です

Alpha channelを合成するには、mask255
→mask2 = mask2*255の部分です
*255なしで出力するとよく分かりますが、マスク画像は0か1かにしかなっておらず、これだとアルファチャンネルを混ぜてもほとんど見た目は変わりませんね。なので、int8に合わせて0か255かにしないといけませんね。

---

投稿そのままだとループしてしまうので、必要なところだけ抜き出しました。

import numpy as np
import cv2

img = cv2.imread("woman.jpg" )
mask = np.zeros(img.shape[:2], np.uint8)

bgdModel = np.zeros((1,65),np.float64)
fgdModel = np.zeros((1,65),np.float64)
iter = 5
rect = (39,5,100,250)

cv2.grabCut(img, mask, rect, bgdModel, fgdModel, iter, cv2.GC_INIT_WITH_RECT)

mask2 = np.where((mask==2)|(mask==0),0,1).astype('uint8')

img = img*mask2[:,:,np.newaxis]
cv2.imshow("img",img)
cv2.imwrite("grabbed.png", img)

# ---------------------------------

img_bgr = cv2.split(img)
mask2 = mask2*255

# cv2.imwrite("aaa.png",mask2*255)

img_alpha = cv2.merge(img_bgr+[mask2])

cv2.imshow("alpha",img_alpha)
cv2.waitKey(0)
cv2.imwrite("alpha.png",img_alpha)

Comments

[daisuke_factex]

回答有り難うございます。私自身素人ですが、詳らかに解説してもらえ非常に嬉しいです。
BAにさせていただきました！