normalize.js 581 B

12345678910111213
  1. import { __spreadArrays } from "tslib";
  2. import * as tf from '@tensorflow/tfjs-core';
  3. export function normalize(x, meanRgb) {
  4. return tf.tidy(function () {
  5. var r = meanRgb[0], g = meanRgb[1], b = meanRgb[2];
  6. var avg_r = tf.fill(__spreadArrays(x.shape.slice(0, 3), [1]), r);
  7. var avg_g = tf.fill(__spreadArrays(x.shape.slice(0, 3), [1]), g);
  8. var avg_b = tf.fill(__spreadArrays(x.shape.slice(0, 3), [1]), b);
  9. var avg_rgb = tf.concat([avg_r, avg_g, avg_b], 3);
  10. return tf.sub(x, avg_rgb);
  11. });
  12. }
  13. //# sourceMappingURL=normalize.js.map