如何使分割掩码更大?

时间:2021-03-05 21:50:59

标签: python deep-learning image-segmentation scikit-image

我建立了一个分割模型来分割图像中的某些对象。我希望能够按指定的像素数(例如,ndim = 10 像素)放大我的预测分割掩码,我想知道最好和最有效的方法是什么。

目前,我正在使用 view_as_windows 的 skimage 内置函数,每个蒙版需要大约 30 秒才能放大。

我用 ndim-1 填充我的图像并将 window_shape 指定为 (ndim, ndim)

然后我按如下方式返回新的掩码:

new_x = np.pad(x, ((pad_size, pad_size), (pad_size, pad_size)), 'constant', constant_values=(0, 0))
new_x = view_as_windows(new_x, window_shape)
new_x = new_x.max(axis=(2,3))
new_x = new_x[half_pad:-half_pad, half_pad:-half_pad]

有没有更好(更有效)的方法来做到这一点?

0 个答案:

没有答案