Función Tensorflow.js tf.dilation2d()

Tensorflow.js es una biblioteca de código abierto desarrollada por Google para ejecutar modelos de aprendizaje automático, así como redes neuronales de aprendizaje profundo en el entorno del navegador o del Node.

La función .dilation2d() se usa para evaluar la dilatación en escala de grises sobre el tensor de entrada indicado.

Sintaxis:

tf.dilation2d(x, filter, strides, pad, dilations?, dataFormat?)

Parámetros:

• x: El tensor de entrada indicado que es de rango 3 o de rango 4 y de forma: [lote, altura, ancho, canales]. Además, en caso de que el rango sea 3, entonces se presume el lote de tamaño 1. Puede ser del tipo tf.Tensor3D, tf.Tensor4D, TypedArray o Array.
• filter: El tensor de filtro indicado de rango 3 y forma: [filterHeight, filterWidth, depth]. Puede ser de tipo tf.Tensor3D, TypedArray o Array.
• zancadas: Las zancadas indicadas de la ventana deslizante para cada tamaño del tensor de forma de entrada dado: [strideHeight, strideWidth]. En caso de que los pasos indicados sean un solo número, entonces strideHeight == strideWidth. Puede ser de tipo [número, número] o número.
• pad: el tipo de algoritmo indicado para el relleno. Puede ser de tipo válido o igual.
  1. Aquí, para mismo y zancada 1, la salida tendría el mismo tamaño que la entrada, independientemente del tamaño del filtro.
  2. Para ‘válido’, la salida será más pequeña que la entrada en caso de que el tamaño del filtro sea mayor que 1*1×1.
• dilataciones: Las tasas de dilatación indicadas: [dilationHeight, dilatationWidth] en el sentido de que los valores de entrada se muestrean sobre las dimensiones de altura y anchura a favor de la dilatación morfológica atrosa. El valor predeterminado es [1, 1]. Además, en caso de que las dilataciones sean un solo número, dilataciónAltura == dilataciónAncho. Y si es mayor que 1, entonces todos los valores de los pasos deben ser 1. Es opcional y es de tipo [número, número], número.
• dataFormat: especifica el formato de datos de la entrada indicada, así como los datos de salida. El valor predeterminado es ‘NHWC’. Además, los datos aquí se almacenan en el orden de: [lote, alto, ancho, canales]. Es opcional y es de tipo ‘NHWC’.

Valor devuelto: Devuelve tf.Tensor3D o tf.Tensor4D.

Ejemplo 1:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Defining input tensor
const x = tf.tensor3d([1, 2, 3, 4], [2, 2, 1]);
 
// Defining filter tensor
const y = tf.tensor3d([1, 1, 0, 4], [1, 1, 4]);
 
// Calling dilation2d() method
const result = tf.dilation2d(x, y, 2, 'valid');
 
// Printing output
result.print();

Producción:

Tensor
     [ [[2],]]

Ejemplo 2:

// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
 
// Calling dilation2d() method with
// all its parameters
tf.tensor3d([1.1, 2.2, 3.3, 4.4], [2, 2, 1]).dilation2d(
 tf.tensor3d([1.3, 1.2, null, -4], [1, 1, 4]),
             2, 'valid', [3, 2], 'NHWC').print();

Producción:

Tensor
     [ [[2.4000001],]]

Referencia: https://js.tensorflow.org/api/latest/#dilation2d