diff --git a/setup.py b/setup.py index 34d0cb3..b70dcdc 100644 --- a/setup.py +++ b/setup.py @@ -14,7 +14,7 @@ __license__ = 'MIT License' # 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 0000000..08c2ffc --- /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 0000000..5f08527 --- /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 0000000..330aee0 --- /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" + } + } +}