Fixed memory management bug.

novaia · May 12, 2023 · 96fa23f · 96fa23f
1 parent 296ce7a
commit 96fa23f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/Assets/NeuralTerrainGeneration/Scripts/NeuralTerrainGeneratorTool.cs b/Assets/NeuralTerrainGeneration/Scripts/NeuralTerrainGeneratorTool.cs
@@ -114,9 +114,9 @@ private void GenerateMaskedBrushHeightmap()
                 modelOutputHeight,
                 upSampleFactor, 
                 true,
-                WorkerFactory.Type.ComputePrecompiled
+                workerType
             );
-            brushMaskTensor = barraUpSampler.Execute(brushMaskTensor);
+            Tensor upSampledBrushMaskTensor = barraUpSampler.Execute(brushMaskTensor);
             // Consider smoothing upsample brush mask, otherwise it makes heightmap jagged.
 
             /*
@@ -136,17 +136,17 @@ private void GenerateMaskedBrushHeightmap()
 
             for(int i = 0; i < brushHeightmapColors.Length; i++)
             {
-                Color brushMaskColor = new Color(brushMaskTensor[i], 0, 0, 1);
+                Color brushMaskColor = new Color(upSampledBrushMaskTensor[i], 0, 0, 1);
                 brushHeightmapMaskedColors[i] = brushHeightmapColors[i] * brushMaskColor.r;
             }
 
             brushMaskTensor.Dispose();
+            upSampledBrushMaskTensor.Dispose();
             barraUpSampler.Dispose();
 
             brushHeightmapMasked = new Texture2D(brushHeightmap.width, brushHeightmap.height);
             brushHeightmapMasked.SetPixels(brushHeightmapMaskedColors);
             brushHeightmapMasked.Apply();
-            Debug.Log(brushHeightmapMasked.width);
         }
 
         public override void OnInspectorGUI(Terrain terrain, IOnInspectorGUI editContext)