mmcv.ops.contour_expand¶
- mmcv.ops.contour_expand(kernel_mask: Union[numpy.array, torch.Tensor], internal_kernel_label: Union[numpy.array, torch.Tensor], min_kernel_area: int, kernel_num: int) → list[source]¶
Expand kernel contours so that foreground pixels are assigned into instances.
- Parameters
kernel_mask (np.array or torch.Tensor) – The instance kernel mask with size hxw.
internal_kernel_label (np.array or torch.Tensor) – The instance internal kernel label with size hxw.
min_kernel_area (int) – The minimum kernel area.
kernel_num (int) – The instance kernel number.
- Returns
The instance index map with size hxw.
- Return type