Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
chhluo committed Aug 22, 2022
1 parent 59e2baf commit 8126d07
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions mmdet/models/roi_heads/mask_heads/fcn_mask_head.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Expand Down

0 comments on commit 8126d07

Please sign in to comment.