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

Expand kernel contours so that foreground pixels are assigned into instances.

参数

• 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.

返回

The instance index map with size hxw.

返回类型

list