normalize.js 696 B

12345678910111213141516
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. var tslib_1 = require("tslib");
  4. var tf = require("@tensorflow/tfjs-core");
  5. function normalize(x, meanRgb) {
  6. return tf.tidy(function () {
  7. var r = meanRgb[0], g = meanRgb[1], b = meanRgb[2];
  8. var avg_r = tf.fill(tslib_1.__spreadArrays(x.shape.slice(0, 3), [1]), r);
  9. var avg_g = tf.fill(tslib_1.__spreadArrays(x.shape.slice(0, 3), [1]), g);
  10. var avg_b = tf.fill(tslib_1.__spreadArrays(x.shape.slice(0, 3), [1]), b);
  11. var avg_rgb = tf.concat([avg_r, avg_g, avg_b], 3);
  12. return tf.sub(x, avg_rgb);
  13. });
  14. }
  15. exports.normalize = normalize;
  16. //# sourceMappingURL=normalize.js.map