hongw.qin commited on
Commit ·
1dde65b
1
Parent(s): c3891f1
update readme
Browse files
README.md
CHANGED
|
@@ -58,20 +58,22 @@ pip install -r requirements-eval.txt
|
|
| 58 |
|
| 59 |
### Test & Evaluation
|
| 60 |
|
| 61 |
-
- Run inference on images
|
| 62 |
|
| 63 |
```bash
|
| 64 |
python onnx_inference.py --onnx psfrgan_nchw_fp32.onnx --latent latent.npy --input /Path/To/Image --out-dir outputs
|
| 65 |
python onnx_inference.py --onnx psfrgan_nhwc_int8.onnx --latent latent.npy --input /Path/To/Image --out-dir outputs
|
| 66 |
```
|
| 67 |
|
| 68 |
-
|
| 69 |
|
| 70 |
-
|
| 71 |
-
- `--latent`: (Optional) Path to the latent code file (.npy). If not provided, random latent values will be generated with a fixed seed for reproducibility.
|
| 72 |
-
- `--out-dir`: Output directory where the restored images will be saved.
|
| 73 |
|
| 74 |
-
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 75 |
|
| 76 |
```bash
|
| 77 |
# eval fp32
|
|
|
|
| 58 |
|
| 59 |
### Test & Evaluation
|
| 60 |
|
| 61 |
+
- **Run inference on images**
|
| 62 |
|
| 63 |
```bash
|
| 64 |
python onnx_inference.py --onnx psfrgan_nchw_fp32.onnx --latent latent.npy --input /Path/To/Image --out-dir outputs
|
| 65 |
python onnx_inference.py --onnx psfrgan_nhwc_int8.onnx --latent latent.npy --input /Path/To/Image --out-dir outputs
|
| 66 |
```
|
| 67 |
|
| 68 |
+
_Arguments:_
|
| 69 |
|
| 70 |
+
`--input`: Accepts either a single image file path or a directory path. If it's a file, the script will process that image only. If it's a directory, the script will recursively scan for .png, .jpg, and .jpeg files and process all of them.
|
|
|
|
|
|
|
| 71 |
|
| 72 |
+
`--latent`: (Optional) Path to the latent code file (.npy). If not provided, random latent values will be generated with a fixed seed for reproducibility.
|
| 73 |
+
|
| 74 |
+
`--out-dir`: Output directory where the restored images will be saved.
|
| 75 |
+
|
| 76 |
+
- **Evaluate the quantized model**
|
| 77 |
|
| 78 |
```bash
|
| 79 |
# eval fp32
|