我建立了一个分割模型来分割图像中的某些对象。我希望能够按指定的像素数(例如,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]
有没有更好(更有效)的方法来做到这一点?