wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-09-09 11:33:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

ENH Tony's comments on random walker

Browse Source
This commit is contained in:
emmanuelle
2012-01-23 19:47:16 +01:00
parent 6559b8f31d
commit 46d330591f
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/segmentation/random_walker_segmentation.py
+6 -4
View File
@@ -46,8 +46,10 @@ def _make_graph_edges_3d(n_x, n_y, n_z):
Returns
-------
edges : ndarray of shape (2, n_x * n_y * (nz - 1) +
n_x * (n_y - 1) * nz + (n_x - 1) * n_y * nz)
edges : (2, N) ndarray
with the total number of edges N = n_x * n_y * (nz - 1) +
n_x * (n_y - 1) * nz +
(n_x - 1) * n_y * nz
Graph edges with each column describing a node-id pair.
"""
vertices = np.arange(n_x * n_y * n_z).reshape((n_x, n_y, n_z))
@@ -102,8 +104,8 @@ def _clean_labels_ar(X, labels):
def _buildAB(lap_sparse, labels):
"""
Build the matrix A and rhs B of the linear system to solve.
A and B are two block of the laplacian of the image graph.
Build the matrix A and rhs B of the linear system to solve.
A and B are two block of the laplacian of the image graph.
"""
labels = labels[labels >= 0]
indices = np.arange(labels.size)