Add MaskRCNN with saturation issue (openvinotoolkit#798)

XianFuWongIntel · Jun 25, 2021 · 4d4d332 · 4d4d332
1 parent 48c242a
commit 4d4d332
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/examples/tensorflow/segmentation/README.md b/examples/tensorflow/segmentation/README.md
@@ -129,5 +129,5 @@ To export a model to the OpenVINO IR and run it using the Intel® Deep Learning
 |**Model**|**Compression algorithm**|**Dataset**|**TensorFlow mAP, %**|**NNCF config file**|**TensorFlow checkpoint**|
 | :---: | :---: | :---: | :---: | :---: | :---: |
 |MaskRCNN|None|COCO2017|bbox: 37.33<br/>segm: 33.56|[mask_rcnn_coco.json](configs/mask_rcnn_coco.json)|[Link](https://storage.openvinotoolkit.org/repositories/nncf/tensorflow/models/develop/mask_rcnn_baseline.tar.gz)|
-|MaskRCNN|INT8 w:sym,per-tensor a:sym,per-tensor|COCO2017|bbox: 37.12<br/>segm: 33.39|[mask_rcnn_coco_int8.json](configs/quantization/mask_rcnn_coco_int8.json)|[Link](https://storage.openvinotoolkit.org/repositories/nncf/tensorflow/models/develop/mask_rcnn_int8_w_sym_t_a_sym_t.tar.gz)|
+|MaskRCNN|INT8 w:sym,per-tensor a:sym,per-tensor|COCO2017|bbox: 37.12<br/>segm: 33.52|[mask_rcnn_coco_int8.json](configs/quantization/mask_rcnn_coco_int8.json)|[Link](https://storage.openvinotoolkit.org/repositories/nncf/tensorflow/models/develop/mask_rcnn_int8_w_sym_t_half_a_sym_t.tar.gz)|
 |MaskRCNN|Sparsity 50% (Magnitude)|COCO2017|bbox: 36.93<br/>segm: 33.23|[mask_rcnn_coco_magnitude_sparsity.json](configs/sparsity/mask_rcnn_coco_magnitude_sparsity.json)|[Link](https://storage.openvinotoolkit.org/repositories/nncf/tensorflow/models/develop/mask_rcnn_sparsity_50.tar.gz)|
diff --git a/examples/tensorflow/segmentation/configs/quantization/mask_rcnn_coco_int8.json b/examples/tensorflow/segmentation/configs/quantization/mask_rcnn_coco_int8.json
@@ -21,7 +21,6 @@
 
   "compression": {
       "algorithm": "quantization",
-      "disable_saturation_fix": true,
       "ignored_scopes": [
               "{re}.*multilevel_propose_rois.*",
               "{re}.*sample_proposals.*",

diff --git a/tests/tensorflow/sota_checkpoints_eval.json b/tests/tensorflow/sota_checkpoints_eval.json
@@ -343,13 +343,13 @@
                     "reverse_input_channels": true,
                     "batch": 16
                 },
-                "mask_rcnn_int8_w_sym_t_a_sym_t": {
+                "mask_rcnn_int8_w_sym_t_half_a_sym_t": {
                     "config": "examples/tensorflow/segmentation/configs/quantization/mask_rcnn_coco_int8.json",
                     "reference": "mask_rcnn_baseline",
                     "target": 37.12,
-                    "resume": "mask_rcnn_int8_w_sym_t_a_sym_t",
+                    "resume": "mask_rcnn_int8_w_sym_t_half_a_sym_t",
                     "metric_type": "mAP",
-                    "model_description": "mask_rcnn_int8_w_sym_t_a_sym_t",
+                    "model_description": "mask_rcnn_int8_w_sym_t_half_a_sym_t",
                     "compression_description": "INT8",
                     "batch_per_gpu": 4,
                     "reverse_input_channels": true