Python＋OpenCVで色抽出すると、グリーンバックの謎が解けた #Python

目的

いきなりむさいおじさん画像で申し訳ありません。

「赤黄青緑の4色の木の玉を、それぞれ抽出できないか」、やってみました。

Python歴OpenCV歴ともに1ヶ月程度ですので、拙い点はぜひご指摘下さい（本当に）。

フィルタのコード

以下は、

指定した色相（hue）のピクセル（hTarget）を明るくして
該当しないピクセル（hMask2）をグレースケールにする

コードです。

filters.py

...

def hueMask(src, dst, hue, hueRange):
    src = cv2.cvtColor(src, cv2.COLOR_BGR2HSV)
    h, s, v = cv2.split(src)
    hOrg = h.copy()
    hTarget = h.copy()

    cv2.threshold(hTarget, hue + hueRange, hue, cv2.THRESH_TOZERO_INV, hTarget)
    # 「src(x,y)がthreshより大きければ、dst(x,y)は0」

    # 0
    # 0
    # 0
    # thresh
    # src
    # src
    # src
    # src
    # src

    cv2.threshold(hTarget, hue - hueRange, hue, cv2.THRESH_BINARY, hTarget)
    # 「src(x,y)がthreshより小さければ、dst(x,y)は0」

    # 0(src)
    # 0(src)
    # 0(src)
    # src
    # src
    # thresh
    # 0
    # 0
    # 0

    # 結果

    # 0
    # 0
    # 0
    # thresh
    # src
    # thresh
    # 0
    # 0
    # 0

    # 蛍光灯の光（黄）や髪の毛（青）を除外するため、
    # 極端に彩度の低いピクセルをターゲット範囲から除外する
    sNotVeryLow = s.copy() # 極端に彩度が低くないところ
    # 彩度が31より高いならターゲット範囲（255）に入れる。
    # さもなくば非ターゲット範囲（0）。
    cv2.threshold(sNotVeryLow, 31, 255, cv2.THRESH_BINARY, sNotVeryLow)

    # hTargetとvNotHighlightの論理積が新しいhTarget
    cv2.bitwise_and(hTarget, vNotVeryLow, hTarget)

    # 明度画像のコピーをとる。
    vBrightened = v.copy()
    # それに+96のガンマ補正をかけ、明るくする
    cv2.addWeighted(v, 0.625, v, 0.0, 96, vBrightened)
    # 明度画像のターゲット範囲のみ、ガンマ補正済み明度画像を入れる
    cv2.bitwise_and(vBrightened, 255, v, hTarget)

    hMask2 = h.copy()
    # hTarget（1チャンネル画像）の該当ピクセルが0のとき、
    # hMask2（1チャンネル画像）の該当ピクセルを255にセットする。
    # さもなくば、0にセットする。
    # 要するにhMask2はhTargetマスク画像を反転させたもの。
    cv2.compare(hTarget, 0, cv2.CMP_EQ, hMask2)

    cv2.bitwise_and(s, 0, s, hMask2) # 論理積

    cv2.merge((hOrg, s, v), src)
    cv2.cvtColor(src, cv2.COLOR_HSV2BGR, dst)

...

なんだか変数の名前が汚いですが・・・これで任意の色相を抽出できるようになりました。

アプリケーションのコード

次に4色つまり赤黄青緑の色相を抽出するようにしてみます。

App.py

...
class App(object):
    def __init__(self):
        ...
        self._shouldHueMask                = False
        ...
    def run(self):
        ...
        while self._windowManager.isWindowCreated:
            ...
            if self._shouldHueMask:
                filters.hueMask(frame, frame, self._hue, self._hueRange)
            ...
    def onKeypress(self, keycode):
        ...
        elif keycode == ord('B'): # 青
            self._hue      = 110
            self._hueRange = 10
            self._shouldHueMask = \
                not self._shouldHueMask
        elif keycode == ord('G'): # 緑
            self._hue      = 70
            self._hueRange = 25
            self._shouldHueMask = \
                not self._shouldHueMask
        elif keycode == ord('R'): # 赤
            self._hue      = 5
            self._hueRange = 5
            self._shouldHueMask = \
                not self._shouldHueMask
        elif keycode == ord('Y'): # 黄
            self._hue      = 30
            self._hueRange = 15
            self._shouldHueMask = \
                not self._shouldHueMask
        ...
if __name__=="__main__":
    App().run()