mhrice / RemFx

General Purpose Audio Effect Removal
https://csteinmetz1.github.io/RemFX/
Apache License 2.0
92 stars 4 forks source link

How to use my own dataset? #46

Open SunHaoRanCN opened 4 weeks ago

SunHaoRanCN commented 4 weeks ago

Hello, I am trying to use RemFx for audio effects removing. However I 'd like to retrain the model with my own dataset. How should I manage my dataset? What should I change for the code?

mhrice commented 4 weeks ago

Thanks for your interest! If you wanted to train with a custom dataset, you'd need to set the DATASET_ROOT env variable, and also modify locate_files() in remfx/datasets.py to find the right files from your dataset.