From 68db83f411d8ac241c909154b906df8a3e427a2b Mon Sep 17 00:00:00 2001
From: Vladislav Sovrasov <sovrasov.vlad@gmail.com>
Date: Mon, 22 Apr 2024 02:38:27 +0200
Subject: [PATCH] Update QAT configs for rotated detection (#3375)

---
 .../efficientnetb2b_maskrcnn/compression_config.json         | 5 +++++
 .../resnet50_maskrcnn/compression_config.json                | 5 +++++
 2 files changed, 10 insertions(+)

diff --git a/src/otx/algorithms/detection/configs/rotated_detection/efficientnetb2b_maskrcnn/compression_config.json b/src/otx/algorithms/detection/configs/rotated_detection/efficientnetb2b_maskrcnn/compression_config.json
index 7e0cba46aa9..c88a5c2744e 100644
--- a/src/otx/algorithms/detection/configs/rotated_detection/efficientnetb2b_maskrcnn/compression_config.json
+++ b/src/otx/algorithms/detection/configs/rotated_detection/efficientnetb2b_maskrcnn/compression_config.json
@@ -20,6 +20,11 @@
           "algorithm": "quantization",
           "initializer": {
             "range": {
+              "type": "percentile",
+              "params": {
+                "min_percentile": 0,
+                "max_percentile": 100
+              },
               "num_init_samples": 1000
             },
             "batchnorm_adaptation": {
diff --git a/src/otx/algorithms/detection/configs/rotated_detection/resnet50_maskrcnn/compression_config.json b/src/otx/algorithms/detection/configs/rotated_detection/resnet50_maskrcnn/compression_config.json
index ab687b9c6a2..af3c5ef8937 100644
--- a/src/otx/algorithms/detection/configs/rotated_detection/resnet50_maskrcnn/compression_config.json
+++ b/src/otx/algorithms/detection/configs/rotated_detection/resnet50_maskrcnn/compression_config.json
@@ -20,6 +20,11 @@
           "algorithm": "quantization",
           "initializer": {
             "range": {
+              "type": "percentile",
+              "params": {
+                "min_percentile": 0,
+                "max_percentile": 100
+              },
               "num_init_samples": 1000
             },
             "batchnorm_adaptation": {