如何从图像生成图形，以像素作为节点和条件边

Question

我有一个 0 ad 1 的二值灰度图像。我想创建一个图表，其中每个节点都是一个像素（值为 1），并在每个像素之间创建一条边缘，这些像素的隐含距离低于阈值。

基本上我使用的是networkx库，但是我实现它的方式非常非常慢（边缘构建的部分）。有没有更快的代码实现（更低的复杂度）？

G = nx.Graph()

rows = img.shape[0]
columns = img.shape[1]

# Generatig Nodes
for x in range(0, rows):
    for y in range(0, columns):
        if img[x, y] == 1:
            G.add_node(n, px=(x, y))

# Generating Edges based on the max distance allowed
max_dist = 10  # threshold 
for x in G.nodes:
    for y in G.nodes:
        dist = distance.euclidean(G.nodes[x]['px'], G.nodes[y]['px'])
        if dist <= max_dist:
            G.add_edge(x, y, weight=dist)

请注意，我保存了像素标签，因为我需要将图形重新转换为图像