Commit b23e6376 by vincent

fixed memory leaks

parent 674e0cda
...@@ -42,8 +42,19 @@ export function faceRecognitionNet(weights: Float32Array) { ...@@ -42,8 +42,19 @@ export function faceRecognitionNet(weights: Float32Array) {
}) })
} }
const computeFaceDescriptor = (input: number[] | ImageData) => forward(input).data() const computeFaceDescriptor = async (input: number[] | ImageData) => {
const computeFaceDescriptorSync = (input: number[] | ImageData) => forward(input).dataSync() 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 { return {
computeFaceDescriptor, computeFaceDescriptor,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment