barvarvara/openunmix-guitar-ft

Дообученная модель Open-Unmix для извлечения акустической гитары. Адаптация выполнена на кастомном датасете (1296 миксов, n_fft=4096). Метрики: SDR=10.7 dB, SIR=17.24 dB, SAR=40.0 dB. Время инференса: ~X сек на CPU.

Воспроизведение инференса

import torch
model = torch.hub.load('sigsep/open-unmix-pytorch', 'umxhq')
ckpt = torch.load('openunmix_guitar.pt')
model.load_state_dict(ckpt['model_state_dict'])
model.eval()