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

use as strided

Browse Source
This commit is contained in:
Vighnesh Birodkar
2016-01-06 16:23:43 -05:00
parent a5f6db59cb
commit ea18fcb99b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/future/graph/rag.py
+2 -2
View File
@@ -354,8 +354,8 @@ def rag_boundary(labels, edge_map, connectivity=2):
n = np.max(labels_large) + 1
# use a dummy broadcast array as data for RAG
ones = np.broadcast_to(np.ones((1,), dtype=np.int_),
labels_small.shape)
ones = as_strided(np.ones((1,), dtype=np.float), shape=labels_small.shape,
strides=(0,))
count_matrix = sparse.coo_matrix((ones, (labels_small, labels_large)),
dtype=np.int_, shape=(n, n)).tocsr()
data = np.concatenate((edge_map[boundaries0], edge_map[boundaries1]))