Mercurial > hg > silvet
diff testdata/evaluation/run-nonorm.sh @ 267:a8c05709e486
Merge from branch "norm"
author | Chris Cannam |
---|---|
date | Wed, 23 Jul 2014 17:51:58 +0100 |
parents | 6b7e78cf96c9 |
children | 40684b5dd478 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/testdata/evaluation/run-nonorm.sh Wed Jul 23 17:51:58 2014 +0100 @@ -0,0 +1,121 @@ +#!/bin/sh + +trios_path="/home/cannam/Music/TRIOS_dataset" +yc="/home/cannam/code/may/bin/yc" + +if [ ! -d "$trios_path" ]; then + echo "TRIOS dataset directory $trios_path not found, giving up" + exit 1 +fi + +if ! "$yc" -v ; then + echo "Failed to run Yeti compiler yc at $yc_path, giving up"; +fi + +if ! sonic-annotator -v ; then + echo "Failed to run sonic-annotator (not in PATH?), giving up" + exit 1 +fi + +rdffile="../../silvet.n3" +if [ ! -f "$rdffile" ] ; then + echo "Failed to find plugin RDF file at $rdffile, giving up" + exit 1 +fi + +case "$trios_path" in +*\ *) echo "TRIOS dataset path $trios_path has a space in it, this script won't handle that"; exit 1;; +esac + +( cd ../.. ; make -f Makefile.linux ) || exit 1 + +VAMP_PATH=../.. +export VAMP_PATH + +outfile="/tmp/$$" + +tmpwav="/tmp/$$norm.wav" + +instfile="/tmp/$$instruments.txt" + +transfile="/tmp/$$transform.ttl" + +trap 'rm -f "$outfile" "$tmpwav" "$instfile" "$transfile" "$outfile.lab"' 0 + +# Use the mix and single-instrument non-synthetic files for a (varied) +# subset of the TRIOS dataset. Omit percussion (but we still use the +# Take Five example, even though it's largely percussion, because it's +# a good test of how we do with other instruments in the presence of +# percussion). +infiles=`find "$trios_path" -name \*.wav -print | egrep '(mozart|lussier|take_five)' | grep -v _syn.wav | grep -v kick.wav | grep -v ride.wav | grep -v snare.wav` + +grep Piano "$rdffile" | sed 's/^.*( *//' | sed 's/ *).*$//' | sed 's/ "/\n/g' | sed 's/"//g' | tr '[A-Z]' '[a-z]' | tail -n +2 | cat -n > "$instfile" + +instrument_for() { + filename="$1" + base=`basename "$filename" .wav` + if [ "$base" = "saxophone" ]; then base="tenorsax"; fi + instrument_no=`grep "$base" "$instfile" | awk '{ print $1; }'` + if [ -z "$instrument_no" ] || [ -z "$base" ]; + then echo 0 + else echo "$instrument_no" + fi +} + +echo +echo "Input files are:" +echo $infiles | fmt -1 + +time for infile in $infiles; do + + echo + echo "Evaluating for file $infile..." + + instrument=`instrument_for "$infile"` + case "$instrument" in + [0-9]*) ;; + *) echo "Instrument extraction failed for infile $infile -- not even default multi-instrument setting returned?"; exit 1;; + esac + + piece=`basename \`dirname "$infile" \`` + arrangement=`basename "$infile" .wav` + + echo + echo "For piece $piece, arrangement $arrangement, using instrument $instrument..." + + # unlike run.sh we don't normalise the input here + cp "$infile" "$tmpwav" + + # generate the transform by interpolating the instrument parameter + cat transform.ttl | sed "s/INSTRUMENT_PARAMETER/$instrument/" > "$transfile" + + sonic-annotator \ + --writer csv \ + --csv-one-file "$outfile" \ + --csv-force \ + --transform "$transfile" \ + "$tmpwav" + + cat "$outfile" | \ + sed 's/^[^,]*,//' | \ + while IFS=, read start duration frequency level label; do + end=`echo "$start $duration + p" | dc` + echo -e "$start\t$end\t$frequency" + done > "$outfile.lab" + + for ms in 50 100; do + mark="" + if [ "$ms" = "50" ]; then mark=" <-- main $piece/$arrangement"; fi; + echo + echo "Validating against ground truth at $ms ms:" + "$yc" ../scripts/evaluate_lab.yeti "$ms" "../TRIOS-groundtruth/$piece/$arrangement.lab" "$outfile.lab" | sed 's,$,'"$mark"',' + echo + echo "Validating against MIREX submission at $ms ms:" + "$yc" ../scripts/evaluate_lab.yeti "$ms" "../TRIOS-mirex2012-matlab/$piece/$arrangement.lab" "$outfile.lab" + echo + echo "Validating MIREX against ground truth at $ms ms": + "$yc" ../scripts/evaluate_lab.yeti "$ms" "../TRIOS-groundtruth/$piece/$arrangement.lab" "../TRIOS-mirex2012-matlab/$piece/$arrangement.lab" + done; + + echo +done