mmcv.ops.contour_expand¶
- mmcv.ops.contour_expand(kernel_mask: array | Tensor, internal_kernel_label: array | 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: