Skip to content

Commit

Permalink
dev(MeanIoU): refine docstrings
Browse files Browse the repository at this point in the history
  • Loading branch information
ice-tong committed Sep 29, 2022
1 parent dfd2aba commit 9226167
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions mmeval/segmentation/mean_iou.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,9 +20,9 @@ class MeanIoU(BaseMetric):
In addition to mean iou, it will also compute and return accuracy, mean
accuracy, mean dice, mean precision, mean recall and mean f-score.
Currently, there are 2 implementations of this metric: NumPy and PyTorch.
Which implementation to use is determined by the type of the calling
parameters. e.g. `numpy.ndarray` or `torch.Tensor`.
This metric supports 2 kinds of inputs, i.e. ``numpy.ndarray`` and
``torch.Tensor``, and the implementation for the calculation depends on
the inputs type.
Args:
num_classes (int, optional): The number of classes. If None, it will be
Expand Down

0 comments on commit 9226167

Please sign in to comment.