Fix a bug occurred when training a RTMdet tiny model (#3266)

change width and hieght order
openvinotoolkit · Apr 5, 2024 · 66d51f8 · 66d51f8
1 parent 3e9e6f8
commit 66d51f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/otx/algo/instance_segmentation/heads/custom_rtmdet_ins_head.py b/src/otx/algo/instance_segmentation/heads/custom_rtmdet_ins_head.py
@@ -76,8 +76,8 @@ def mask_postprocess(
                 F.interpolate(
                     mask_logits[:, inds],
                     size=[
-                        img_w,
                         img_h,
+                        img_w,
                     ],
                     mode="bilinear",
                     align_corners=False,