use MMSE TDOA instead of mode of TDOA distribution

mim · Aug 14, 2015 · 96452a0 · 96452a0
1 parent 23642ce
commit 96452a0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/messlMultichannel.m b/messlMultichannel.m
@@ -141,7 +141,8 @@
 
 if (Np > Ch) && useConsistentTdoa
     % Compute global TDOA at each mic, re-derive pairwise ITDs
-    perPairTdoa = tau(squeeze(argmax(pTauIInit,2)))';
+    perPairTdoa = tau(squeeze(argmax(pTauIInit,2)))';   % posterior mode
+    % perPairTdoa = squeeze(sum(bsxfun(@times, tau, pTauIInit), 2) ./ sum(pTauIInit,2))';  % posterior mean
     [perMicTdoa tauPosInit] = perMicTdoaLs(perPairTdoa(:,1:end-garbageSrc), channelPairs);
 else
     tauPosInit = [];
@@ -298,11 +299,13 @@
 
 % Compute per-mic TDOAs
 pTauI = cat(3, ipdParams.p_tauI);
-perPairTdoa = tau(squeeze(argmax(pTauI,2)))';
+perPairTdoa = tau(squeeze(argmax(pTauI,2)))';   % posterior mode
+% perPairTdoa = squeeze(sum(bsxfun(@times, tau, pTauI), 2) ./ sum(pTauI,2))';  % posterior mean
 perMicTdoa = perMicTdoaLs(perPairTdoa(:,1:end-garbageSrc), channelPairs);
 
 params = struct('ipdParams', ipdParams, 'ildParams', ildParams, ...
-    'spParams', spParams, 'perMicTdoa', perMicTdoa);
+    'spParams', spParams, 'perMicTdoa', perMicTdoa, ...
+    'channelPairs', channelPairs, 'tau', tau);
 
 % Compute hard masks, potentially using the MRF model
 [~,~,~,hardSrcs] = messlMrfApply(nuIld, nuIpd, p_lr_iwt, mrfCompatPot, mrfHardCompatExp, mrfLbpIter, 'max');

diff --git a/perMicTdoaLs.m b/perMicTdoaLs.m
@@ -42,7 +42,7 @@
         break
     end    
 
-    failed = (mean(keep,1) < minKeepPairFrac) | (sum(keep,1) <= Ch);
+    failed = (mean(keep,1) < minKeepPairFrac) | (sum(keep,1) <= Ch+1);
     unRobustObs = any(keep ~= lastKeep, 1) & ~failed;
 end
 %plot(estPerPair, perPair(1:end-1,:), '.')