diff src/matlab/nmf_divergence.m @ 0:c52bc3e8d3ad tip

user: boblsturm
branch 'default'
added README.md
added assets/.DS_Store
added assets/playButton.jpg
added assets/stopButton.png
added assets/swapButton.jpg
added data/.DS_Store
added data/fiveoctaves.mp3
added data/glock2.wav
added data/sinScale.mp3
added data/speech_female.mp3
added data/sweep.wav
added nimfks.m.lnk
added src/.DS_Store
added src/matlab/.DS_Store
added src/matlab/AnalysisCache.m
added src/matlab/CSS.m
added src/matlab/DataHash.m
added src/matlab/ExistsInCache.m
added src/matlab/KLDivCost.m
added src/matlab/LoadFromCache.m
added src/matlab/SA_B_NMF.m
added src/matlab/SaveInCache.m
added src/matlab/Sound.m
added src/matlab/SynthesisCache.m
added src/matlab/chromagram_E.m
added src/matlab/chromagram_IF.m
added src/matlab/chromagram_P.m
added src/matlab/chromsynth.m
added src/matlab/computeSTFTFeat.m
added src/matlab/controller.m
added src/matlab/decibelSliderReleaseCallback.m
added src/matlab/drawClickCallBack.m
added src/matlab/fft2chromamx.m
added src/matlab/hz2octs.m
added src/matlab/ifgram.m
added src/matlab/ifptrack.m
added src/matlab/istft.m
added src/matlab/nimfks.fig
added src/matlab/nimfks.m
added src/matlab/nmfFn.m
added src/matlab/nmf_beta.m
added src/matlab/nmf_divergence.m
added src/matlab/nmf_euclidean.m
added src/matlab/prune_corpus.m
added src/matlab/rot_kernel.m
added src/matlab/templateAdditionResynth.m
added src/matlab/templateDelCb.m
added src/matlab/templateScrollCb.m

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/matlab/nmf_divergence.m	Sun Jun 18 06:26:13 2017 -0400
@@ -0,0 +1,122 @@
+function [Y, cost] = nmf_divergence(V, W, varargin)
+
+if nargin > 2
+    nmf_params = varargin{1};
+    L = nmf_params.Iterations;
+    convergence = nmf_params.Convergence_criteria;
+    r = nmf_params.Repition_restriction;
+    p = nmf_params.Polyphony_restriction;
+    c = nmf_params.Continuity_enhancement;
+    rot = nmf_params.Continuity_enhancement_rot;
+    pattern = nmf_params.Diagonal_pattern;
+    endtime = nmf_params.Modification_application;
+    rng(nmf_params.Random_seed);
+elseif nargin == 2
+    L = 10;
+    convergence = 0;
+    r = -1;
+    p = -1;
+    c = -1;
+    pattern = 'Diagonal';
+    endtime = false;
+    rng('shuffle');
+end
+
+waitbarHandle = waitbar(0, 'Starting NMF synthesis...');
+
+cost=0;
+K=size(W, 2);
+M=size(V, 2);
+
+H=random('unif',0, 1, K, M);
+
+P=zeros(K, M);
+R=zeros(K, M);
+C=zeros(K, M);
+
+V = V+1E-6;
+W = W+1E-6;
+den = sum(W);
+
+for l=1:L-1
+    waitbar(l/(L-1), waitbarHandle, ['Computing approximation...Iteration: ', num2str(l), '/', num2str(L-1)])
+    
+    recon = W*H;
+    for mm = 1:size(H,2)
+      num = V(:,mm).*(1./recon(:,mm));
+      num2 = num'*W./den;
+      H(:,mm) = H(:, mm).*num2';
+    end
+    
+    if((r > 0 && ~endtime) || (r > 0 && endtime && l==L-1))
+        waitbar(l/(L-1), waitbarHandle, ['Repition Restriction...Iteration: ', num2str(l), '/', num2str(L-1)])
+        for k = 1:size(H, 1)
+            for m = 1:size(H, 2)
+                if(m>r && (m+r)<=M && H(k,m)==max(H(k,m-r:m+r)))
+                    R(k,m)=H(k,m);
+                else
+                    R(k,m)=H(k,m)*(1-(l+1)/L);
+                end
+            end
+        end
+        
+        H = R;
+    end
+    
+    if((p > 0 && ~endtime) || (p > 0 && endtime && l==L-1))
+        waitbar(l/(L-1), waitbarHandle, ['Polyphony Restriction...Iteration: ', num2str(l), '/', num2str(L-1)])
+      P = zeros(size(H));
+      mask = zeros(size(H,1),1);
+      for m = 1:size(H, 2)
+        [~, sortedIndices] = sort(H(:, m),'descend');
+        mask(sortedIndices(1:p)) = 1;
+        mask(sortedIndices(p+1:end)) = (1-(l+1)/L);
+        P(:,m)=H(:,m).*mask;
+      end
+      H = P;
+    end
+    
+    if((c > 0 && ~endtime) || (c > 0 && endtime && l==L-1))
+        waitbar(l/(L-1), waitbarHandle, ['Continuity Enhancement...Iteration: ', num2str(l), '/', num2str(L-1)])
+        switch pattern
+            case 'Diagonal'
+                C = conv2(H, rot_kernel( eye(c), rot ), 'same'); %Default
+            case 'Reverse'
+                C = conv2(H, flip(eye(c)), 'same'); %Reverse
+            case 'Blur'
+                C = conv2(H, flip(ones(c)), 'same'); %Blurring
+            case 'Vertical'
+                M = zeros(c, c); %Vertical
+                M(:, floor(c/2)) = 1;
+                C = conv2(P, M, 'same');
+        end
+        
+        H = C;
+    end
+    
+    if ~endtime
+        recon = W*H;
+        for mm = 1:size(H,2)
+            num = V(:,mm).*(1./recon(:,mm));
+            num2 = num'*W./den;
+            H(:,mm) = H(:, mm).*num2';
+        end
+    end
+    
+    cost(l)=KLDivCost(V, W*H);
+    if(l>3 && (abs(((cost(l)-cost(l-1)))/max(cost))<=convergence))
+        break;
+    end
+end
+
+fprintf('Iterations: %i/%i\n', l, L);
+fprintf('Convergence Criteria: %i\n', convergence*100);
+fprintf('Repitition: %i\n', r);
+fprintf('Polyphony: %i\n', p);
+fprintf('Continuity: %i\n', c);
+
+Y=H;
+Y = Y./max(max(Y)); %Normalize activations
+
+close(waitbarHandle);
+end
\ No newline at end of file