Method classify

electron/image_classifier.js:53–77

* Perform classification on a batch of image tensors. * * @param {tf.Tensor} images Batch image tensor of shape * `[numExamples, height, width, channels]`. The values of `height`, * `width` and `channel` must match the underlying MobileNetV2 model * (default: 224, 224, 3). *

(images, topK = 5)

   *   of `images`.
   */
  async classify(images, topK = 5) {
    await this.ensureModelLoaded();
    return tf.tidy(() => {
      const probs = this.model.predict(images);
      const sorted = true;
      const {values, indices} = tf.topk(probs, topK, sorted);

      const classProbs = values.arraySync();
      const classIndices = indices.arraySync();

      const results = [];
      classIndices.forEach((indices, i) => {
        const classesAndProbs = [];
        indices.forEach((index, j) => {
          classesAndProbs.push({
            className: IMAGENET_CLASSES[index],
            prob: classProbs[i][j]
          });
        });
        results.push(classesAndProbs);
      })

      return results;
    });
  }

  /**
   * If the underlying model is not loaded, load it.