sklearn.feature_extraction.image.img_to_graph¶
- sklearn.feature_extraction.image.img_to_graph(img, mask=None, return_as=<class 'scipy.sparse.coo.coo_matrix'>, dtype=None)¶
Graph of the pixel-to-pixel gradient connections
Edges are weighted with the gradient values.
Parameters : img: ndarray, 2D or 3D :
2D or 3D image
mask : ndarray of booleans, optional
An optional mask of the image, to consider only part of the pixels.
return_as: np.ndarray or a sparse matrix class, optional :
The class to use to build the returned adjacency matrix.
dtype: None or dtype, optional :
The data of the returned sparse matrix. By default it is the dtype of img