二つの要素を比較し小さい方を0にしバイナリマスクを作成したい

Question

前提・実現したいこと

Pythonのプログラミングでの質問です
音源分離のためのバイナリマスクの生成で躓いています。
下記の計算式でバイナリマスク MA, MB を作成したいです
MA(t, ω)＝｛1 , A1(t, ω) > A2(t, ω)
．　　　　　 0 , A1(t, ω) ≤ A2(t, ω)
XA(t, ω) = MA(t, ω) ∗ X1(t, ω),
MB(t, ω)＝｛1 , A1(t, ω) > A2(t, ω)
・　　　　　 0 , A1(t, ω) ≤ A2(t, ω)
XB(t, ω) = MB(t, ω) ∗ X2(t, ω),
正直全く持ってやり方が分からない状態なので、どうすればいいかご教授願いたいです。
お手柔らかにお願いします

ソースコード

# モジュールのインポート
import numpy as np
import matplotlib.pyplot as plt
import soundfile as sf
import scipy
from scipy import signal as sg

# 音声の読み込み
x1, fs = sf.read('stereo_R.wav')
x2, fs = sf.read('stereo_L.wav')

# スペクトログラムの計算
f, t, X1 = sg.stft(x1, fs=fs)
f, t, X2 = sg.stft(x2, fs=fs)

# スペクトログラムの絶対値の計算
A1 = np.abs(X1)
A2 = np.abs(X2)

# バイナリマスクを作成(ここが分からない)
"""
ここに下記の式を適用したい
MA(t, ω)＝｛1 , A1(t, ω) > A2(t, ω)
　　　　　　 0 , A1(t, ω) ≤ A2(t, ω)

MB(t, ω)＝｛1 , A1(t, ω) > A2(t, ω)
　　　　　　0 , A1(t, ω) ≤ A2(t, ω)
"""
XA = Ma*X1
XB = Mb*X2

# 逆ISFT
_, Xa = sg.istft(XA, fs)
_, Xb = sg.istft(XB, fs)

# グラフに表示
# - スペクトログラム
plt.figure('Spectrogram1')
plt.imshow( np.log10(np.abs(X1)), origin='low', cmap='jet', aspect="auto")

plt.figure('Spectrogram2')
plt.imshow( np.log10(np.abs(X2)), origin='low', cmap='jet', aspect="auto")

plt.figure('SpectrogramA')
plt.imshow( np.log10(np.abs(Xa)), origin='low', cmap='jet', aspect="auto")

plt.figure('SpectrogramB')
plt.imshow( np.log10(np.abs(Xb)), origin='low', cmap='jet', aspect="auto")

plt.show()

試したこと

二分化できればいいのかなと思いcv2.thresholdを使ってみたのでしたが案の定できず、正直全く持ってやり方が分からない状態です

Answer 1

音声データの処理についてはあまりよく知らないので違っているかもしれませんが、

MA = (A1 > A2).astype(np.int)
MB = (A1 > A2).astype(np.int)

はご希望のものになりませんか? (MAとMBは同じ?)

A1 > A2は各要素ごとの比較結果の boolean の配列になります。
astype(np.int)で整数に変換してやるとTrue→1, False→0に変換されます。

Comments

[NEK0o0]

助かりました！
そもそも0.1で判別するならTrue,Falseで判別すればいいですもんね！
完全に失念していました！ありがとうございます！