Update README.md
Browse files
README.md
CHANGED
|
@@ -63,7 +63,7 @@ Typical applications:
|
|
| 63 |
|
| 64 |
## Inference
|
| 65 |
|
| 66 |
-
This repo includes a reference script for running the TFLite models on WAV files using streaming-style, frame-by-frame inference: `
|
| 67 |
|
| 68 |
### Setup
|
| 69 |
|
|
@@ -78,7 +78,7 @@ pip install tflite-runtime
|
|
| 78 |
|
| 79 |
By default, the script loads models from:
|
| 80 |
|
| 81 |
-
*
|
| 82 |
|
| 83 |
Create the folder and place the `.tflite` files there (or edit `TFLITE_DIR` in the script to match your layout).
|
| 84 |
|
|
@@ -87,7 +87,7 @@ Create the folder and place the `.tflite` files there (or edit `TFLITE_DIR` in t
|
|
| 87 |
The script processes `*.wav` files non-recursively and writes enhanced outputs as 16-bit PCM WAVs:
|
| 88 |
|
| 89 |
```bash
|
| 90 |
-
python
|
| 91 |
```
|
| 92 |
|
| 93 |
Available `--model_name` options: `baseline`, `dpdfnet2`, `dpdfnet4`, `dpdfnet8`.
|
|
|
|
| 63 |
|
| 64 |
## Inference
|
| 65 |
|
| 66 |
+
This repo includes a reference script for running the TFLite models on WAV files using streaming-style, frame-by-frame inference: `run_tflite.py`.
|
| 67 |
|
| 68 |
### Setup
|
| 69 |
|
|
|
|
| 78 |
|
| 79 |
By default, the script loads models from:
|
| 80 |
|
| 81 |
+
* `./<model_name>.tflite`
|
| 82 |
|
| 83 |
Create the folder and place the `.tflite` files there (or edit `TFLITE_DIR` in the script to match your layout).
|
| 84 |
|
|
|
|
| 87 |
The script processes `*.wav` files non-recursively and writes enhanced outputs as 16-bit PCM WAVs:
|
| 88 |
|
| 89 |
```bash
|
| 90 |
+
python run_tflite.py --noisy_dir /path/to/noisy_wavs --enhanced_dir /path/to/out --model_name dpdfnet8
|
| 91 |
```
|
| 92 |
|
| 93 |
Available `--model_name` options: `baseline`, `dpdfnet2`, `dpdfnet4`, `dpdfnet8`.
|