TensorFlow 2 version
|
View source on GitHub
|
Returns a tensor holding Sobel edge maps.
tf.image.sobel_edges(
image
)
Arguments | |
|---|---|
image
|
Image tensor with shape [batch_size, h, w, d] and type float32 or float64. The image(s) must be 2x2 or larger. |
Returns | |
|---|---|
| Tensor holding edge maps for each channel. Returns a tensor with shape [batch_size, h, w, d, 2] where the last two dimensions hold [[dy[0], dx[0]], [dy[1], dx[1]], ..., [dy[d-1], dx[d-1]]] calculated using the Sobel filter. |