Skip to content

Commit 72b9d62

Browse files
hmk114NicolasHug
andauthored
Update docstrings of MaskRCNN (#9083)
Co-authored-by: Nicolas Hug <[email protected]>
1 parent 01f87c7 commit 72b9d62

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/models/detection/mask_rcnn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ class MaskRCNN(FasterRCNN):
5050
``0 <= x1 < x2 <= W`` and ``0 <= y1 < y2 <= H``.
5151
- labels (Int64Tensor[N]): the predicted labels for each image
5252
- scores (Tensor[N]): the scores or each prediction
53-
- masks (UInt8Tensor[N, 1, H, W]): the predicted masks for each instance, in 0-1 range. In order to
53+
- masks (FloatTensor[N, 1, H, W]): the predicted masks for each instance, in 0-1 range. In order to
5454
obtain the final segmentation masks, the soft masks can be thresholded, generally
5555
with a value of 0.5 (mask >= 0.5)
5656

0 commit comments

Comments
 (0)