1
2
3
4
5
6
7
8
9
10
11
12
13
| import * as tf from '../../dist/tfjs.esm';
|
| export function normalize(x: tf.Tensor4D, meanRgb: number[]): tf.Tensor4D {
| return tf.tidy(() => {
| const [r, g, b] = meanRgb;
| const avg_r = tf.fill([...x.shape.slice(0, 3), 1], r, 'float32');
| const avg_g = tf.fill([...x.shape.slice(0, 3), 1], g, 'float32');
| const avg_b = tf.fill([...x.shape.slice(0, 3), 1], b, 'float32');
| const avg_rgb = tf.concat([avg_r, avg_g, avg_b], 3);
|
| return tf.sub(x, avg_rgb);
| });
| }
|
|
