从标签上的numpy数组中选择子数组

时间:2017-12-28 18:46:15

标签: python arrays numpy

我使用skimage库的分割模块处理了一个图像。它生成一个方形numpy数组,所有段都标有唯一编号。假设我们有一个400 * 400的图像,让我们说500个用500个唯一数字标记的段。

我需要为每个片段选择500个子阵列,并将每个子片段保存为单独的图像。

我可以使用一个非常残酷的简单算法和几个for循环,但我坚信使用numpy掩码方法有一个更简单的方法。问题是我自己无法应对: - (

生成细分的代码非常简单:

from skimage import future, graph, segmentation
import numpy as np

img_name = 'lion.jpg' 
img = io.imread(img_name)    

labels = segmentation.slic(img, compactness=10, n_segments=500)

因此我们有一个2D数组标签。

我的目标是运行循环

for label in labels:

并将每个标签的子阵列剪切为正方形,其中像素标记为零,剩余像素为零。

谢谢, Franek

0 个答案:

没有答案