我使用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