Spaces:
Runtime error
Runtime error
File size: 979 Bytes
153628e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
# Copyright (C) 2021-2024, Mindee.
# This program is licensed under the Apache License 2.0.
# See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.
import tensorflow as tf
__all__ = ["erode", "dilate"]
def erode(x: tf.Tensor, kernel_size: int) -> tf.Tensor:
"""Performs erosion on a given tensor
Args:
----
x: boolean tensor of shape (N, H, W, C)
kernel_size: the size of the kernel to use for erosion
Returns:
-------
the eroded tensor
"""
return 1 - tf.nn.max_pool2d(1 - x, kernel_size, strides=1, padding="SAME")
def dilate(x: tf.Tensor, kernel_size: int) -> tf.Tensor:
"""Performs dilation on a given tensor
Args:
----
x: boolean tensor of shape (N, H, W, C)
kernel_size: the size of the kernel to use for dilation
Returns:
-------
the dilated tensor
"""
return tf.nn.max_pool2d(x, kernel_size, strides=1, padding="SAME")
|