From acf6bead2126dafda6b1620214c699dd3f58eef4 Mon Sep 17 00:00:00 2001
From: Kozyrev Vladislav <76097937+dalvine@users.noreply.github.com>
Date: Wed, 3 May 2023 10:34:08 +0300
Subject: [PATCH] fix memory leak in histogramEqualization

Bug was introduced in cc4650c after rgb variable had been renamed.
---
 src/image/enhance.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/image/enhance.ts b/src/image/enhance.ts
index dc3e098e..65ad6c9b 100644
--- a/src/image/enhance.ts
+++ b/src/image/enhance.ts
@@ -24,7 +24,7 @@ export async function histogramEqualization(inputImage: Tensor): Promise<Tensor>
     const enh = [tf.mul(sub[0], factor), tf.mul(sub[1], factor), tf.mul(sub[2], factor)];
     const stack = tf.stack([enh[0], enh[1], enh[2]], 2);
     final = tf.reshape(stack, [1, squeeze.shape[0] || 0, squeeze.shape[1] || 0, 3]);
-    tf.dispose([...sub, ...range, ...enh]);
+    tf.dispose([...sub, ...range, ...enh, stack]);
   } else {
     final = tf.expandDims(squeeze, 0);
   }