2020-10-09 03:31:31 +02:00
|
|
|
import * as tf from '@tensorflow/tfjs';
|
2020-08-18 13:54:53 +02:00
|
|
|
|
|
|
|
|
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)
|
|
|
|
|
const avg_g = tf.fill([...x.shape.slice(0, 3), 1], g)
|
|
|
|
|
const avg_b = tf.fill([...x.shape.slice(0, 3), 1], b)
|
|
|
|
|
const avg_rgb = tf.concat([avg_r, avg_g, avg_b], 3)
|
|
|
|
|
|
|
|
|
|
return tf.sub(x, avg_rgb)
|
|
|
|
|
})
|
|
|
|
|
}
|
