|
217 | 217 | "fg_mask_arr = fg_mask.mask\n", |
218 | 218 | "\n", |
219 | 219 | "# Multiply the masks to the magnitude spectrogram\n", |
| 220 | + "mix.stft()\n", |
220 | 221 | "mix_mag_spec = mix.magnitude_spectrogram_data\n", |
221 | 222 | "bg_no_phase_spec = mix_mag_spec * bg_mask_arr\n", |
222 | 223 | "fg_no_phase_spec = mix_mag_spec * fg_mask_arr\n", |
|
394 | 395 | } |
395 | 396 | ], |
396 | 397 | "source": [ |
397 | | - "print('REPET Background without Phase')\n", |
| 398 | + "print('REPET Background with Phase')\n", |
398 | 399 | "_ = bg_phase.embed_audio()\n", |
399 | 400 | "\n", |
400 | | - "print('REPET Foreground without Phase')\n", |
| 401 | + "print('REPET Foreground with Phase')\n", |
401 | 402 | "_ = fg_phase.embed_audio()" |
402 | 403 | ] |
403 | 404 | }, |
|
605 | 606 | "source": [ |
606 | 607 | "bss_eval = nussl.evaluation.BSSEvalScale(\n", |
607 | 608 | " true_sources_list=[gt_bg, gt_fg],\n", |
608 | | - " estimated_sources_list=[repet_bg, repet_bg]\n", |
| 609 | + " estimated_sources_list=[repet_bg, repet_fg]\n", |
609 | 610 | ")\n", |
610 | 611 | "repet_eval = bss_eval.evaluate()\n", |
611 | 612 | "\n", |
|
1076 | 1077 | "name": "python", |
1077 | 1078 | "nbconvert_exporter": "python", |
1078 | 1079 | "pygments_lexer": "ipython3", |
1079 | | - "version": "3.7.6" |
| 1080 | + "version": "3.8.1" |
1080 | 1081 | } |
1081 | 1082 | }, |
1082 | 1083 | "nbformat": 4, |
|
0 commit comments