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