Function customLayerDemo

custom-layer/index.js:22–34

()

import {antirectifier} from './custom_layer';

function customLayerDemo() {
  let imgElement = document.getElementById('cat');
  // Layer expects first dimension to be batch, therefore expandDims.
  const img = tf.browser.fromPixels(imgElement).toFloat().expandDims(0);
  const layer = antirectifier();
  const [posTensor, negTensor] = tf.split(layer.apply(img), 2, 3);
  const posCanvas = document.createElement('canvas');
  tensorToCanvas(posTensor, posCanvas);
  document.getElementById('output_image_1').appendChild(posCanvas);
  const negCanvas = document.createElement('canvas');
  tensorToCanvas(negTensor, negCanvas);
  document.getElementById('output_image_2').appendChild(negCanvas);
}

function tensorToCanvas(tensor, canvas) {
  const ctx = canvas.getContext('2d');