Opencvで画像の色を抽出してアルファチャンネル付きで保存したい
解決済
回答 2
投稿
- 評価
- クリップ 0
- VIEW 1,855
コラージュを作るときのいわゆるBB(ブルーバック?)のような画像で、背景の単色だけを
抜いてその部分を透過させたいです。
画像はBGRです。最終的に入力に対して出力が周りが透過されてて、ぽつんと撮影対象が写っている感じにしたいです。
一応色を摘出するところまではできたのですが、アルファチャンネルを追加するところで詰まってしまいました。
皆様のお力をお借りしたいです。参考になりそうなサイトなどでもかまいません。
よろしくお願いいたします。
pythonはver2. 、OpenCVはver3です。
import cv2
import numpy as np
import sys
cap = cv2.VideoCapture(0)
lower_yellow = np.array([20, 50, 30])
upper_yellow = np.array([100, 255, 255])
gaussian =7
num = 0
def video2frame(cap):
ret, frame = cap.read()
image = frame
return image
def masking(image, num):
#ノイズがやなのでぼかす
image = cv2.GaussianBlur(image, (gaussian, gaussian), 0)
hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
#今回は黄色のバックカラーを想定
aim2masking_color = cv2.inRange(hsv, lower_yellow, upper_yellow)
mask = cv2.bitwise_not(image, image, mask=aim2masking_color)
masked_upstate = cv2.bitwise_and(image, mask)
mask_img = cv2.cvtColor(aim2masking_color, cv2.COLOR_GRAY2RGB)
#ココでアルファチャンネルを追加(失敗)
'''
mask_b, mask_g, mask_r = cv2.split(mask_img)
alpha_channel = np.ones(mask_b.shape, dtype=mask_b.dtype) * 50
#alpha_mask = cv2.merge((mask_b, mask_g, mask_r, alpha_channel))
obj_image = cv2.addWeighted(masked_upstate, 1, alpha_mask, 1, 0)
'''
masked_replaced_white = cv2.addWeighted(masked_upstate, 1, mask_img, 1, 0)
cv2.imshow("test", masked_replaced_white)
#cv2.imwrite("test_%d.png"%(num), obj_image)
cv2.waitKey(1)
def main(cap, num):
masking(video2frame(cap), num)
while True:
main(cap, num)
num += 1
cap.release()
cv2.destroyAllWindows()-
気になる質問をクリップする
クリップした質問は、後からいつでもマイページで確認できます。
またクリップした質問に回答があった際、通知やメールを受け取ることができます。
クリップを取り消します
-
良い質問の評価を上げる
以下のような質問は評価を上げましょう
- 質問内容が明確
- 自分も答えを知りたい
- 質問者以外のユーザにも役立つ
評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。
質問の評価を上げたことを取り消します
-
評価を下げられる数の上限に達しました
評価を下げることができません
- 1日5回まで評価を下げられます
- 1日に1ユーザに対して2回まで評価を下げられます
質問の評価を下げる
teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。
- プログラミングに関係のない質問
- やってほしいことだけを記載した丸投げの質問
- 問題・課題が含まれていない質問
- 意図的に内容が抹消された質問
- 広告と受け取られるような投稿
評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。
質問の評価を下げたことを取り消します
この機能は開放されていません
評価を下げる条件を満たしてません
質問の評価を下げる機能の利用条件
この機能を利用するためには、以下の事項を行う必要があります。
- 質問回答など一定の行動
-
メールアドレスの認証
メールアドレスの認証
-
質問評価に関するヘルプページの閲覧
質問評価に関するヘルプページの閲覧
check解決した方法
+1
なんかいろいろやってたら上手くいきました。
ご協力ありがとうございました。
import cv2
import numpy as np
import sys
cap = cv2.VideoCapture(0)
lower_yellow = np.array([20, 50, 30])
upper_yellow = np.array([100, 255, 255])
gaussian =7
num = 0
def video2frame(cap):
ret, frame = cap.read()
image = frame
return image
def channels(image):
if len(image.shape) == 3:
height, width, channels = image.shape[:3]
print channels
else:
height, width = image.shape[:2]
channels = 1
print channels
def masking(image, num):
image = cv2.GaussianBlur(image, (gaussian, gaussian), 0)
h, w = image.shape[:2]
hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
aim2masking_color = cv2.inRange(hsv, lower_yellow, upper_yellow)
mask = cv2.bitwise_not(image, image, mask=aim2masking_color)
mask = cv2.bitwise_not(aim2masking_color)
bgr = cv2.split(image)
img_alpha = cv2.merge(bgr+[mask])
print mask
cv2.imshow("test", mask)
cv2.imwrite("test_%d.png"%(num), img_alpha)
cv2.waitKey(1)
def main(cap, num):
masking(video2frame(cap), num)
while True:
main(cap, num)
num += 1
cap.release()
cv2.destroyAllWindows()投稿
-
回答の評価を上げる
以下のような回答は評価を上げましょう
- 正しい回答
- わかりやすい回答
- ためになる回答
評価が高い回答ほどページの上位に表示されます。
-
回答の評価を下げる
下記のような回答は推奨されていません。
- 間違っている回答
- 質問の回答になっていない投稿
- スパムや攻撃的な表現を用いた投稿
評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。
+1
Pythonでopencvを使用していないので、正確な回答でないかもしれない事をお許しください。
アルファチャネルを使用する時は、確実な方法として一旦ファイルにimwriteで書き出し、読み込みの際(imread)にC++であればIMREAD_UNCHANGEDを引数に。pythonであれば引数に-1を渡せば読み込めると認識しています。
念のためググってみたサイト（検証していないので正解かどうかは判断できませんが、参考になれば）は以下の通り。
投稿
-
回答の評価を上げる
以下のような回答は評価を上げましょう
- 正しい回答
- わかりやすい回答
- ためになる回答
評価が高い回答ほどページの上位に表示されます。
-
回答の評価を下げる
下記のような回答は推奨されていません。
- 間違っている回答
- 質問の回答になっていない投稿
- スパムや攻撃的な表現を用いた投稿
評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。