sklearn.feature_extraction.img_to_graph

sklearn.feature_extraction.image.img_to_graph(img, *, mask=None, return_as=<class 'scipy.sparse.coo.coo_matrix'>, dtype=None)

[源码]

图的像素到像素的梯度连接

边缘用梯度值加权。

用户指南中阅读更多内容。

参数 说明
img ndarray of shape (height, width) or (height, width, channel)
2D或3D图像。
mask ndarray of shape (height, width) or (height, width, channel), dtype=bool, default=None
可选的图像遮罩,只考虑部分像素。
return_as np.ndarray or a sparse matrix class, default=sparse.coo_matrix
用来构建返回邻接矩阵的类。
dtype dtype, default=None
返回稀疏矩阵的数据。默认情况下,它是img的dtype

注意

对于scikit-learn版本0.14.1和之前版本,return_as=np。ndarray是通过返回稠密np来处理的。矩阵实例。展望未来,np。ndarray返回一个np。ndarray,如预期。

为了兼容性,依赖于此方法的用户代码应该将其调用封装在np中。避免类型问题。