From fc1e3d9a2f620c481f9d876ee142be8787abc3d9 Mon Sep 17 00:00:00 2001
From: romi1502 <rhennequin@deezer.com>
Date: Fri, 27 Dec 2019 14:34:27 +0100
Subject: [PATCH] Added new config for processing until 16kHz

---
 setup.py                             |  2 +-
 spleeter/resources/2stems-16kHz.json | 28 +++++++++++++++++++++++++
 spleeter/resources/4stems-16kHz.json | 31 ++++++++++++++++++++++++++++
 spleeter/resources/5stems-16kHz.json | 31 ++++++++++++++++++++++++++++
 4 files changed, 91 insertions(+), 1 deletion(-)
 create mode 100644 spleeter/resources/2stems-16kHz.json
 create mode 100644 spleeter/resources/4stems-16kHz.json
 create mode 100644 spleeter/resources/5stems-16kHz.json

diff --git a/setup.py b/setup.py
index 34d0cb31..b70dcdc3 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@
 
 # Default project values.
 project_name = 'spleeter'
-project_version = '1.4.8'
+project_version = '1.4.9'
 tensorflow_dependency = 'tensorflow'
 tensorflow_version = '1.14.0'
 here = path.abspath(path.dirname(__file__))
diff --git a/spleeter/resources/2stems-16kHz.json b/spleeter/resources/2stems-16kHz.json
new file mode 100644
index 00000000..08c2ffca
--- /dev/null
+++ b/spleeter/resources/2stems-16kHz.json
@@ -0,0 +1,28 @@
+{
+    "train_csv": "path/to/train.csv",
+    "validation_csv": "path/to/test.csv",
+    "model_dir": "2stems",
+    "mix_name": "mix",
+    "instrument_list": ["vocals", "accompaniment"],
+    "sample_rate":44100,
+    "frame_length":4096,
+    "frame_step":1024,
+    "T":512,
+    "F":1536,
+    "n_channels":2,
+    "separation_exponent":2,
+    "mask_extension":"zeros",
+    "learning_rate": 1e-4,
+    "batch_size":4,
+    "training_cache":"training_cache",
+    "validation_cache":"validation_cache",
+    "train_max_steps": 1000000,
+    "throttle_secs":300,
+    "random_seed":0,
+    "save_checkpoints_steps":150,
+    "save_summary_steps":5,
+    "model":{
+            "type":"unet.unet",
+            "params":{}
+            }
+}
diff --git a/spleeter/resources/4stems-16kHz.json b/spleeter/resources/4stems-16kHz.json
new file mode 100644
index 00000000..5f085277
--- /dev/null
+++ b/spleeter/resources/4stems-16kHz.json
@@ -0,0 +1,31 @@
+{
+    "train_csv": "path/to/train.csv",
+    "validation_csv": "path/to/val.csv",
+    "model_dir": "4stems",
+    "mix_name": "mix",
+    "instrument_list": ["vocals", "drums", "bass", "other"],
+    "sample_rate":44100,
+    "frame_length":4096,
+    "frame_step":1024,
+    "T":512,
+    "F":1536,
+    "n_channels":2,
+    "separation_exponent":2,
+    "mask_extension":"zeros",
+    "learning_rate": 1e-4,
+    "batch_size":4,
+    "training_cache":"training_cache",
+    "validation_cache":"validation_cache",
+    "train_max_steps": 1500000,
+    "throttle_secs":600,
+    "random_seed":3,
+    "save_checkpoints_steps":300,
+    "save_summary_steps":5,
+    "model":{
+        "type":"unet.unet",
+        "params":{
+               "conv_activation":"ELU",
+               "deconv_activation":"ELU"
+        }
+    }
+}
diff --git a/spleeter/resources/5stems-16kHz.json b/spleeter/resources/5stems-16kHz.json
new file mode 100644
index 00000000..330aee08
--- /dev/null
+++ b/spleeter/resources/5stems-16kHz.json
@@ -0,0 +1,31 @@
+{
+    "train_csv": "path/to/train.csv",
+    "validation_csv": "path/to/test.csv",
+    "model_dir": "5stems",
+    "mix_name": "mix",
+    "instrument_list": ["vocals", "piano", "drums", "bass", "other"],
+    "sample_rate":44100,
+    "frame_length":4096,
+    "frame_step":1024,
+    "T":512,
+    "F":1536,
+    "n_channels":2,
+    "separation_exponent":2,
+    "mask_extension":"zeros",
+    "learning_rate": 1e-4,
+    "batch_size":4,
+    "training_cache":"training_cache",
+    "validation_cache":"validation_cache",
+    "train_max_steps": 2500000,
+    "throttle_secs":600,
+    "random_seed":8,
+    "save_checkpoints_steps":300,
+    "save_summary_steps":5,
+    "model":{
+        "type":"unet.softmax_unet",
+        "params":{
+               "conv_activation":"ELU",
+               "deconv_activation":"ELU"
+        }
+    }
+}