- Notifications
You must be signed in to change notification settings - Fork1
Souden MVDR beamformer on GPU with CuPy
License
NotificationsYou must be signed in to change notification settings
desh2608/beamformer
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
This package is modified from the core parts ofpb_bssand modifies it to useCuPy for accelerated GPU-based inference.
At the moment, it is meant to be used with theGSS toolkit,but it can also be used as a general beamformer.
> pip install cupy-cuda102# modify according to your CUDA version (https://docs.cupy.dev/en/stable/install.html#installing-cupy)> pip install beamformer-gpu
frombeamformerimportbeamform_mvdrimportcupyascpX=cp.random.rand(4,1000,513)# D, T, FX_mask=cp.random.rand(1000,513)# T, FN_mask=cp.random.rand(1000,513)# T, FX_hat=beamform_mvdr(X,X_mask,N_mask,ban=True)