Method predict

deeplab/src/index.ts:127–132

* Segments an arbitrary image and generates a two-dimensional tensor with * class labels assigned to each cell of the grid overlayed on the image ( the * maximum number of cells on the side is fixed to 513). * * @param input :: * `ImageData|HTMLImageElement|HTMLCanvasElement|HTMLV

(input: DeepLabInput)

   * The segmentation map of the image
   */
  public predict(input: DeepLabInput): tf.Tensor2D {
    return tf.tidy(() => {
      const data = tf.cast(toInputTensor(input), 'int32');
      return tf.squeeze(this.model.execute(data) as tf.Tensor);
    });
  }

  /**
   * Segments an arbitrary image and generates a two-dimensional tensor with