Wanli commited on
Commit
8b5437e
·
1 Parent(s): b81d9fd

update benchmark (#97)

Browse files
benchmark/README.md CHANGED
@@ -20,12 +20,20 @@ Time is measured from data preprocess (resize is excluded), to a forward pass of
20
  Run the following command to benchmark on a given config:
21
 
22
  ```shell
23
- PYTHONPATH=.. python benchmark.py --cfg ./config/face_detection_yunet.yaml
 
24
  ```
25
 
26
  If you are a Windows user and wants to run in CMD/PowerShell, use this command instead:
 
 
 
 
 
 
 
27
  ```shell
28
- set PYTHONPATH=..
29
  python benchmark.py --cfg ./config/face_detection_yunet.yaml
30
  ```
31
  <!--
 
20
  Run the following command to benchmark on a given config:
21
 
22
  ```shell
23
+ export PYTHONPATH=$PYTHONPATH:..
24
+ python benchmark.py --cfg ./config/face_detection_yunet.yaml
25
  ```
26
 
27
  If you are a Windows user and wants to run in CMD/PowerShell, use this command instead:
28
+ - CMD
29
+ ```shell
30
+ set PYTHONPATH=%PYTHONPATH%;..
31
+ python benchmark.py --cfg ./config/face_detection_yunet.yaml
32
+ ```
33
+
34
+ - PowerShell
35
  ```shell
36
+ $env:PYTHONPATH=$env:PYTHONPATH+";.."
37
  python benchmark.py --cfg ./config/face_detection_yunet.yaml
38
  ```
39
  <!--
benchmark/benchmark.py CHANGED
@@ -30,7 +30,7 @@ def prepend_pythonpath(cfg):
30
  prepend_pythonpath(v)
31
  else:
32
  if 'path' in k.lower():
33
- cfg[k] = os.path.join(os.environ['PYTHONPATH'], v)
34
 
35
  class Benchmark:
36
  def __init__(self, **kwargs):
 
30
  prepend_pythonpath(v)
31
  else:
32
  if 'path' in k.lower():
33
+ cfg[k] = os.path.join(os.environ['PYTHONPATH'].split(os.pathsep)[-1], v)
34
 
35
  class Benchmark:
36
  def __init__(self, **kwargs):
tools/quantize/requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  opencv-python>=4.5.4.58
2
  onnx
3
  onnxruntime
 
4
  neural-compressor
 
1
  opencv-python>=4.5.4.58
2
  onnx
3
  onnxruntime
4
+ onnxruntime-extensions
5
  neural-compressor