From 8126d07422088c7e133d68968ef160617b2c1d86 Mon Sep 17 00:00:00 2001 From: luochunhua Date: Mon, 22 Aug 2022 07:13:23 +0000 Subject: [PATCH] update --- mmdet/models/roi_heads/mask_heads/fcn_mask_head.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mmdet/models/roi_heads/mask_heads/fcn_mask_head.py b/mmdet/models/roi_heads/mask_heads/fcn_mask_head.py index 9fc1974976b..55f238db2ea 100644 --- a/mmdet/models/roi_heads/mask_heads/fcn_mask_head.py +++ b/mmdet/models/roi_heads/mask_heads/fcn_mask_head.py @@ -303,7 +303,7 @@ def _predict_by_feat_single(self, Tensor: Encoded masks, has shape (n, img_w, img_h) Example: - >>> import mmcv + >>> from mmengine.config import Config >>> from mmdet.models.roi_heads.mask_heads.fcn_mask_head import * # NOQA >>> N = 7 # N = number of extracted ROIs >>> C, H, W = 11, 32, 32 @@ -314,7 +314,7 @@ def _predict_by_feat_single(self, >>> # Each input is associated with some bounding box >>> bboxes = torch.Tensor([[1, 1, 42, 42 ]] * N) >>> labels = torch.randint(0, C, size=(N,)) - >>> rcnn_test_cfg = mmcv.Config({'mask_thr_binary': 0, }) + >>> rcnn_test_cfg = Config({'mask_thr_binary': 0, }) >>> ori_shape = (H * 4, W * 4) >>> scale_factor = (1, 1) >>> rescale = False