背景色を素早く除去したい

Question

OpenCVを使い、画像処理を行なっているものです。

処理したい画像は白色の背景をしています。

そして、この白色が画像処理の邪魔になってしまっています。

def deleteWhite(img):#imgはnumpy.array
        for row in img:
            for pixel in row:
                if all(x >= 250 for x in pixel):
                    np.delete(img, np.where(pixel))
        return img

上記のように作ってみましたが、時間掛かりすぎてタイムエラーになりました。

短時間で白色のみ除去するにはどうしたらよいのでしょうか？

Answer 1

cv2.inRangeで指定範囲の色のみのマスク画像を生成し、cv2.bitwise_andで元画像とマスク画像を合成する手順はいかがでしょうか？
参考：Python3 OpenCV3で指定した色のみを抽出して表示する

画素（配列要素）自体を削除すると画像として成り立たなくなるため、「ほぼ白」の領域を黒にする例です。

import cv2
img = cv2.imread('lena.png', cv2.IMREAD_COLOR)

# 「ほぼ白」の色範囲
lower = np.array([192,192,192]) 
upper = np.array([255,255,255])
#マスク作成
img_mask = cv2.inRange(img, lower, upper)     # 「ほぼ白」のみの抽出マスク
img_mask = cv2.bitwise_not(img_mask,img_mask) # 反転＝「ほぼ白」以外の抽出マスク
# 「ほぼ白」以外を抽出→「ほぼ白」を黒に
img = cv2.bitwise_and(img, img, mask=img_mask)

cv2.imshow('img', img)
cv2.waitKey(0)
cv2.destroyAllWindows()

Comments

[退会済みユーザー]

lower = np.array([0,0,0])
upper = np.array([250,250,250])
#マスク作成
img_mask = cv2.inRange(img, lower, upper)
#マスクとの共通領域のみ抽出
img_ = cv2.bitwise_and(img, img, mask=img_mask)

こんな感じでつくってみましたが、これだとlistの中から除去されません。
「白色のないlist」をつくるにはどうしたらよいのでしょうか？

[can110]

すみません。やりたいことを読み違えていたようです。
とりあえず白を黒にする例のみ追記しておきます。

[退会済みユーザー]

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