sigsep / open-unmix-pytorch

Open-Unmix - Music Source Separation for PyTorch
https://sigsep.github.io/open-unmix/
MIT License
1.27k stars 191 forks source link

Can't just evaluate one goal #88

Closed BYR-HZ closed 3 years ago

BYR-HZ commented 3 years ago

when I run “python openunmix/evaluate.py --root /local/musdb18 --targets vocals --model my_model --residual acc” appear: Traceback (most recent call last): File "openunmix/evaluate.py", line 260, in results.add_track(scores) File "/home/bianyuren/anaconda3/envs/umx-gpu-pytorch_1_8/lib/python3.8/site-packages/museval/aggregate.py", line 183, in add_track self.df = self.df.append(track.df, ignore_index=True) File "/home/bianyuren/anaconda3/envs/umx-gpu-pytorch_1_8/lib/python3.8/site-packages/museval/aggregate.py", line 113, in df return json2df(simplejson.loads(self.json), self.track_name) File "/home/bianyuren/anaconda3/envs/umx-gpu-pytorch_1_8/lib/python3.8/site-packages/museval/aggregate.py", line 413, in json2df df = pd.melt( File "/home/bianyuren/anaconda3/envs/umx-gpu-pytorch_1_8/lib/python3.8/site-packages/pandas/core/reshape/melt.py", line 64, in melt raise KeyError( KeyError: "The following 'id_vars' are not present in the DataFrame: ['name', 'time']"

faroit commented 3 years ago

@BYR-HZ will have to look into this to reproduce. Do you have some more information or advice how to quickly reproduce this?

BYR-HZ commented 3 years ago

I've only trained one Vocals model and wanted to be able to evaluate it。When I follow your instructions run“python openunmix/evaluate.py --root /local/musdb18 --targets vocals --model my_model --residual acc”。 Error occurred: raise KeyError(KeyError: "The following 'id_vars' are not present in the DataFrame: ['name', 'time']" Because there is no correct data format to pass into museval。 When I change line 337 of the model.py
estimates_dict["residual"] = estimates[:, -1, ...]-->estimates_dict["accompaniment"] = estimates[:, -1, ...] The problem is solved.