diff --git a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt index a8eebb0551a40d643aa9c03e7a797e6e19deb0d0..15060e503517e4d7651fd69e4a650629ce97ace0 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/cascade_rcnn/cascade_mask_rcnn_r50_ fpgm_export:tools/export_model.py -c configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt index 6a5396dcf2dc1b2ac544e10998edc1d954ba6407..acd89de7bd9838921c4432202c2c58ad4d5703d1 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/cascade_rcnn/cascade_mask_rcnn_r50_ fpgm_export:tools/export_model.py -c configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/cascade_rcnn/cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_infer_python.txt index c84223cee84e4458dbebc6b7aea361cd8ea98d3d..c92e7053ac57dc832c05b03ed2829f76d768f811 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/cascade_rcnn/cascade_rcnn_r50_fpn_1 fpgm_export:tools/export_model.py -c configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt index 9af29c5fe121864216cb6672c4deebf12d5523ef..1bd0eed60ce81962ea32305b0e5fcd214e85ae42 100644 --- a/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/cascade_rcnn/cascade_rcnn_r50_vd_fp fpgm_export:tools/export_model.py -c configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/cascade_rcnn/cascade_rcnn_r50_vd_fpn_ssld_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_infer_python.txt index bb35fbd046da7484486be21db461bbfdb54a67fb..c9434b9cdf457ae4b38264759d7f0d2e06f34ea2 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r101_1x_coc fpgm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r101_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/faster_rcnn/faster_rcnn_r101_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_infer_python.txt index 2663ed6823c61346209043ca6052af5db177c154..da3523a1f424e5392532bcf14ecd6266a4835cfc 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r101_fpn_1x fpgm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt index 7e03d9ed47b399c123284a57b7c903ac172a8126..35a4bfd7df04b18782667e824d8a8cbabe277d57 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r101_vd_fpn fpgm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/faster_rcnn/faster_rcnn_r101_vd_fpn_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_infer_python.txt index 15eb022486155a7bde09292aff7363f36e9dca81..49587c5bc7f7d8e119716553c8a85bccbee69081 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r34_fpn_1x_ fpgm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/faster_rcnn/faster_rcnn_r34_fpn_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_infer_python.txt index e63e6ebd77b71b42c61e1b435af1646af8379a56..cc9c72a76ea275b8634e73e6bb3af1d6c0c2012d 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r34_vd_fpn_ fpgm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/faster_rcnn/faster_rcnn_r34_vd_fpn_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_infer_python.txt index b9938cbe6b92efe12b33513bc077d9852ed51890..a2f701da4eb9c9e574e64ecc0f470b2d2bf9e832 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r50_1x_coco fpgm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r50_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/faster_rcnn/faster_rcnn_r50_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_infer_python.txt index d4ca96fab3134334ea3a3d28e3dbdc89d9cbb675..bb176533824922ee9865e73539509700e3dc4dd9 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r50_fpn_1x_ fpgm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_infer_python.txt index 024208b33fbeef4e748a44c58f9cd9e977bac4ba..758a94cdab20b69e281a6f26c0fcc0f80465b28e 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r50_vd_1x_c fpgm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/faster_rcnn/faster_rcnn_r50_vd_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt index 3b2522fcba571d72ab51f1c603aec47028fbffec..de865f501ec352345e61852d4f16d9488a23590d 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r50_vd_fpn_ fpgm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/faster_rcnn/faster_rcnn_r50_vd_fpn_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_infer_python.txt index 9c59a78b5799295217250939fbbaf64bd105b96a..207c95b35d95842b957648028216b31dd482fe2b 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_swin_tiny_f fpgm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/faster_rcnn/faster_rcnn_swin_tiny_fpn_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt index a47c34c5ffb64ff4dea95f04ba2af504c2259aef..297cbfe0985bf59249dd67b4c135a326a17c323a 100644 --- a/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_x101_vd_64x fpgm_export:tools/export_model.py -c configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/faster_rcnn/faster_rcnn_x101_vd_64x4d_fpn_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_infer_python.txt index eea535fdbc4f51e69f60768819db5bb86735190c..da4b45aea4535c9e953c8fb097b0ff6a629c93dc 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coc fpgm_export:tools/export_model.py -c configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt index 96812063c8a27d0b71f7a5498d4b091e7f122cca..6e658ce9745b082a95ec2e58f666ffbc873b4b61 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_ fpgm_export:tools/export_model.py -c configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/mask_rcnn/mask_rcnn_r101_vd_fpn_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_infer_python.txt index d2563152680e82c98365a5d382e86d358b3dd42c..d3d91a9de5e431a8121bd512e7a71ed5bfaaf932 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/mask_rcnn/mask_rcnn_r50_1x_coco.yml fpgm_export:tools/export_model.py -c configs/mask_rcnn/mask_rcnn_r50_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/mask_rcnn/mask_rcnn_r50_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt index f4d95bcdc0e5d491a5c8f6c3c2ff7afc9dfd65e4..fe15d42c30dfd74bbba0dcb19bfa750ecdc435f6 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco fpgm_export:tools/export_model.py -c configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt index fd3e542a6e8b6f6d9e5da551251e9f8400d188fc..f1156ff3ca6982192c3eb1ce3c2195eacfe4292c 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_c fpgm_export:tools/export_model.py -c configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/mask_rcnn/mask_rcnn_r50_vd_fpn_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt b/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt index ec7e0f4a72d3238451c693b446845aff3a99b5ad..4ad43fabd79f1acf22c547eb44ea5845cff9f6af 100644 --- a/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt +++ b/test_tipc/configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco_train_infer_python.txt @@ -32,7 +32,7 @@ pact_export:tools/export_model.py -c configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_f fpgm_export:tools/export_model.py -c configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco.yml --slim_config _template_fpgm -o distill_export:null export1:null -export2:null +export_onnx:null kl_quant_export:tools/post_quant.py -c configs/mask_rcnn/mask_rcnn_x101_vd_64x4d_fpn_1x_coco.yml --slim_config _template_kl_quant -o ## infer_mode:norm diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh index da8cb37e022016490188165c7dad648522fcc556..d0eb77810b827b87b185ad95544c5c7f8ec3217d 100644 --- a/test_tipc/test_train_inference_python.sh +++ b/test_tipc/test_train_inference_python.sh @@ -62,7 +62,7 @@ fpgm_export=$(func_parser_value "${lines[31]}") distill_export=$(func_parser_value "${lines[32]}") export_key1=$(func_parser_key "${lines[33]}") export_value1=$(func_parser_value "${lines[33]}") -export_key2=$(func_parser_key "${lines[34]}") +export_onnx_key=$(func_parser_key "${lines[34]}") export_value2=$(func_parser_value "${lines[34]}") kl_quant_export=$(func_parser_value "${lines[35]}") @@ -297,15 +297,27 @@ else fi # run export model if [ ${run_export} != "null" ]; then - # run export model + save_export_model_dir="${save_log}/${model_name}" set_export_weight=$(func_set_params "${export_weight_key}" "${save_log}/${model_name}/${train_model_name}") set_save_export_dir=$(func_set_params "${save_export_key}" "${save_log}") + if [ ${export_onnx_key} = "export_onnx" ]; then + # run export onnx model for rcnn + export_cmd="${python} ${run_export} ${set_export_weight} ${set_filename} export_onnx=True ${set_save_export_dir} " + eval $export_cmd + status_check $? "${export_cmd}" "${status_log}" + # copy model for inference benchmark + eval "cp ${save_export_model_dir}/* ${save_log}/" + fi + # run export model export_cmd="${python} ${run_export} ${set_export_weight} ${set_filename} ${set_save_export_dir} " eval $export_cmd status_check $? "${export_cmd}" "${status_log}" #run inference - save_export_model_dir="${save_log}/${model_name}" + if [ ${export_onnx_key} != "export_onnx" ]; then + # copy model for inference benchmark + eval "cp ${save_export_model_dir}/* ${save_log}/" + fi eval $env func_inference "${python}" "${inference_py}" "${save_export_model_dir}" "${LOG_PATH}" "${train_infer_img_dir}" "${flag_quant}"