如何在python中提取skimage骨架信息到NetworkX节点和边以进行进一步的高级分析

目前，我在Python中使用skimage从一些二值化地图中提取开放空间的骨架，如下图所示，

使用以下 python 代码：

from skimage.morphology import skeletonize
from skimage import draw
from skimage.io import imread, imshow
from skimage.color import rgb2gray

# load image from file
img_fname=os.path.join('images','mall1_2F_schema.png') 
image=imread(img_fname)

# Change RGB color to gray 
image=rgb2gray(image)

# Change gray image to binary
image=np.where(image>np.mean(image),1.0,0.0)

# perform skeletonization
skeleton = skeletonize(image)

现在我想从骨架中提取端点和交叉点作为 Networkx 图对象的节点，同时计算相邻节点之间的距离作为 Networkx 图对象的边。目前，我必须手动获取点坐标并输入到 NetworkX 对象初始化过程中，我们是否有更智能的方法来自动完成所有操作？

顺便说一句，我发现亚龙·卡汉诺维奇 https://stackoverflow.com/users/1811651/yaron-kahanovitch的回答stackoverflow上的问题 https://stackoverflow.com/questions/27648195/how-to-find-the-length-of-a-path-curve-in-a-picture#提出了类似的方法。不过，没有给出更多的实现建议，我认为 NetworkX 可能是一种方法。

非常感谢您的任何建议。

干得好。

这段代码适用于 2D 骨架图像，您可以轻松地将其扩展到 3D。

import networkx as nx
import numpy as np

def skeleton_image_to_graph(skeIm, connectivity=2):
    assert(len(skeIm.shape) == 2)
    skeImPos = np.stack(np.where(skeIm))
    skeImPosIm = np.zeros_like(skeIm, dtype=np.int)
    skeImPosIm[skeImPos[0], skeImPos[1]] = np.arange(0, skeImPos.shape[1])
    g = nx.Graph()
    if connectivity == 1:
        neigh = np.array([[0, 1], [0, -1], [1, 0], [-1, 0]])
    elif connectivity == 2:
        neigh = np.array([[0, 1], [0, -1], [1, 0], [-1, 0], [1, 1], [1, -1], [-1, 1], [-1, -1]])
    else:
        raise ValueError(f'unsupported connectivity {connectivity}')
    for idx in range(skeImPos[0].shape[0]):
        for neighIdx in range(neigh.shape[0]):
            curNeighPos = skeImPos[:, idx] + neigh[neighIdx]
            if np.any(curNeighPos<0) or np.any(curNeighPos>=skeIm.shape):
                continue
            if skeIm[curNeighPos[0], curNeighPos[1]] > 0:
                g.add_edge(skeImPosIm[skeImPos[0, idx], skeImPos[1, idx]], skeImPosIm[curNeighPos[0], curNeighPos[1]], weight=np.linalg.norm(neigh[neighIdx]))
    g.graph['physicalPos'] = skeImPos.T
    return g