annotate mirex2012-matlab/doMultiF0.m @ 2:8017dd4a650d

Add MIREX 2012 code
author Chris Cannam
date Wed, 19 Mar 2014 09:09:23 +0000
parents
children e92376d450b0
rev   line source
Chris@2 1 function [] = doMultiF0(inputFile,outputFile)
Chris@2 2
Chris@2 3
Chris@2 4 % Transcribe file
Chris@2 5 fprintf('%s',['Preprocessing............']);
Chris@2 6 [ph pz sumY] = transcriptionMultipleTemplates(inputFile,12,1.1,1.3);
Chris@2 7 fprintf('\n');
Chris@2 8 fprintf('%s',['Postprocessing...........']);
Chris@2 9 pianoRoll = repmat(sumY,88,1).*pz(1:88,:);
Chris@2 10 pianoRoll = pianoRoll';
Chris@2 11 for j=[1:15 74:88] pianoRoll(:,j)=0; end;
Chris@2 12 pianoRoll = medfilt1(pianoRoll,3);
Chris@2 13
Chris@2 14
Chris@2 15 % Max polyphony 5
Chris@2 16 [B,IX] =sort(pianoRoll,2,'descend');
Chris@2 17 tempPianoRoll = zeros(size(pianoRoll,1),88);
Chris@2 18 for j=1:size(pianoRoll,1) for k=1:5 tempPianoRoll(j,IX(j,k)) = B(j,k); end; end;
Chris@2 19 pianoRoll = tempPianoRoll;
Chris@2 20
Chris@2 21
Chris@2 22 % Expand piano-roll and perform thresholding
Chris@2 23 expandedPianoRoll = zeros(4*size(pianoRoll,1),88);
Chris@2 24 for j=1:4*size(pianoRoll,1)
Chris@2 25 expandedPianoRoll(j,:) = pianoRoll(floor((j-1)/4)+1,:);
Chris@2 26 end;
Chris@2 27 finalPianoRoll = (expandedPianoRoll>4.8)';
Chris@2 28
Chris@2 29
Chris@2 30 % Create output and perform minimum duration pruning
Chris@2 31 auxPianoRoll = diff([zeros(1,88); finalPianoRoll'; zeros(1,88);],1); k=0;
Chris@2 32 for i=1:88
Chris@2 33 onsets = find(auxPianoRoll(:,i)==1);
Chris@2 34 offsets = find(auxPianoRoll(:,i)==-1);
Chris@2 35 for j=1:length(onsets)
Chris@2 36 if((offsets(j)/100-0.01) - (onsets(j)/100) > 0.05)
Chris@2 37 k=k+1;
Chris@2 38 nmat(k,1) = onsets(j)/100;
Chris@2 39 nmat(k,2) = offsets(j)/100-0.01;
Chris@2 40 nmat(k,3) = 27.5*2.^((( i-1)*10 )/120);
Chris@2 41 end;
Chris@2 42 end;
Chris@2 43 end;
Chris@2 44 nmat = sortrows(nmat,1);
Chris@2 45
Chris@2 46
Chris@2 47 % Print output
Chris@2 48 fid=fopen(outputFile,'w');
Chris@2 49 for i=1:size(nmat,1)
Chris@2 50 fprintf(fid,'%.2f\t%.2f\t%.2f\n',nmat(i,1),nmat(i,2),nmat(i,3));
Chris@2 51 end;
Chris@2 52 fclose(fid);
Chris@2 53 fprintf('%s','done');
Chris@2 54 fprintf('\n');
Chris@2 55
Chris@2 56
Chris@2 57 exit;