We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 01f87c7 commit 72b9d62Copy full SHA for 72b9d62
torchvision/models/detection/mask_rcnn.py
@@ -50,7 +50,7 @@ class MaskRCNN(FasterRCNN):
50
``0 <= x1 < x2 <= W`` and ``0 <= y1 < y2 <= H``.
51
- labels (Int64Tensor[N]): the predicted labels for each image
52
- 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
+ - masks (FloatTensor[N, 1, H, W]): the predicted masks for each instance, in 0-1 range. In order to
54
obtain the final segmentation masks, the soft masks can be thresholded, generally
55
with a value of 0.5 (mask >= 0.5)
56
0 commit comments