Skip to content

tangy1227/UNet-Audio-Source-Separation

Repository files navigation

Audio Source Separation Based on U-Net

Spleeter Baseline

U-Net based audio source separation by Deezer

Modification

Experiment with different loss functions to compare their performance. Different spectro losses: mel-spectrogram and magnitude. Different distance costs: L1 and MSE.

Files Note:

logs contains training loss log with tensorboard

data.py dataloader, experimented with both original baseline Spleeter dataloader and Open-Unmix dataloader dataloader

display_mask.ipynb contains what model outputs (a ratio mask) and spectrogram comparison between ground truth audio source (vocal) with separated audio source (mixture * ratio mask)

run.py model training code

splitter.py inferencing code

Project Poster

Poster

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published