https://github.com/desh2608/beamformer
Souden MVDR beamformer on GPU with CuPy
https://github.com/desh2608/beamformer
beamformer mvdr
Last synced: 2 months ago
JSON representation
Souden MVDR beamformer on GPU with CuPy
- Host: GitHub
- URL: https://github.com/desh2608/beamformer
- Owner: desh2608
- License: mit
- Created: 2022-04-06T13:44:06.000Z (about 3 years ago)
- Default Branch: main
- Last Pushed: 2022-04-11T15:49:38.000Z (about 3 years ago)
- Last Synced: 2024-10-11T09:26:24.305Z (8 months ago)
- Topics: beamformer, mvdr
- Language: Python
- Homepage:
- Size: 10.7 KB
- Stars: 1
- Watchers: 3
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Souden MVDR beamformer in CuPy
This package is modified from the core parts of [pb_bss](https://github.com/fgnt/pb_bss)
and modifies it to use [CuPy](https://github.com/cupy/cupy) for accelerated GPU-based inference.
At the moment, it is meant to be used with the [GSS](https://github.com/desh2608/gss) toolkit,
but it can also be used as a general beamformer.
## Installation
```bash
> pip install cupy-cuda102 # modify according to your CUDA version (https://docs.cupy.dev/en/stable/install.html#installing-cupy)
> pip install beamformer-gpu
```
## Usage
```python
from beamformer import beamform_mvdr
import cupy as cp
X = cp.random.rand(4, 1000, 513) # D, T, F
X_mask = cp.random.rand(1000, 513) # T, F
N_mask = cp.random.rand(1000, 513) # T, F
X_hat = beamform_mvdr(X, X_mask, N_mask, ban=True)
```