From f699193f78fb0fc557f0eef2de19906474ecfc54 Mon Sep 17 00:00:00 2001 From: Eugene Liu Date: Tue, 16 Apr 2024 08:54:31 +0100 Subject: [PATCH 1/2] add fix --- .../instance_segmentation/heads/test_custom_rtmdet_ins_head.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/algo/instance_segmentation/heads/test_custom_rtmdet_ins_head.py b/tests/unit/algo/instance_segmentation/heads/test_custom_rtmdet_ins_head.py index 2ff04967585..77066e88584 100644 --- a/tests/unit/algo/instance_segmentation/heads/test_custom_rtmdet_ins_head.py +++ b/tests/unit/algo/instance_segmentation/heads/test_custom_rtmdet_ins_head.py @@ -71,7 +71,7 @@ def test_mask_pred(self, mocker) -> None: mask_head._bbox_mask_post_process( results[0], mask_feat, - cfg=None, + cfg=test_cfg, ) def test_predict_by_feat_ov(self) -> None: From 3d5a4d7e6c0b8cacc070d648a6f3003fa1b5feec Mon Sep 17 00:00:00 2001 From: Eugene Liu Date: Tue, 16 Apr 2024 09:42:54 +0100 Subject: [PATCH 2/2] fix --- .../heads/test_custom_rtmdet_ins_head.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/tests/unit/algo/instance_segmentation/heads/test_custom_rtmdet_ins_head.py b/tests/unit/algo/instance_segmentation/heads/test_custom_rtmdet_ins_head.py index 77066e88584..12f4926dc91 100644 --- a/tests/unit/algo/instance_segmentation/heads/test_custom_rtmdet_ins_head.py +++ b/tests/unit/algo/instance_segmentation/heads/test_custom_rtmdet_ins_head.py @@ -68,12 +68,6 @@ def test_mask_pred(self, mocker) -> None: cfg=test_cfg, ) - mask_head._bbox_mask_post_process( - results[0], - mask_feat, - cfg=test_cfg, - ) - def test_predict_by_feat_ov(self) -> None: with tempfile.TemporaryDirectory() as tmpdirname: lit_module = RTMDetInst(num_classes=1, variant="tiny")