Skip to content

Commit 20559d2

Browse files
mengniwang95mengniwang95
authored
Update onnx example for new API (#190)
Signed-off-by: Mengni Wang <mengni.wang@intel.com>
1 parent 01adc8b commit 20559d2

File tree

148 files changed

+482
-6934
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

148 files changed

+482
-6934
lines changed

‎examples/.config/model_params_onnxrt.json‎

Lines changed: 16 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -176,12 +176,10 @@
176176
},
177177
"squeezenet": {
178178
"model_src_dir": "image_recognition/onnx_model_zoo/squeezenet/quantization/ptq",
179-
"dataset_location": "/tf_dataset2/datasets/imagenet/ImagenetRaw/ILSVRC2012_img_val",
179+
"dataset_location": "/tf_dataset2/datasets/imagenet/ImagenetRaw",
180180
"input_model": "/tf_dataset2/models/onnx/squeezenet/squeezenet1.0-12.onnx",
181-
"yaml": "squeezenet.yaml",
182-
"strategy": "basic",
183-
"batch_size": 1,
184-
"new_benchmark": true
181+
"main_script": "main.py",
182+
"batch_size": 1
185183
},
186184
"caffenet": {
187185
"model_src_dir": "image_recognition/onnx_model_zoo/caffenet/quantization/ptq",
@@ -239,12 +237,10 @@
239237
},
240238
"fcn": {
241239
"model_src_dir": "image_recognition/onnx_model_zoo/fcn/quantization/ptq",
242-
"dataset_location": "/tf_dataset2/datasets/coco2017/coco/val2017",
240+
"dataset_location": "/tf_dataset2/datasets/coco2017/coco",
243241
"input_model": "/tf_dataset2/models/onnx/fcn/fcn-resnet50-11.onnx",
244-
"yaml": "fcn_rn50.yaml",
245-
"strategy": "basic",
246-
"batch_size": 1,
247-
"new_benchmark": true
242+
"main_script": "main.py",
243+
"batch_size": 1
248244
},
249245
"ssd_mobilenet_v1-2": {
250246
"model_src_dir": "object_detection/onnx_model_zoo/ssd_mobilenet_v1/quantization/ptq",
@@ -516,12 +512,10 @@
516512
},
517513
"fcn_qdq": {
518514
"model_src_dir": "image_recognition/onnx_model_zoo/fcn/quantization/ptq",
519-
"dataset_location": "/tf_dataset2/datasets/coco2017/coco/val2017",
515+
"dataset_location": "/tf_dataset2/datasets/coco2017/coco",
520516
"input_model": "/tf_dataset2/models/onnx/fcn/fcn-resnet50-13.onnx",
521-
"yaml": "fcn_rn50_qdq.yaml",
522-
"strategy": "basic",
523-
"batch_size": 1,
524-
"new_benchmark": true
517+
"main_script": "main.py",
518+
"batch_size": 1
525519
},
526520
"ssd_mobilenet_v1-2_qdq": {
527521
"model_src_dir": "object_detection/onnx_model_zoo/ssd_mobilenet_v1/quantization/ptq",
@@ -576,31 +570,25 @@
576570
"batch_size": 1
577571
},
578572
"arcface": {
579-
"model_src_dir": "image_recognition/onnx_model_zoo/arcface/quantization/ptq",
573+
"model_src_dir": "body_analysis/onnx_model_zoo/arcface/quantization/ptq",
580574
"dataset_location": "/tf_dataset2/datasets/faces_ms1m_112x112/lfw.bin",
581575
"input_model": "/tf_dataset2/models/onnx/arcface/arcfaceresnet100-11.onnx",
582-
"yaml": "arcface.yaml",
583-
"strategy": "basic",
584-
"batch_size": 1,
585-
"new_benchmark": true
576+
"main_script": "main.py",
577+
"batch_size": 1
586578
},
587579
"emotion_ferplus": {
588580
"model_src_dir": "body_analysis/onnx_model_zoo/emotion_ferplus/quantization/ptq",
589581
"dataset_location": "/tf_dataset2/datasets/emotion_data/emotion_ferplus/fer2013/fer2013.csv",
590582
"input_model": "/tf_dataset2/models/onnx/emotion_ferplus/emotion-ferplus-12.onnx",
591-
"yaml": "emotion_ferplus.yaml",
592-
"strategy": "basic",
593-
"batch_size": 1,
594-
"new_benchmark": true
583+
"main_script": "main.py",
584+
"batch_size": 1
595585
},
596586
"ultraface": {
597587
"model_src_dir": "body_analysis/onnx_model_zoo/ultraface/quantization/ptq",
598588
"dataset_location": "/tf_dataset2/datasets/wider_face_data/WIDER_val",
599589
"input_model": "/tf_dataset2/models/onnx/ultraface/version-RFB-320-12.onnx",
600-
"yaml": "ultraface.yaml",
601-
"strategy": "basic",
602-
"batch_size": 1,
603-
"new_benchmark": true
590+
"main_script": "main.py",
591+
"batch_size": 1
604592
},
605593
"unet": {
606594
"model_src_dir": "image_recognition/unet/quantization/ptq",

‎examples/README.md‎

Lines changed: 19 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -815,55 +815,55 @@ Intel® Neural Compressor validated examples with multiple compression technique
815815
<td>ResNet50 V1.5</td>
816816
<td>Image Recognition</td>
817817
<td>Post-Training Static Quantization</td>
818-
<td><a href="./onnxrt/image_recognition/resnet50/quantization/ptq">qlinearops</a> / <a href="./onnxrt/image_recognition/resnet50/quantization/ptq">qdq</a></td>
818+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/resnet50/quantization/ptq">qlinearops</a> / <a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/resnet50/quantization/ptq">qdq</a></td>
819819
</tr>
820820
<tr>
821821
<td>ResNet50 V1.5 MLPerf</td>
822822
<td>Image Recognition</td>
823823
<td>Post-Training Static Quantization</td>
824-
<td><a href="./onnxrt/image_recognition/resnet50/quantization/ptq">qlinearops</a> / <a href="./onnxrt/image_recognition/resnet50/quantization/ptq">qdq</a></td>
824+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/resnet50/quantization/ptq">qlinearops</a> / <a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/resnet50/quantization/ptq">qdq</a></td>
825825
</tr>
826826
<tr>
827827
<td>VGG16</td>
828828
<td>Image Recognition</td>
829829
<td>Post-Training Static Quantization</td>
830-
<td><a href="./onnxrt/image_recognition/vgg16/quantization/ptq">qlinearops</a> / <a href="./onnxrt/image_recognition/vgg16/quantization/ptq">qdq</a></td>
830+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/vgg16/quantization/ptq">qlinearops</a> / <a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/vgg16/quantization/ptq">qdq</a></td>
831831
</tr>
832832
<tr>
833833
<td>MobileNet V2</td>
834834
<td>Image Recognition</td>
835835
<td>Post-Training Static Quantization</td>
836-
<td><a href="./onnxrt/image_recognition/mobilenet_v2/quantization/ptq">qlinearops</a> / <a href="./onnxrt/image_recognition/mobilenet_v2/quantization/ptq">qdq</a></td>
836+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/mobilenet_v2/quantization/ptq">qlinearops</a> / <a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/mobilenet_v2/quantization/ptq">qdq</a></td>
837837
</tr>
838838
<tr>
839839
<td>MobileNet V3 MLPerf</td>
840840
<td>Image Recognition</td>
841841
<td>Post-Training Static Quantization</td>
842-
<td><a href="./onnxrt/image_recognition/mobilenet_v3/quantization/ptq">qlinearops</a> / <a href="./onnxrt/image_recognition/mobilenet_v3/quantization/ptq">qdq</a></td>
842+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/mobilenet_v3/quantization/ptq">qlinearops</a> / <a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/mobilenet_v3/quantization/ptq">qdq</a></td>
843843
</tr>
844844
<tr>
845845
<td>AlexNet</td>
846846
<td>Image Recognition</td>
847847
<td>Post-Training Static Quantization</td>
848-
<td><a href="./onnxrt/image_recognition/onnx_model_zoo/alexnet/quantization/ptq">qlinearops</a> / <a href="./onnxrt/image_recognition/onnx_model_zoo/alexnet/quantization/ptq">qdq</a></td>
848+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/alexnet/quantization/ptq">qlinearops</a> / <a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/alexnet/quantization/ptq">qdq</a></td>
849849
</tr>
850850
<tr>
851851
<td>CaffeNet</td>
852852
<td>Image Recognition</td>
853853
<td>Post-Training Static Quantization</td>
854-
<td><a href="./onnxrt/image_recognition/onnx_model_zoo/caffenet/quantization/ptq">qlinearops</a> / <a href="./onnxrt/image_recognition/onnx_model_zoo/caffenet/quantization/ptq">qdq</a></td>
854+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/caffenet/quantization/ptq">qlinearops</a> / <a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/caffenet/quantization/ptq">qdq</a></td>
855855
</tr>
856856
<tr>
857857
<td>DenseNet</td>
858858
<td>Image Recognition</td>
859859
<td>Post-Training Static Quantization</td>
860-
<td><a href="./onnxrt/image_recognition/onnx_model_zoo/densenet/quantization/ptq">qlinearops</a></td>
860+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/densenet/quantization/ptq">qlinearops</a></td>
861861
</tr>
862862
<tr>
863863
<td>EfficientNet</td>
864864
<td>Image Recognition</td>
865865
<td>Post-Training Static Quantization</td>
866-
<td><a href="./onnxrt/image_recognition/onnx_model_zoo/efficientnet/quantization/ptq">qlinearops</a> / <a href="./onnxrt/image_recognition/onnx_model_zoo/efficientnet/quantization/ptq">qdq</a></td>
866+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/efficientnet/quantization/ptq">qlinearops</a> / <a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/efficientnet/quantization/ptq">qdq</a></td>
867867
</tr>
868868
<tr>
869869
<td>FCN</td>
@@ -875,37 +875,37 @@ Intel® Neural Compressor validated examples with multiple compression technique
875875
<td>GoogleNet</td>
876876
<td>Image Recognition</td>
877877
<td>Post-Training Static Quantization</td>
878-
<td><a href="./onnxrt/image_recognition/onnx_model_zoo/googlenet/quantization/ptq">qlinearops</a> / <a href="./onnxrt/image_recognition/onnx_model_zoo/googlenet/quantization/ptq">qdq</a></td>
878+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/googlenet/quantization/ptq">qlinearops</a> / <a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/googlenet/quantization/ptq">qdq</a></td>
879879
</tr>
880880
<tr>
881881
<td>Inception V1</td>
882882
<td>Image Recognition</td>
883883
<td>Post-Training Static Quantization</td>
884-
<td><a href="./onnxrt/image_recognition/onnx_model_zoo/inception/quantization/ptq">qlinearops</a> / <a href="./onnxrt/image_recognition/onnx_model_zoo/inception/quantization/ptq">qdq</a></td>
884+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/inception/quantization/ptq">qlinearops</a> / <a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/inception/quantization/ptq">qdq</a></td>
885885
</tr>
886886
<tr>
887887
<td>MNIST</td>
888888
<td>Image Recognition</td>
889889
<td>Post-Training Static Quantization</td>
890-
<td><a href="./onnxrt/image_recognition/onnx_model_zoo/mnist/quantization/ptq">qlinearops</a></td>
890+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/mnist/quantization/ptq">qlinearops</a></td>
891891
</tr>
892892
<tr>
893893
<td>MobileNet V2 (ONNX Model Zoo)</td>
894894
<td>Image Recognition</td>
895895
<td>Post-Training Static Quantization</td>
896-
<td><a href="./onnxrt/image_recognition/onnx_model_zoo/mobilenet/quantization/ptq">qlinearops</a> / <a href="./onnxrt/image_recognition/onnx_model_zoo/mobilenet/quantization/ptq">qdq</a></td>
896+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/mobilenet/quantization/ptq">qlinearops</a> / <a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/mobilenet/quantization/ptq">qdq</a></td>
897897
</tr>
898898
<tr>
899899
<td>ResNet50 V1.5 (ONNX Model Zoo)</td>
900900
<td>Image Recognition</td>
901901
<td>Post-Training Static Quantization</td>
902-
<td><a href="./onnxrt/image_recognition/onnx_model_zoo/resnet50/quantization/ptq">qlinearops</a> / <a href="./onnxrt/image_recognition/onnx_model_zoo/resnet50/quantization/ptq">qdq</a></td>
902+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/resnet50/quantization/ptq">qlinearops</a> / <a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/resnet50/quantization/ptq">qdq</a></td>
903903
</tr>
904904
<tr>
905905
<td>ShuffleNet V2</td>
906906
<td>Image Recognition</td>
907907
<td>Post-Training Static Quantization</td>
908-
<td><a href="./onnxrt/image_recognition/onnx_model_zoo/shufflenet/quantization/ptq">qlinearops</a> / <a href="./onnxrt/image_recognition/onnx_model_zoo/shufflenet/quantization/ptq">qdq</a></td>
908+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/shufflenet/quantization/ptq">qlinearops</a> / <a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/shufflenet/quantization/ptq">qdq</a></td>
909909
</tr>
910910
<tr>
911911
<td>SqueezeNet</td>
@@ -917,19 +917,19 @@ Intel® Neural Compressor validated examples with multiple compression technique
917917
<td>VGG16 (ONNX Model Zoo)</td>
918918
<td>Image Recognition</td>
919919
<td>Post-Training Static Quantization</td>
920-
<td><a href="./onnxrt/image_recognition/onnx_model_zoo/vgg16/quantization/ptq">qlinearops</a> / <a href="./onnxrt/image_recognition/onnx_model_zoo/vgg16/quantization/ptq">qdq</a></td>
920+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/vgg16/quantization/ptq">qlinearops</a> / <a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/vgg16/quantization/ptq">qdq</a></td>
921921
</tr>
922922
<tr>
923923
<td>ZFNet</td>
924924
<td>Image Recognition</td>
925925
<td>Post-Training Static Quantization</td>
926-
<td><a href="./onnxrt/image_recognition/onnx_model_zoo/zfnet/quantization/ptq">qlinearops</a> / <a href="./onnxrt/image_recognition/onnx_model_zoo/zfnet/quantization/ptq">qdq</a></td>
926+
<td><a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/zfnet/quantization/ptq">qlinearops</a> / <a href="https://github.com/intel/neural-compressor/tree/old_api_examples/examples/onnxrt/image_recognition/onnx_model_zoo/zfnet/quantization/ptq">qdq</a></td>
927927
</tr>
928928
<tr>
929929
<td>ArcFace</td>
930930
<td>Image Recognition</td>
931931
<td>Post-Training Static Quantization</td>
932-
<td><a href="./onnxrt/image_recognition/onnx_model_zoo/arcface/quantization/ptq">qlinearops</a></td>
932+
<td><a href="./onnxrt/body_analysis/onnx_model_zoo/arcface/quantization/ptq">qlinearops</a></td>
933933
</tr>
934934
<tr>
935935
<td>BERT base MRPC</td>
@@ -1109,7 +1109,7 @@ Intel® Neural Compressor validated examples with multiple compression technique
11091109
<td>Emotion FERPlus</td>
11101110
<td>Body Analysis</td>
11111111
<td>Post-Training Static Quantization</td>
1112-
<td><a href="./onnxrt/body_analysis/onnx_model_zoo/emotion_ferplus/quantization/ptq">qlinearops</a></td>
1112+
<td><a href="/onnxrt/body_analysis/onnx_model_zoo/emotion_ferplus/quantization/ptq">qlinearops</a></td>
11131113
</tr>
11141114
<tr>
11151115
<td>Ultra Face</td>

‎examples/onnxrt/image_recognition/onnx_model_zoo/arcface/quantization/ptq/README.md‎ renamed to ‎examples/onnxrt/body_analysis/onnx_model_zoo/arcface/quantization/ptq/README.md‎

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -18,19 +18,15 @@ wget https://github.com/onnx/models/raw/main/vision/body_analysis/arcface/model/
1818

1919
```bash
2020
bash run_tuning.sh --input_model=path/to/model \ # model path as *.onnx
21-
--config=arcface.yaml \
22-
--data_path=/path/to/faces_ms1m_112x112/task.bin \
23-
--nfolds=nfolds_num \ # optional, data type is int, default is 1
21+
--dataset_location=/path/to/faces_ms1m_112x112/task.bin \
2422
--output_model=path/to/save
2523
```
2624

2725
### Benchmark
2826

2927
```bash
3028
bash run_benchmark.sh --input_model=path/to/model \ # model path as *.onnx
31-
--config=arcface.yaml \
32-
--data_path=/path/to/faces_ms1m_112x112/task.bin \
33-
--nfolds=nfolds_num \ # optional, data type is int, default is 1
29+
--dataset_location=/path/to/faces_ms1m_112x112/task.bin \
3430
--mode=performance # or accuracy
3531
```
3632

0 commit comments

Comments
 (0)