diff options
| -rw-r--r-- | .gitignore | 3 | ||||
| -rwxr-xr-x | datasets/dataset.pl | 4 | ||||
| -rw-r--r-- | datasets/silence.sh | 0 | ||||
| -rwxr-xr-x | latest.pl | 6 | ||||
| -rwxr-xr-x | process_fns | 109 |
5 files changed, 119 insertions, 3 deletions
@@ -110,3 +110,6 @@ node_modules/ results/old_checkpoints/ +mgb +run_* + diff --git a/datasets/dataset.pl b/datasets/dataset.pl index edde43f..a6eb984 100755 --- a/datasets/dataset.pl +++ b/datasets/dataset.pl @@ -25,6 +25,10 @@ sub process($) { return; } + if ($ext eq 'mp3') { + system('ffmpeg', '-i', $filename, $name . '.wav') + $filename = $name . '.wav'; + } my $soxi = `soxi $filename`; my @lines = split("\n", $soxi); diff --git a/datasets/silence.sh b/datasets/silence.sh new file mode 100644 index 0000000..e69de29 --- /dev/null +++ b/datasets/silence.sh @@ -69,9 +69,9 @@ sub process($){ } $epoch[0] =~ /(\d+)/; - if ($1 < 4) { - return 0; - } + #if ($1 < 4) { + # return 0; + #} if (-e "output/" . $name . ".mp3") { return 1; diff --git a/process_fns b/process_fns new file mode 100755 index 0000000..ad47a28 --- /dev/null +++ b/process_fns @@ -0,0 +1,109 @@ +workon samplernn + +function generate () { + exp_name=$1 + n_samples=$2 + sample_rate=$3 + duration=$4 + let sample_length=$3*$4 + + echo "" + echo "###################################################" + echo "###################################################" + echo "###################################################" + echo "" + echo ">> generating $exp_name" + echo "" + echo "###################################################" + echo "###################################################" + echo "###################################################" + echo "" + python generate.py \ + --exp $exp_name --dataset $exp_name \ + --frame_sizes 8 2 \ + --n_rnn 2 --dim 1024 --q_levels 256 \ + --seq_len 1024 --batch_size 128 \ + --val_frac 0.1 --test_frac 0.1 \ + --sample_rate $sample_rate \ + --sample_length $sample_length \ + --keep_old_checkpoints False \ + --n_samples $n_samples \ + --epoch_limit 1 \ + --resume True + sleep 1 +} +function run () { + exp_name=$1 + epoch_limit=$2 + n_samples=$3 + sample_rate=$4 + duration=$5 + let sample_length=$4*$5 + + echo "" + echo ">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" + echo ">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" + echo ">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" + echo "" + echo ">> running $exp_name" + echo "" + echo ">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" + echo ">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" + echo ">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>" + echo "" + python train.py \ + --exp $exp_name --dataset $exp_name \ + --frame_sizes 8 2 \ + --n_rnn 2 --dim 1024 --q_levels 256 \ + --seq_len 1024 --batch_size 128 \ + --val_frac 0.1 --test_frac 0.1 \ + --sample_rate $sample_rate \ + --sample_length $sample_length \ + --n_samples $n_samples \ + --keep_old_checkpoints False \ + --epoch_limit $epoch_limit \ + --resume True + sleep 1 +} +function standard () { + dataset=$1 + run $1 6 6 44100 5 + ./latest.pl -l $1 +} +function quick () { + dataset=$1 + run $1 4 6 44100 5 + ./latest.pl -l $1 +} +function half () { + dataset=$1 + run $1 2 6 44100 5 + ./latest.pl -l $1 +} +function fast () { + dataset=$1 + run $1 1 6 44100 10 + ./latest.pl -l $1 +} +function emph () { + dataset=$1 + run $1 1 6 44100 10 + run $1 3 6 44100 5 + ./latest.pl -l $1 +} + +function start () { + dataset=$1 + cd datasets/ + perl dataset.pl "${dataset}.wav" + cd .. + quick $dataset +} +function start_standard () { + dataset=$1 + cd datasets/ + perl dataset.pl "${dataset}.wav" + cd .. + standard $dataset +} + |