optimise perf

openvinotoolkit · Oct 23, 2023 · 506bfaa · 506bfaa
1 parent fd67555
commit 506bfaa
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/src/otx/algorithms/detection/configs/instance_segmentation/rtmdet_ins_tiny/deployment.py b/src/otx/algorithms/detection/configs/instance_segmentation/rtmdet_ins_tiny/deployment.py
@@ -7,4 +7,11 @@
     output_names=["boxes", "labels", "masks"],
 )
 
+codebase_config = dict(
+    post_processing=dict(
+        max_output_boxes_per_class=100,
+        pre_top_k=300,
+    )
+)
+
 scale_ir_input = False
diff --git a/src/otx/algorithms/detection/configs/instance_segmentation/rtmdet_ins_tiny/model.py b/src/otx/algorithms/detection/configs/instance_segmentation/rtmdet_ins_tiny/model.py
@@ -54,11 +54,11 @@
         assigner=dict(type="DynamicSoftLabelAssigner", topk=13), allowed_border=-1, pos_weight=-1, debug=False
     ),
     test_cfg=dict(
-        nms_pre=1000,
+        nms_pre=300,
         min_bbox_size=0,
         score_thr=0.05,
         nms=dict(type="nms", iou_threshold=0.6),
-        max_per_img=200,
+        max_per_img=100,
         mask_thr_binary=0.5,
     ),
 )

diff --git a/src/otx/algorithms/detection/configs/instance_segmentation/rtmdet_ins_tiny/template.yaml b/src/otx/algorithms/detection/configs/instance_segmentation/rtmdet_ins_tiny/template.yaml
@@ -36,7 +36,7 @@ hyper_parameters:
       learning_rate_warmup_iters:
         default_value: 3
       num_iters:
-        default_value: 200
+        default_value: 100
     pot_parameters:
       stat_requests_number:
         default_value: 1