ヒートマップの色から元の画像をぼかす方法について

Question

ヒートマップの特定の色を指定し、その色と同じ位置を別の画像でぼかす

ヒートマップの画像のRGB値のうちBのみ指定した場合は上手くぼかしたものと入れ替えれたが、RとGの値両方指定し、ぼかしたものと入れ替えようとしたらプログラム自体は終了しているのですが入れ替わった画像が出力されませんでした。

やりたい処理としましては、全体的にぼかし処理を加えた画像と何も処理していない元の画像を用意し、下のようなヒートマップ画像のRGB値のRとGが０となっている部分(主に濃い青色の部分)のみをぼかしている画像と入れ替え特定の部分をぼかした画像を作成したいと思っています。

しかし、上記でも触れたようにBのみの値をを指定した場合ですと上手く特定の部分をボカせたのですが、RとGが０の部分のみぼかすといったことができないでいます。

先ほど回答いただいたように訂正したところ青ではなく画像でいうと真っ赤な部分がぼけてしまいました。

---

追記
下記回答のおかげで差し替えることができました。
しかし、画像は以下のようになり部分的に上手くいってないような感じがします。
画像的には大部分ができているので問題はないのですがなるべく綺麗に差し替えれた方がいいのでもし、もう少し改善できるようであればご教授願いたいです。
画像はぼかしではなくどこが再帰られているか見やすくするために緑の画像に差し替えたものです。

該当のソースコード

import cv2
import numpy as np
import math

x=None
y=None
xmax=224
ymax=224
xcenter=112
ycenter=112
Rfigs=[0,128]
Gfigs=[0,128]

img = cv2.imread(r"/Users/1.jpg")#ぼかしたい画像
imgheat = cv2.imread(r"/Users/2jpg")#ヒートマップ画像
blur2 = cv2.imread(r"/Users/3.jpg")#ぼかしてある画像

img2 = cv2.resize(img,(224,224))
blur2 = cv2.resize(blur2,(224,224))
count = 0

for Rfig  in Rfigs:
    for Gfig  in Gfigs:
        for x in range(xmax):
            for y in range(ymax):
                pixelValue = imgheat[x,y,[0]]
                if pixelValue == Rfig and Gfig:
                   if Rfig==0 and Gfig==0:
                      img2[x,y,[0]] = blur2[x,y,[0]]
                      img2[x,y,[1]] = blur2[x,y,[1]]
                      img2[x,y,[2]] = blur2[x,y,[2]]

        if Rfig==0 and Gfig==0:
           cv2.imwrite(r"/Users/gazou.jpg',img2)

動作したBの値飲みを指定した時のコード

for Bfig  in Bfigs:
    for x in range(xmax):
        for y in range(ymax):
            pixelValue = imgheat[x,y,[0]]
            if pixelValue == Bfig:
               if Bfig==128:
                  img2[x,y,[0]] = blur3[x,y,[0]]
                  img2[x,y,[1]] = blur3[x,y,[1]]
                  img2[x,y,[2]] = blur3[x,y,[2]]

    if Bfig==128:
        cv2.imwrite(r"/Users/gazou.jpg',img2)

###回答から変更したコード(変更部分のみ)

b, g, r = cv2.split(imgheat)

# heatmap の red == 0 かつ green == 0 の画素は差し替える
img2[(g == 0) & (r == 0)] = blur1[(g == 0) & (r == 0)]
img2[(b == 128)] = blur1[(b == 128)]
cv2.imwrite(r"/Users/gazou.jpg',img2)

Comments

[tiitoi]

どのような処理を行いたかったのか文章で説明していただけますか？(コードの各変数の説明等がないため、処理の概要がわからない)

[satoUmino]

わかりにくかもしれませんが質問の方に説明加えました。

Answer 1

やりたい処理としましては、全体的にぼかし処理を加えた画像と何も処理していない元の画像を用意し、下のようなヒートマップ画像のRGB値のRとGが０となっている部分(主に濃い青色の部分)のみをぼかしている画像と入れ替え特定の部分をぼかした画像を作成したいと思っています。

(R, G, B) = (0, 0, 128) の画素は用意した別の画像の画素と差し替えるということであれば、以下のようにすればいいと思います。

import cv2
import numpy as np

# 画像の代わり
img1 = np.random.randint(0, 256, (500, 500, 3), dtype=np.uint8)

# 差し替える画像の代わり
img2 = np.random.randint(0, 256, (500, 500, 3), dtype=np.uint8)

# ヒートマップ画像の代わり
heatmap = np.random.randint(0, 256, (500, 500, 3), dtype=np.uint8)

# heatmap をチャンネルごとに分解する。
b, g, r = cv2.split(heatmap)

# heatmap の (R, G, B) = (0, 0, 128) の画素は差し替える
mask = (r == 0) & (g == 0) & (b == 128)
img1[mask] = img2[mask]

cv2.imwrite("output.png", img)

Comments

[satoUmino]

上記のコードにて部分的にぼかすことはできましたが、なぜかR=０、G=0、B=128の部分のみ差し替えができませんでした。
なので、R=０とG=0の時に差し替えるコードに加えて、B＝128の部分を差し替えるコードを付け加え差し替えできていなかった部分を差し替えるようにしました。
しかし、一部上手く差し替えることができていないような部分があるため改善できるようであればコメントいただけると助かります。

[tiitoi]

質問のコードを編集しました。画像 img のうち、(R, G, B) = (0, 0, 128) の画素だけ取り出す場合は、img[(r == 0) & (g == 0) & (b == 128)] です。

Answer 2

   if pixelValue == Rfig and Gfig:             

↓

   if pixelValue == Rfig or pixelValue == Gfig:             

ではないでしょうか。

Comments

[satoUmino]

回答ありがとうございます。
試してみたところ入れ替えてぼかすことはできたのですが、ヒートマップ上のRとGが０ではない部分が変わってしまいました。