Question

双通连通分量算法检测一个图像中的单独分量，并且在每次检测后，我将每个component保存为不同的图像。要在单独的图像上显示每个component，我使用多个if条件，但只要每个组件的图像中有很多形状，这些if conditions都在增加，到目前为止我已经使用了7条件，但它是increaing。任何想法如何使用循环或如何处理它。

 for (x, y) in labels:
            component = uf.find(labels[(x, y)])
            labels[(x, y)] = component
            ############################################################
            if labels[(x, y)]==0:
                Zero[y][x]=int(255)
                count=count+1
                if count<=43:
                    continue
                elif count>43:
                    Zeroth = Image.fromarray(Zero)
                    Zeroth.save(os.path.join(dirs, 'Zero.png'), 'png')
            #############################################################
            if labels[(x, y)]==1:
                One[y][x]=int(255)
                count1=count1+1
                if count1<=43:
                    continue
                elif count1>43:
                    First = Image.fromarray(One)
                    First.save(os.path.join(dirs, 'First.png'),'png')

Answer 1

我不确定我是否完全理解您的代码，但我认为可能有一个简单的解决方法可以解决太多变量和if语句的问题。您应该将它们放在列表中并索引这些列表以获取要更新和保存的值，而不是使用单独的变量和代码来保存每个图像。

以下是您的代码的查找方式：

# at some point above, create an "images" list instead of separate Zero, One, etc variables
# also create a "counts" list instead of count, count1, etc

    for (x, y) in labels:
        component = uf.find(labels[(x, y)])
        labels[(x, y)] = component

        # optionally, add code here to create a new image if needed

        images[component][y][x] = 255   # update image
        counts[component] += 1          # update count

        if counts[component] > 43:      # if count is high enough, save out image
            img = images[component] = Image.fromarray(Zero)
            img.save(os.path.join(dirs, 'image{:02d}.png'.format(component), 'png')

请注意，您需要以编程方式生成图片文件名，因此我选择Zero.png和One.png代替image00.png和image01.png等。如果你想保留相同的名称系统，你可以创建一个从数字到英文名称的映射，但我怀疑使用数字会更方便你以后使用。

如果您不知道提前需要多少图像和计数，可以在循环中添加一些额外的逻辑，根据需要创建新的逻辑，并将它们附加到images和counts列表。我在上面的代码中添加了一条注释，显示了您想要的逻辑。

两个通过连接组件，组件数量问题

1 个答案: