fix: truncated-normal generator

Author: piercus

lib/backend/tensorflow-backend.js

@@ -188,9 +188,9 @@ class TensorflowBackend {
 
 	addNoiseOne(img, noiseImg) {
 		return this._tf.tidy(() => {
-			const black = this._tf.zerosLike(img);
-			const white = this._tf.onesLike(img).mul(255);
-			const res = img.add(noiseImg).maximum(black).minimum(white).toInt();
+			const black = this._tf.zerosLike(img, 'int32');
+			const white = this._tf.onesLike(img, 'int32').mul(255);
+			const res = img.add(noiseImg.round().toInt()).maximum(black).minimum(white);
 			return res;
 		});
 	}

lib/generators/truncated-normal-noise.js

@@ -80,6 +80,8 @@ new ia.TruncatedNormalNoiseGenerator({
 class TruncatedNormalNoiseGenerator extends GaussianNoiseGenerator {
 	buildHasard(o) {
 		if (typeof (this.backend.truncatedNormal) === 'function') {
+			// Tensorflow specific code
+			// TO DO move this into tfjs backend folder
 			const imagesProps = h.array({
 				size: o.nImages,
 				value: h.object({

test/augmenters/additive-truncated-normal-noise.js

@@ -7,7 +7,7 @@ const nImages = 5;
 
 test('additiveTruncatedNormalNoise not perChannel', macroAugmenter, AdditiveTruncatedNormalNoise, {
 	inputFilenames: new Array(nImages).fill('lenna.png'),
-	// BackendLibs: [require('opencv4nodejs')],
+	backendLibs: [require('@tensorflow/tfjs-node')],
 	expectImg(t, mats1, mats2, backend) {
 		const metadatas = backend.getMetadata(mats1);
 		const metadata = metadatas[0];
@@ -23,8 +23,8 @@ test('additiveTruncatedNormalNoise not perChannel', macroAugmenter, AdditiveTrun
 		const m2 = backend.imageToArray(mats2);
 
 		backend.forEachPixel(diff, ([b, g, r], batchIndex, rowIndex, colIndex) => {
-			// Console.log(m2[batchIndex] && m2[batchIndex][rowIndex])
 			if (m2[batchIndex][rowIndex][colIndex].slice(0, 3).indexOf(255) === -1 && m2[batchIndex][rowIndex][colIndex].slice(0, 3).indexOf(0) === -1 && (r !== g || g !== b)) {
+				console.log(m2[batchIndex] && m2[batchIndex][rowIndex][colIndex], [r, g, b]);
 				count++;
 			}
 		});