Fixed bug in algorithm when changing images

episphere · Jan 9, 2024 · 9f38be8 · 9f38be8
1 parent 7d7bcc3
commit 9f38be8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/UI.js b/UI.js
@@ -253,6 +253,7 @@ function resetApplication() {
   window.sortedCoresData = [];
   window.loadedImg = null;
   window.preprocessingData = null;
+  window.neuralNetworkResult = null;
 
   // Reset sliders and output elements to their default values
   // resetSlidersAndOutputs();

diff --git a/drawCanvas.js b/drawCanvas.js
@@ -207,7 +207,7 @@ function findNearestCoreIndex(x, y) {
   return nearestIndex;
 }
 
-let neuralNetworkResult = null;
+window.neuralNetworkResult = null;
 
 async function visualizeSegmentationResults(
   originalImage,
@@ -220,17 +220,18 @@ async function visualizeSegmentationResults(
     originalImage.naturalWidth,
     originalImage.naturalHeight,
   ];
+
   const canvas = document.getElementById(canvasID);
   const ctx = canvas.getContext("2d");
   canvas.width = width;
   canvas.height = height;
 
   ctx.drawImage(originalImage, 0, 0, width, height);
 
-  if (!neuralNetworkResult) {
-    neuralNetworkResult = await processPredictions(predictions);
+  if (!window.neuralNetworkResult) {
+    window.neuralNetworkResult = await processPredictions(predictions);
   }
-  drawMask(ctx, neuralNetworkResult, alpha, width, height);
+  drawMask(ctx, window.neuralNetworkResult, alpha, width, height);
   drawProperties(ctx, properties);
 
   addSegmentationCanvasEventListeners(canvas);