Function handleClickTrigger

tfjs-tflite/demo/src/script.ts:89–113

(
    trigger: HTMLElement, tfliteModel: tflite.TFLiteModel)

}

function handleClickTrigger(
    trigger: HTMLElement, tfliteModel: tflite.TFLiteModel) {
  // Get the source media (either a picture or the cam video).
  const imageContainer = trigger.closest('.img-container')!;
  let srcMedia: HTMLImageElement|HTMLVideoElement =
      imageContainer.querySelector('img')!;
  if (!srcMedia) {
    srcMedia = imageContainer.querySelector('video')!;
    removeClass('.take-pic', 'hide');
  }

  // Run inference and draw the result on the corresponding canvas.
  const canvas = imageContainer.querySelector('canvas')!;
  const ctx = canvas.getContext('2d')!;
  const inferenceStart = Date.now();
  const imageData = cartoonize(tfliteModel, srcMedia);
  const latency = Date.now() - inferenceStart;
  ctx.putImageData(imageData, 0, 0);
  canvas.classList.add('show');

  // Show latency stat.
  const stats = trigger.closest('.img-container')!.querySelector('.stats')!;
  stats.classList.add('show');
  stats.innerHTML = latency.toFixed(1) + ' ms';
}

function cartoonize(
    tfliteModel: tflite.TFLiteModel,