Fixed some typos in comments

This commit is contained in:
Elm Forest
2023-04-07 00:50:18 +08:00
parent aac76a1fb0
commit 20114f5138
8 changed files with 11 additions and 11 deletions

View File

@@ -85,8 +85,8 @@ class Sam(nn.Module):
(list(dict)): A list over input images, where each element is
as dictionary with the following keys.
'masks': (torch.Tensor) Batched binary mask predictions,
with shape BxCxHxW, where B is the number of input promts,
C is determiend by multimask_output, and (H, W) is the
with shape BxCxHxW, where B is the number of input prompts,
C is determined by multimask_output, and (H, W) is the
original size of the image.
'iou_predictions': (torch.Tensor) The model's predictions
of mask quality, in shape BxC.