fixed memory leaks

justadudewhohacks · justadudewhohacks · commit b23e637636c4 · 2018-06-09T09:20:43.000+02:00
diff --git a/src/faceRecognitionNet/index.ts b/src/faceRecognitionNet/index.ts
@@ -42,8 +42,19 @@ export function faceRecognitionNet(weights: Float32Array) {
     })
   }
 
-  const computeFaceDescriptor = (input: number[] | ImageData) => forward(input).data()
-  const computeFaceDescriptorSync = (input: number[] | ImageData) => forward(input).dataSync()
+  const computeFaceDescriptor = async (input: number[] | ImageData) => {
+    const result = forward(input)
+    const data = await result.data()
+    result.dispose()
+    return data
+  }
+
+  const computeFaceDescriptorSync = (input: number[] | ImageData) => {
+    const result = forward(input)
+    const data = result.dataSync()
+    result.dispose()
+    return data
+  }
 
   return {
     computeFaceDescriptor,
