img_to_graph

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

Граф связей градиента от пикселя к пикселю.

Ребра взвешены значениями градиента.

Подробнее в Руководство пользователя.

Параметры:

imgмассивоподобный формы (высота, ширина) или (высота, ширина, канал)

2D или 3D изображение.

маскаndarray формы (высота, ширина) или (высота, ширина, канал), dtype=bool, по умолчанию=None

Необязательная маска изображения, чтобы учитывать только часть пикселей.

return_asnp.ndarray или класс разреженной матрицы, по умолчанию sparse.coo_matrix

Класс, используемый для построения возвращаемой матрицы смежности.

dtypedtype, по умолчанию=None

Данные возвращаемой разреженной матрицы. По умолчанию это dtype изображения.

Возвращает:

графndarray или класс разреженной матрицы

Вычисленная матрица смежности.

Примеры

>>> import numpy as np
>>> from sklearn.feature_extraction.image import img_to_graph
>>> img = np.array([[0, 0], [0, 1]])
>>> img_to_graph(img, return_as=np.ndarray)
array([[0, 0, 0, 0],
       [0, 0, 0, 1],
       [0, 0, 0, 1],
       [0, 1, 1, 1]])