danielr-ceva commited on
Commit
1a260a9
·
verified ·
1 Parent(s): 27c8c7f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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: `enhance.py`.
67
 
68
  ### Setup
69
 
@@ -78,7 +78,7 @@ pip install tflite-runtime
78
 
79
  By default, the script loads models from:
80
 
81
- * `./model_zoo/tflite/<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,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 enhance.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`.
 
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`.